youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-04 01:18:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor app

Browse Source
This commit is contained in:
takatost
2024-02-29 17:33:52 +08:00
parent 896c200211
commit 799db69e4f
94 changed files with 991 additions and 721 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/tools/tool/dataset_retriever_tool.py
View File

@ -4,7 +4,7 @@ from langchain.tools import BaseTool
from core.callback_handler.index_tool_callback_handler import DatasetIndexToolCallbackHandler
from core.entities.application_entities import DatasetRetrieveConfigEntity, InvokeFrom
from core.features.dataset_retrieval.dataset_retrieval import DatasetRetrievalFeature
from core.rag.retrieval.dataset_retrieval import DatasetRetrieval
from core.tools.entities.common_entities import I18nObject
from core.tools.entities.tool_entities import ToolDescription, ToolIdentity, ToolInvokeMessage, ToolParameter
from core.tools.tool.tool import Tool
@ -30,7 +30,7 @@ class DatasetRetrieverTool(Tool):
if retrieve_config is None:
return []
feature = DatasetRetrievalFeature()
feature = DatasetRetrieval()
# save original retrieve strategy, and set retrieve strategy to SINGLE
# Agent only support SINGLE mode