AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-14 06:45:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix misspell. (#4219)

Browse Source 
### What problem does this PR solve?
#4216

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu 2024-12-25 10:48:59 +08:00 committed by GitHub
parent 4e5f92f01b
commit bc3288d390
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 33 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
api/apps/sdk/session.py
View File

@ -71,7 +71,7 @@ def create_agent_session(tenant_id, agent_id):
if not e: if not e:
return get_error_data_result("Agent not found.") return get_error_data_result("Agent not found.")
if not UserCanvasService.query(user_id=tenant_id,id=agent_id): if not UserCanvasService.query(user_id=tenant_id, id=agent_id):
return get_error_data_result("You cannot access the agent.") return get_error_data_result("You cannot access the agent.")
if not isinstance(cvs.dsl, str): if not isinstance(cvs.dsl, str):
@ -96,7 +96,7 @@ def create_agent_session(tenant_id, agent_id):
conv = { conv = {
"id": get_uuid(), "id": get_uuid(),
"dialog_id": cvs.id, "dialog_id": cvs.id,
"user_id": req.get("usr_id","") if isinstance(req, dict) else "", "user_id": req.get("user_id", "") if isinstance(req, dict) else "",
"message": [{"role": "assistant", "content": canvas.get_prologue()}], "message": [{"role": "assistant", "content": canvas.get_prologue()}],
"source": "agent", "source": "agent",
"dsl": cvs.dsl "dsl": cvs.dsl
@ -133,11 +133,11 @@ def update(tenant_id, chat_id, session_id):
def chat_completion(tenant_id, chat_id): def chat_completion(tenant_id, chat_id):
req = request.json req = request.json
if not req or not req.get("session_id"): if not req or not req.get("session_id"):
req = {"question":""} req = {"question": ""}
if not DialogService.query(tenant_id=tenant_id,id=chat_id,status=StatusEnum.VALID.value): if not DialogService.query(tenant_id=tenant_id, id=chat_id, status=StatusEnum.VALID.value):
return get_error_data_result(f"You don't own the chat {chat_id}") return get_error_data_result(f"You don't own the chat {chat_id}")
if req.get("session_id"): if req.get("session_id"):
if not ConversationService.query(id=req["session_id"],dialog_id=chat_id): if not ConversationService.query(id=req["session_id"], dialog_id=chat_id):
return get_error_data_result(f"You don't own the session {req['session_id']}") return get_error_data_result(f"You don't own the session {req['session_id']}")
if req.get("stream", True): if req.get("stream", True):
resp = Response(rag_completion(tenant_id, chat_id, **req), mimetype="text/event-stream") resp = Response(rag_completion(tenant_id, chat_id, **req), mimetype="text/event-stream")
@ -164,16 +164,16 @@ def agent_completions(tenant_id, agent_id):
return get_error_data_result(f"You don't own the agent {agent_id}") return get_error_data_result(f"You don't own the agent {agent_id}")
if req.get("session_id"): if req.get("session_id"):
dsl = cvs[0].dsl dsl = cvs[0].dsl
if not isinstance(dsl,str): if not isinstance(dsl, str):
dsl = json.dumps(dsl) dsl = json.dumps(dsl)
canvas=Canvas(dsl,tenant_id) canvas = Canvas(dsl, tenant_id)
if canvas.get_preset_param(): if canvas.get_preset_param():
req["question"]="" req["question"] = ""
conv = API4ConversationService.query(id=req["session_id"], dialog_id=agent_id) conv = API4ConversationService.query(id=req["session_id"], dialog_id=agent_id)
if not conv: if not conv:
return get_error_data_result(f"You don't own the session {req['session_id']}") return get_error_data_result(f"You don't own the session {req['session_id']}")
else: else:
req["question"]="" req["question"] = ""
if req.get("stream", True): if req.get("stream", True):
resp = Response(agent_completion(tenant_id, agent_id, **req), mimetype="text/event-stream") resp = Response(agent_completion(tenant_id, agent_id, **req), mimetype="text/event-stream")
resp.headers.add_header("Cache-control", "no-cache") resp.headers.add_header("Cache-control", "no-cache")
@ -447,5 +447,3 @@ def agent_bot_completions(agent_id):
for answer in agent_completion(objs[0].tenant_id, agent_id, **req): for answer in agent_completion(objs[0].tenant_id, agent_id, **req):
return get_result(data=answer) return get_result(data=answer)