Browse Source

fix: disable template string in query (#160)

tags/0.2.2
John Wang 2 years ago
parent
commit
219011b62a
No account linked to committer's email address
1 changed files with 7 additions and 0 deletions
  1. 7
    0
      api/core/completion.py

+ 7
- 0
api/core/completion.py View File

chain_output: Optional[str], chain_output: Optional[str],
memory: Optional[ReadOnlyConversationTokenDBBufferSharedMemory]) -> \ memory: Optional[ReadOnlyConversationTokenDBBufferSharedMemory]) -> \
Union[str | List[BaseMessage]]: 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 pre_prompt = PromptBuilder.process_template(pre_prompt) if pre_prompt else pre_prompt
if mode == 'completion': if mode == 'completion':
prompt_template = OutLinePromptTemplate.from_template( prompt_template = OutLinePromptTemplate.from_template(

Loading…
Cancel
Save