This commit is contained in:
Timothy J. Baek committed 2024-10-14 18:47:41 -07:00
1 parent 1db1ef7c26
commit 8eb45acf10
2 files changed
+6 -6

No files matched your search

+4 -4
View File
@@ -403,9 +403,7 @@ def get_rag_context(
)
contexts.append(
(", ".join(file_names) + ":\n\n")
if file_names
else ""
((", ".join(file_names) + ":\n\n") if file_names else "")
+ "\n\n".join(
[text for text in context["documents"][0] if text is not None]
)
@@ -422,7 +420,9 @@ def get_rag_context(
except Exception as e:
log.exception(e)
print(contexts, citations)
print("contexts", contexts)
print("citations", citations)
return contexts, citations
+2 -2
View File
@@ -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.
<context>
[context]
{{CONTEXT}}
</context>
<rules>
@@ -1055,7 +1055,7 @@ DEFAULT_RAG_TEMPLATE = """You are given a user query, some textual context and r
</rules>
<user_query>
[query]
{{QUERY}}
</user_query>
"""