Kevin Hu
commited on
Commit
·
d657ebb
1
Parent(s):
f76d985
Fix session API issues. (#3939)
Browse files### What problem does this PR solve?
### Type of change
- [x] Bug Fix (non-breaking change which fixes an issue)
api/apps/sdk/session.py
CHANGED
|
@@ -374,14 +374,14 @@ def chatbot_completions(dialog_id):
|
|
| 374 |
req["quote"] = False
|
| 375 |
|
| 376 |
if req.get("stream", True):
|
| 377 |
-
resp = Response(iframe_completion(
|
| 378 |
resp.headers.add_header("Cache-control", "no-cache")
|
| 379 |
resp.headers.add_header("Connection", "keep-alive")
|
| 380 |
resp.headers.add_header("X-Accel-Buffering", "no")
|
| 381 |
resp.headers.add_header("Content-Type", "text/event-stream; charset=utf-8")
|
| 382 |
return resp
|
| 383 |
|
| 384 |
-
for answer in
|
| 385 |
return get_result(data=answer)
|
| 386 |
|
| 387 |
|
|
|
|
| 374 |
req["quote"] = False
|
| 375 |
|
| 376 |
if req.get("stream", True):
|
| 377 |
+
resp = Response(iframe_completion(dialog_id, **req), mimetype="text/event-stream")
|
| 378 |
resp.headers.add_header("Cache-control", "no-cache")
|
| 379 |
resp.headers.add_header("Connection", "keep-alive")
|
| 380 |
resp.headers.add_header("X-Accel-Buffering", "no")
|
| 381 |
resp.headers.add_header("Content-Type", "text/event-stream; charset=utf-8")
|
| 382 |
return resp
|
| 383 |
|
| 384 |
+
for answer in iframe_completion(dialog_id, **req):
|
| 385 |
return get_result(data=answer)
|
| 386 |
|
| 387 |
|
api/db/services/canvas_service.py
CHANGED
|
@@ -89,6 +89,8 @@ def completion(tenant_id, agent_id, question, session_id=None, stream=True, **kw
|
|
| 89 |
assert e, "Session not found!"
|
| 90 |
canvas = Canvas(json.dumps(conv.dsl), tenant_id)
|
| 91 |
|
|
|
|
|
|
|
| 92 |
messages = conv.message
|
| 93 |
question = {
|
| 94 |
"role": "user",
|
|
|
|
| 89 |
assert e, "Session not found!"
|
| 90 |
canvas = Canvas(json.dumps(conv.dsl), tenant_id)
|
| 91 |
|
| 92 |
+
if not conv.message:
|
| 93 |
+
conv.message = []
|
| 94 |
messages = conv.message
|
| 95 |
question = {
|
| 96 |
"role": "user",
|
api/db/services/conversation_service.py
CHANGED
|
@@ -176,6 +176,8 @@ def iframe_completion(dialog_id, question, session_id=None, stream=True, **kwarg
|
|
| 176 |
e, conv = API4ConversationService.get_by_id(session_id)
|
| 177 |
assert e, "Session not found!"
|
| 178 |
|
|
|
|
|
|
|
| 179 |
messages = conv.message
|
| 180 |
question = {
|
| 181 |
"role": "user",
|
|
|
|
| 176 |
e, conv = API4ConversationService.get_by_id(session_id)
|
| 177 |
assert e, "Session not found!"
|
| 178 |
|
| 179 |
+
if not conv.message:
|
| 180 |
+
conv.message = []
|
| 181 |
messages = conv.message
|
| 182 |
question = {
|
| 183 |
"role": "user",
|