From 59b6c56262c37f12363356d49aef29af265eb189 Mon Sep 17 00:00:00 2001 From: ManishMadan2882 Date: Sat, 21 Sep 2024 01:50:02 +0530 Subject: [PATCH] feat(stream): yield sources in stream --- application/api/answer/routes.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/application/api/answer/routes.py b/application/api/answer/routes.py index 62fb6b93..53005cae 100644 --- 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: