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

增加post接口

parent 40105bef
......@@ -12,6 +12,7 @@ from config.system import config
from model.entity.AiChatPermissionEntity import AiChatPermissionEntity
from model.entity.AiChatRecommendQuestionEntity import AiChatRecommendQuestionEntity
from model.entity.AiChatRecordEntity import AiChatRecordEntity
from model.param.AiChatParam import AiChatParam
from model.vo.ResultVo import ResultVo
......@@ -99,6 +100,16 @@ async def chat(request: Request, message: str):
return EventSourceResponse(mcp_client_instance.process_query(message, id_str.split("-")[0]),
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__":
cors = config.get("cors", {})
......
from pydantic import BaseModel
class AiChatRecordParam(BaseModel):
id: int
answer: str
class AiChatParam(BaseModel):
message: 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