jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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
commit dcb72e0067
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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])