feat(stream): yield sources in stream

2025-11-29 16:43:16 +00:00 · 2024-09-21 01:50:02 +05:30
parent bf7cce52db
commit 59b6c56262
1 changed files with 7 additions and 1 deletions
--- a/application/api/answer/routes.py
+++ b/application/api/answer/routes.py
@@ -221,7 +221,13 @@ def complete_stream(
                "timestamp": datetime.datetime.now(datetime.timezone.utc),
            }
        )
-
+        sources = retriever.search()
+        for source in sources:
+            if("text" in source):
+                source["text"] = source["text"][:100].strip()+"..."
+        if(len(sources) > 0):
+            data = json.dumps({"type":"source","source":sources})
+            yield f"data: {data}\n\n"
        data = json.dumps({"type": "end"})
        yield f"data: {data}\n\n"
    except Exception as e: