diff --git a/api/apps/sdk/session.py b/api/apps/sdk/session.py index 269198560..cdf07b7ed 100644 --- a/api/apps/sdk/session.py +++ b/api/apps/sdk/session.py @@ -898,7 +898,7 @@ def ask_about_embedded(): def stream(): nonlocal req, uid try: - for ans in ask(req["question"], req["kb_ids"], uid, search_config): + for ans in ask(req["question"], req["kb_ids"], uid, search_config=search_config): yield "data:" + json.dumps({"code": 0, "message": "", "data": ans}, ensure_ascii=False) + "\n\n" except Exception as e: yield "data:" + json.dumps({"code": 500, "message": str(e), "data": {"answer": "**ERROR**: " + str(e), "reference": []}}, ensure_ascii=False) + "\n\n"