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(objs[0].tenant_id, 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 agent_completion(objs[0].tenant_id, dialog_id, **req):
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",