瀏覽代碼

fix: disable template string in query (#160)

John Wang 1 年之前
父節點
當前提交
219011b62a
共有 1 個文件被更改,包括 7 次插入0 次删除
  1. 7 0
      api/core/completion.py

+ 7 - 0
api/core/completion.py

@@ -124,6 +124,13 @@ class Completion:
                             chain_output: Optional[str],
                             memory: Optional[ReadOnlyConversationTokenDBBufferSharedMemory]) -> \
             Union[str | List[BaseMessage]]:
+        # disable template string in query
+        query_params = OutLinePromptTemplate.from_template(template=query).input_variables
+        if query_params:
+            for query_param in query_params:
+                if query_param not in inputs:
+                    inputs[query_param] = '{' + query_param + '}'
+
         pre_prompt = PromptBuilder.process_template(pre_prompt) if pre_prompt else pre_prompt
         if mode == 'completion':
             prompt_template = OutLinePromptTemplate.from_template(