Added initial Griptape support.

Author: kirk-marple

README.md

@@ -4,7 +4,7 @@
 
 ## Overview
 
-The Graphlit Agent Tools for Python enables easy interaction with agent frameworks such as [CrewAI](https://crewai.com), allowing developers to easily integrate the Graphlit service with agentic workflows. This document outlines the setup process and provides a basic example of using the tools.
+The Graphlit Agent Tools for Python enables easy interaction with agent frameworks such as [CrewAI](https://crewai.com) or [Griptape](https://www.griptape.ai/), allowing developers to easily integrate the Graphlit service with agentic workflows. This document outlines the setup process and provides a basic example of using the tools.
 
 ## Prerequisites
 
@@ -21,13 +21,25 @@ To install the Graphlit Agent Tools with CrewAI, use pip:
 pip install graphlit-tools[crewai]
 ```
 
+To install the Graphlit Agent Tools with Griptape, use pip:
+
+```bash
+pip install graphlit-tools[griptape]
+```
+
 ### Using the Graphlit agent tools
 
 We have example Google Colab notebooks using CrewAI, which provide an example for [analyzing the web marketing strategy of a company](https://colab.research.google.com/github/graphlit/graphlit-samples/blob/main/python/Notebook%20Examples/Graphlit_2024_12_07_CrewAI_Web_Marketing_Analyzer.ipynb), and for [structured data extraction of products from scraped web pages](https://colab.research.google.com/github/graphlit/graphlit-samples/blob/main/python/Notebook%20Examples/Graphlit_2024_12_08_CrewAI_Product_Data_Extraction.ipynb).
 
 Once you have configured the Graphlit client, as shown below, you will pass the client to the tool constructor.
 
-For use in CrewAI, you will need to convert the tool to the CrewAI tool schema with the `CrewAIConverter.from_tool()` function.  We will provide support for additional agent frameworks, such as LangGraph and AutoGen in future.
+For use in CrewAI, you will need to convert the tool to the CrewAI tool schema with the `CrewAIConverter.from_tool()` function.  
+
+For use in Griptape, you will need to convert the tool to the CrewAI tool schema with the `GriptapeConverter.from_tool()` function.
+
+We will provide support for additional agent frameworks, such as LangGraph and AutoGen in future.
+
+#### CrewAI
 
 ```python
 from graphlit_tools import WebSearchTool, CrewAIConverter
@@ -44,6 +56,23 @@ web_search_agent = Agent(
 )
 ```
 
+#### Griptape
+
+```python
+from graphlit_tools import WebSearchTool, CrewAIConverter
+
+web_search_tool = GriptapeConverter.from_tool(WebSearchTool(graphlit))
+
+web_search_agent = Agent(
+    role="Web Researcher",
+    goal="Find the {company} website.",
+    backstory="",
+    verbose=True,
+    allow_delegation=False,
+    tools=[web_search_tool],
+)
+```
+
 ## Configuration
 
 The Graphlit Client supports environment variables to be set for authentication and configuration:

__init__.py

@@ -33,5 +33,6 @@
     MicrosoftTeamsIngestTool,
     DiscordIngestTool,
     SlackIngestTool,
-    CrewAIConverter
+    CrewAIConverter,
+    GriptapeConverter,
 )

graphlit_tools/__init__.py

@@ -1,5 +1,6 @@
 from .base_tool import BaseTool
 from .crewai_converter import CrewAIConverter
+from .griptape_converter import GriptapeConverter
 from .exceptions import ToolException
 from .retrieval.content_retrieval_tool import ContentRetrievalTool
 from .retrieval.person_retrieval_tool import PersonRetrievalTool

graphlit_tools/griptape_converter.py

@@ -0,0 +1,78 @@
+from typing import Any, cast
+from schema import Schema
+from .base_tool import BaseTool
+
+GriptapeBaseTool: Any = None
+
+try:
+    from griptape.tools import BaseTool as GriptapeBaseTool
+    from griptape.utils.decorators import activity
+    from griptape.artifacts import TextArtifact    
+except ImportError:
+    GriptapeBaseTool = None
+
+if GriptapeBaseTool:
+    class GriptapeConverter(GriptapeBaseTool):
+        """Tool to convert Graphlit tools into Griptape tools."""
+
+        graphlit_tool: Any
+
+        def _run(
+            self,
+            *args: Any,
+            **kwargs: Any,
+        ) -> Any:
+            tool = cast(BaseTool, self.graphlit_tool)
+
+            return tool.run(*args, **kwargs)
+
+        async def _arun(
+            self,
+            *args: Any,
+            **kwargs: Any,
+        ) -> Any:
+            tool = cast(BaseTool, self.graphlit_tool)
+
+            return await tool.arun(*args, **kwargs)
+
+        @classmethod
+        def from_tool(cls, tool: Any, **kwargs: Any) -> "GriptapeConverter":
+            if not isinstance(tool, BaseTool):
+                raise ValueError(f"Expected a Graphlit tool, got {type(tool)}")
+
+            tool = cast(BaseTool, tool)
+
+            if tool.args_schema is None:
+                raise ValueError("Invalid arguments JSON schema.")
+
+            def generate(self, params: dict[str, Any]) -> TextArtifact:
+                return TextArtifact(str(self.graphlit_tool.run(**params)))
+
+            tool_schema = Schema(tool.json_schema)
+
+            decorated_generate = activity(
+                config={
+                    "description": tool.description,
+                    "schema": tool_schema,
+                }
+            )(generate)
+
+            new_cls = type(
+                f"{tool.name.capitalize()}GriptapeConverter",
+                (cls,),
+                {"graphlit_tool": tool, "generate": decorated_generate},
+            )
+
+            return new_cls(
+                name=tool.name,
+                graphlit_tool=tool,
+                **kwargs,
+            )
+else:
+    class GriptapeConverter:
+        """Fallback GriptapeConverter if griptape is not installed."""
+        def __init__(self, *args, **kwargs):
+            raise ImportError(
+                "GriptapeConverter requires the griptape package. "
+                "Install it using pip install graphlit-tools[griptape]."
+            )

setup.py

@@ -15,7 +15,8 @@
         'graphlit-client'        
     ],
     extras_require={
-        "crewai": ["crewai-tools"]  # Extras for CrewAI support
+        "crewai": ["crewai-tools"],  # Extras for CrewAI support
+        "griptape": ["griptape"]  # Extras for Griptape support
     },
     python_requires='>=3.6',
     author='Unstruk Data Inc.',