浏览代码

fix: backslash rag content issue

Timothy J. Baek 1 年之前
父节点
当前提交
072b499a50
共有 1 个文件被更改,包括 2 次插入3 次删除
  1. 2 3
      backend/apps/rag/utils.py

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

@@ -91,9 +91,8 @@ def query_collection(
 
 
 
 
 def rag_template(template: str, context: str, query: str):
 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
     return template