Timothy J. Baek 6 months ago
parent
commit
8eb45acf10
2 changed files with 6 additions and 6 deletions
  1. 4 4
      backend/open_webui/apps/retrieval/utils.py
  2. 2 2
      backend/open_webui/config.py

+ 4 - 4
backend/open_webui/apps/retrieval/utils.py

@@ -403,9 +403,7 @@ def get_rag_context(
                 )
                 )
 
 
                 contexts.append(
                 contexts.append(
-                    (", ".join(file_names) + ":\n\n")
-                    if file_names
-                    else ""
+                    ((", ".join(file_names) + ":\n\n") if file_names else "")
                     + "\n\n".join(
                     + "\n\n".join(
                         [text for text in context["documents"][0] if text is not None]
                         [text for text in context["documents"][0] if text is not None]
                     )
                     )
@@ -422,7 +420,9 @@ def get_rag_context(
         except Exception as e:
         except Exception as e:
             log.exception(e)
             log.exception(e)
 
 
-    print(contexts, citations)
+    print("contexts", contexts)
+    print("citations", citations)
+
     return contexts, citations
     return contexts, citations
 
 
 
 

+ 2 - 2
backend/open_webui/config.py

@@ -1042,7 +1042,7 @@ CHUNK_OVERLAP = PersistentConfig(
 DEFAULT_RAG_TEMPLATE = """You are given a user query, some textual context and rules, all inside xml tags. You have to answer the query based on the context while respecting the rules.
 DEFAULT_RAG_TEMPLATE = """You are given a user query, some textual context and rules, all inside xml tags. You have to answer the query based on the context while respecting the rules.
 
 
 <context>
 <context>
-[context]
+{{CONTEXT}}
 </context>
 </context>
 
 
 <rules>
 <rules>
@@ -1055,7 +1055,7 @@ DEFAULT_RAG_TEMPLATE = """You are given a user query, some textual context and r
 </rules>
 </rules>
 
 
 <user_query>
 <user_query>
-[query]
+{{QUERY}}
 </user_query>
 </user_query>
 """
 """