Fix: Remove core/tools from pyrightconfig.json and fix type errors (#26413)

Co-authored-by: google-labs-jules[bot] <161369871+google-labs-jules[bot]@users.noreply.github.com>
2025-09-29 20:22:38 +09:00
parent b6cea71023
commit b2bcb6d21a
4 changed files with 43 additions and 36 deletions
--- a/api/core/tools/utils/dataset_retriever/dataset_retriever_base_tool.py
+++ b/api/core/tools/utils/dataset_retriever/dataset_retriever_base_tool.py
@@ -18,6 +18,10 @@ class DatasetRetrieverBaseTool(BaseModel, ABC):
    retriever_from: str
    model_config = ConfigDict(arbitrary_types_allowed=True)

+    def run(self, query: str) -> str:
+        """Use the tool."""
+        return self._run(query)
+
    @abstractmethod
    def _run(self, query: str) -> str:
        """Use the tool.