Commit a8552474 authored by 何家明's avatar 何家明

增加post接口

parent 40105bef
...@@ -12,6 +12,7 @@ from config.system import config ...@@ -12,6 +12,7 @@ from config.system import config
from model.entity.AiChatPermissionEntity import AiChatPermissionEntity from model.entity.AiChatPermissionEntity import AiChatPermissionEntity
from model.entity.AiChatRecommendQuestionEntity import AiChatRecommendQuestionEntity from model.entity.AiChatRecommendQuestionEntity import AiChatRecommendQuestionEntity
from model.entity.AiChatRecordEntity import AiChatRecordEntity from model.entity.AiChatRecordEntity import AiChatRecordEntity
from model.param.AiChatParam import AiChatParam
from model.vo.ResultVo import ResultVo from model.vo.ResultVo import ResultVo
...@@ -99,6 +100,16 @@ async def chat(request: Request, message: str): ...@@ -99,6 +100,16 @@ async def chat(request: Request, message: str):
return EventSourceResponse(mcp_client_instance.process_query(message, id_str.split("-")[0]), return EventSourceResponse(mcp_client_instance.process_query(message, id_str.split("-")[0]),
media_type="text/event-stream", headers={"Cache-Control": "no-cache"}) media_type="text/event-stream", headers={"Cache-Control": "no-cache"})
@router.post(path="/chat", description="AI对话")
async def chat(request: Request, param: AiChatParam):
id_str = request.headers.get("id", "")
if not id_str:
def error_generator():
yield ServerSentEvent(event="error")
return EventSourceResponse(error_generator())
return EventSourceResponse(mcp_client_instance.process_query(param.message, id_str.split("-")[0]),
media_type="text/event-stream", headers={"Cache-Control": "no-cache"})
if __name__ == "__main__": if __name__ == "__main__":
cors = config.get("cors", {}) cors = config.get("cors", {})
......
from pydantic import BaseModel from pydantic import BaseModel
class AiChatRecordParam(BaseModel): class AiChatParam(BaseModel):
id: int message: str
answer: str
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment