hotfix source doc data

arc53 · Dec 22, 2023 · 015bc7c · 015bc7c · vercel · Dec 22, 2023
1 parent 3da2a00
commit 015bc7c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/application/api/answer/routes.py b/application/api/answer/routes.py
@@ -117,10 +117,10 @@ def complete_stream(question, docsearch, chat_history, api_key, prompt_id, conve
     source_log_docs = []
     for doc in docs:
         if doc.metadata:
-            data = json.dumps({"type": "source", "doc": doc.page_content, "metadata": doc.metadata})
+            data = json.dumps({"type": "source", "doc": doc.page_content[:10], "metadata": doc.metadata})
             source_log_docs.append({"title": doc.metadata['title'].split('/')[-1], "text": doc.page_content})
         else:
-            data = json.dumps({"type": "source", "doc": doc.page_content})
+            data = json.dumps({"type": "source", "doc": doc.page_content[:10]})
             source_log_docs.append({"title": doc.page_content, "text": doc.page_content})
         yield f"data:{data}\n\n"