feat: change dataset / wikipedia tool name

api/core/orchestrator_rule_parser.py

@@ -231,6 +231,7 @@ class OrchestratorRuleParser:
 
     def to_wikipedia_tool(self) -> Optional[BaseTool]:
         return WikipediaQueryRun(
+            name="wikipedia",
             api_wrapper=WikipediaAPIWrapper(doc_content_chars_max=4000),
             callbacks=[DifyStdOutCallbackHandler()]
         )

api/core/tool/dataset_retriever_tool.py

@@ -22,7 +22,7 @@ class DatasetRetrieverToolInput(BaseModel):
 
 class DatasetRetrieverTool(BaseTool):
     """Tool for querying a Dataset."""
-    name: str = "dataset_retriever"
+    name: str = "dataset"
     args_schema: Type[BaseModel] = DatasetRetrieverToolInput
     description: str = "use this to retrieve a dataset. "