Update astradb.py

langflow-ai · Jan 28, 2025 · 731f82b · 731f82b
1 parent 4ade9f3
commit 731f82b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/backend/base/langflow/components/tools/astradb.py b/src/backend/base/langflow/components/tools/astradb.py
@@ -111,7 +111,7 @@ def create_args_schema(self) -> dict[str, BaseModel]:
         model = create_model("ToolInput", **args, __base__=BaseModel)
         return {"ToolInput": model}
 
-    def build_tool(self) -> StructuredTool:
+    def build_tool(self) -> Tool:
         """Builds an Astra DB Collection tool.
 
         Returns: