fix: backslash rag content issue

072b499a · Timothy J. Baek · cd92cf0d · 072b499a
Commit 072b499a authored Mar 15, 2024 by Timothy J. Baek
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 3 deletions

backend/apps/rag/utils.py backend/apps/rag/utils.py +2 -3

No files found.
--- a/backend/apps/rag/utils.py
+++ b/backend/apps/rag/utils.py
@@ -91,9 +91,8 @@ def query_collection(


 def rag_template(template: str, context: str, query: str):
-    template = re.sub(r"\[context\]", context, template)
-    template = re.sub(r"\[query\]", query, template)
-
+    template = template.replace("[context]", context)
+    template = template.replace("[query]", query)
    return template