updated entities param

neo4j-labs · Sep 12, 2024 · ccfabc7 · ccfabc7
1 parent 21dd283
commit ccfabc7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/backend/src/QA_integration.py b/backend/src/QA_integration.py
@@ -155,7 +155,7 @@ def format_documents(documents, model):
 
     formatted_docs = []
     sources = set()
-    lc_entities = {}
+    lc_entities = {'entities':list()}
 
     for doc in sorted_documents:
         try:
@@ -357,12 +357,12 @@ def process_chat_response(messages,history, question, model, graph, document_nam
     try:
         llm, doc_retriever, model_version = setup_chat(model, graph, document_names,chat_mode_settings)
 
-        docs = retrieve_documents(doc_retriever, messages)        
+        docs = retrieve_documents(doc_retriever, messages)   
         if docs:
             content, result, total_tokens = process_documents(docs, question, messages, llm, model,chat_mode_settings)
         else:
             content = "I couldn't find any relevant documents to answer your question."
-            result = {"sources": [], "chunkdetails": []}
+            result = {"sources": [], "chunkdetails": [],"entities":[]}
             total_tokens = 0
 
         ai_response = AIMessage(content=content)

diff --git a/backend/src/shared/constants.py b/backend/src/shared/constants.py
@@ -271,7 +271,6 @@
 }} as outside
 """
 
-
 LOCAL_COMMUNITY_SEARCH_QUERY_SUFFIX = """
 RETURN {chunks: [c in chunks | c.text], 
         communities: [c in communities | c.summary],