hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-09 19:06:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore: apply flake8-comprehensions Ruff rules to improve collection comprehensions (#5652)

Browse Source 
Co-authored-by: -LAN- <laipz8200@outlook.com>
This commit is contained in:
Bowen Liang
2024-06-27 11:21:31 +08:00
committed by GitHub
parent 2e718b85e9
commit dcb72e0067
58 changed files with 123 additions and 136 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/core/prompt/simple_prompt_transform.py
View File

@@ -148,7 +148,7 @@ class SimplePromptTransform(PromptTransform):
special_variable_keys.append('#histories#')
if query_in_prompt:
prompt += prompt_rules['query_prompt'] if 'query_prompt' in prompt_rules else '{{#query#}}'
prompt += prompt_rules.get('query_prompt', '{{#query#}}')
special_variable_keys.append('#query#')
return {
@@ -234,8 +234,8 @@ class SimplePromptTransform(PromptTransform):
)
),
max_token_limit=rest_tokens,
human_prefix=prompt_rules['human_prefix'] if 'human_prefix' in prompt_rules else 'Human',
ai_prefix=prompt_rules['assistant_prefix'] if 'assistant_prefix' in prompt_rules else 'Assistant'
human_prefix=prompt_rules.get('human_prefix', 'Human'),
ai_prefix=prompt_rules.get('assistant_prefix', 'Assistant')
)
# get prompt