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

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

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>
"""