hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-11 03:46:52 +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

2
api/core/rag/datasource/keyword/jieba/jieba.py
View File

@@ -197,7 +197,7 @@ class Jieba(BaseKeyword):
chunk_indices_count[node_id] += 1
sorted_chunk_indices = sorted(
list(chunk_indices_count.keys()),
chunk_indices_count.keys(),
key=lambda x: chunk_indices_count[x],
reverse=True,
)

2
api/core/rag/retrieval/router/multi_dataset_react_route.py
View File

@@ -201,7 +201,7 @@ class ReactMultiDatasetRouter:
tool_strings.append(
f"{tool.name}: {tool.description}, args: {{'query': {{'title': 'Query', 'description': 'Query for the dataset to be used to retrieve the dataset.', 'type': 'string'}}}}")
formatted_tools = "\n".join(tool_strings)
unique_tool_names = set(tool.name for tool in tools)
unique_tool_names = {tool.name for tool in tools}
tool_names = ", ".join('"' + name + '"' for name in unique_tool_names)
format_instructions = format_instructions.format(tool_names=tool_names)
template = "\n\n".join([prefix, formatted_tools, format_instructions, suffix])