move the crewAI tools conversation back in the example

Author: Shashikant86

examples/crewai_semantic_search.py

@@ -19,17 +19,68 @@
 """
 
 import os
+from typing import Any
 
 from crewai import Agent, Crew, Task
+from crewai.tools.base_tool import BaseTool as CrewAIBaseTool
 from dotenv import load_dotenv
+from pydantic import BaseModel, Field
 
 from stackone_ai import StackOneToolSet
+from stackone_ai.models import StackOneTool
 
 load_dotenv()
 
 _account_ids = [aid.strip() for aid in os.getenv("STACKONE_ACCOUNT_ID", "").split(",") if aid.strip()]
 
 
+def _to_crewai_tool(tool: StackOneTool) -> CrewAIBaseTool:
+    """Wrap a StackOneTool as a CrewAI BaseTool.
+
+    CrewAI has its own BaseTool (not LangChain's), so we create a
+    lightweight wrapper that delegates execution to the StackOne tool.
+    """
+    schema_props: dict[str, Any] = {}
+    annotations: dict[str, Any] = {}
+
+    for name, details in tool.parameters.properties.items():
+        python_type: type = str
+        if isinstance(details, dict):
+            type_str = details.get("type", "string")
+            if type_str == "number":
+                python_type = float
+            elif type_str == "integer":
+                python_type = int
+            elif type_str == "boolean":
+                python_type = bool
+            field = Field(description=details.get("description", ""))
+        else:
+            field = Field(description="")
+
+        schema_props[name] = field
+        annotations[name] = python_type
+
+    _schema = type(
+        f"{tool.name.title().replace('_', '')}Args",
+        (BaseModel,),
+        {"__annotations__": annotations, "__module__": __name__, **schema_props},
+    )
+
+    _parent = tool
+    _name = tool.name
+    _description = tool.description
+
+    class WrappedTool(CrewAIBaseTool):
+        name: str = _name
+        description: str = _description
+        args_schema: type[BaseModel] = _schema
+
+        def _run(self, **kwargs: Any) -> Any:
+            return _parent.execute(kwargs)
+
+    return WrappedTool()
+
+
 def crewai_semantic_search() -> None:
     toolset = StackOneToolSet()
 
@@ -38,7 +89,7 @@ def crewai_semantic_search() -> None:
     #   tool definitions. Useful for inspecting what's available before committing.
     preview = toolset.search_action_names(
         "book a meeting or check availability",
-        account_ids=_account_ids
+        account_ids=_account_ids,
     )
     print("Semantic search preview (action names only):")
     for r in preview:
@@ -51,7 +102,7 @@ def crewai_semantic_search() -> None:
     tools = toolset.search_tools(
         "schedule meetings, check availability, list events",
         connector="calendly",
-        account_ids=_account_ids
+        account_ids=_account_ids,
     )
     assert len(tools) > 0, "Expected at least one scheduling tool"
 
@@ -61,7 +112,7 @@ def crewai_semantic_search() -> None:
     print()
 
     # Step 3: Convert to CrewAI format
-    crewai_tools = tools.to_crewai()
+    crewai_tools = [_to_crewai_tool(t) for t in tools]
 
     # Step 4: Create a CrewAI meeting booking agent
     agent = Agent(

stackone_ai/models.py

@@ -450,64 +450,6 @@ def _run(self, **kwargs: Any) -> Any:
 
         return StackOneLangChainTool()
 
-    def to_crewai(self) -> Any:
-        """Convert this tool to CrewAI format
-
-        Requires the ``crewai`` package (``pip install crewai``).
-
-        Returns:
-            Tool as a ``crewai.tools.BaseTool`` instance
-        """
-        try:
-            from crewai.tools.base_tool import BaseTool as CrewAIBaseTool
-        except ImportError as e:
-            raise ImportError("crewai is required for to_crewai(). Install with: pip install crewai") from e
-
-        schema_props: dict[str, Any] = {}
-        annotations: dict[str, Any] = {}
-
-        for name, details in self.parameters.properties.items():
-            python_type: type = str
-            if isinstance(details, dict):
-                type_str = details.get("type", "string")
-                if type_str == "number":
-                    python_type = float
-                elif type_str == "integer":
-                    python_type = int
-                elif type_str == "boolean":
-                    python_type = bool
-
-                field = Field(description=details.get("description", ""))
-            else:
-                field = Field(description="")
-
-            schema_props[name] = field
-            annotations[name] = python_type
-
-        schema_class = type(
-            f"{self.name.title()}Args",
-            (BaseModel,),
-            {
-                "__annotations__": annotations,
-                "__module__": __name__,
-                **schema_props,
-            },
-        )
-
-        parent_tool = self
-        _name = parent_tool.name
-        _description = parent_tool.description
-
-        class StackOneCrewAITool(CrewAIBaseTool):
-            name: str = _name
-            description: str = _description
-            args_schema: type[BaseModel] = schema_class
-
-            def _run(self, **kwargs: Any) -> Any:
-                return parent_tool.execute(kwargs)
-
-        return StackOneCrewAITool()
-
     def set_account_id(self, account_id: str | None) -> None:
         """Set the account ID for this tool
 
@@ -616,15 +558,6 @@ def to_langchain(self) -> Sequence[BaseTool]:
         """
         return [tool.to_langchain() for tool in self.tools]
 
-    def to_crewai(self) -> list[Any]:
-        """Convert all tools to CrewAI format
-
-        Requires the ``crewai`` package (``pip install crewai``).
-
-        Returns:
-            List of tools as ``crewai.tools.BaseTool`` instances
-        """
-        return [tool.to_crewai() for tool in self.tools]
 
     def utility_tools(
         self,