Changes to make graphrag ui work

Signed-off-by: theresa <[email protected]>
opea-project · Nov 13, 2024 · b140381 · b140381
1 parent 60a53c7
commit b140381
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/comps/nginx/nginx.conf.template b/comps/nginx/nginx.conf.template
@@ -43,6 +43,10 @@ server {
         proxy_set_header X-Real-IP $remote_addr;
         proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
         proxy_set_header X-Forwarded-Proto $scheme;
+        proxy_connect_timeout 6000;
+        proxy_send_timeout 6000;
+        proxy_read_timeout 6000;
+        send_timeout 6000;
     }
 
     location /v1/dataprep/get_file {

diff --git a/comps/retrievers/neo4j/llama_index/retriever_community_answers_neo4j.py b/comps/retrievers/neo4j/llama_index/retriever_community_answers_neo4j.py
@@ -192,7 +192,11 @@ async def retrieve(input: Union[ChatCompletionRequest]) -> Union[ChatCompletionR
     if logflag:
         logger.info(input)
     start = time.time()
-    query = input.messages[0]["content"]
+
+    if isinstance(input.messages, str):
+        query = input.messages
+    else:
+        query = input.messages[0]["content"]
     logger.info(f"Query received in retriever: {query}")
 
     if OPENAI_API_KEY: