From 219011b62ae8fcc31f59dd99862be4690ce1af7c Mon Sep 17 00:00:00 2001 From: John Wang Date: Tue, 23 May 2023 12:57:26 +0800 Subject: [PATCH] fix: disable template string in query (#160) --- api/core/completion.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/api/core/completion.py b/api/core/completion.py index afa40b45cd..6b4250c4c5 100644 --- a/api/core/completion.py +++ b/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(