diff --git a/api/apps/sdk/session.py b/api/apps/sdk/session.py index 80f4228dc..1716db9ee 100644 --- a/api/apps/sdk/session.py +++ b/api/apps/sdk/session.py @@ -388,10 +388,10 @@ def agents_completion_openai_compatibility (tenant_id, agent_id): question = next((m["content"] for m in reversed(messages) if m["role"] == "user"), "") if req.get("stream", True): - return Response(completionOpenAI(tenant_id, agent_id, question, session_id=req.get("id", ""), stream=True), mimetype="text/event-stream") + return Response(completionOpenAI(tenant_id, agent_id, question, session_id=req.get("id", req.get("metadata", {}).get("id","")), stream=True), mimetype="text/event-stream") else: # For non-streaming, just return the response directly - response = next(completionOpenAI(tenant_id, agent_id, question, session_id=req.get("id", ""), stream=False)) + response = next(completionOpenAI(tenant_id, agent_id, question, session_id=req.get("id", req.get("metadata", {}).get("id","")), stream=False)) return jsonify(response)