Bye bye tool driver

griptape-ai · Nov 21, 2024 · 251807a · 251807a
1 parent ded439b
commit 251807a
Show file tree

Hide file tree

Showing 12 changed files with 0 additions and 630 deletions.
diff --git a/docs/griptape-framework/drivers/src/griptape_cloud_tool_driver_1.py b/docs/griptape-framework/drivers/src/griptape_cloud_tool_driver_1.py
diff --git a/docs/griptape-framework/drivers/src/griptape_cloud_tool_driver_2.py b/docs/griptape-framework/drivers/src/griptape_cloud_tool_driver_2.py
diff --git a/docs/griptape-framework/drivers/src/tool_driver_1.py b/docs/griptape-framework/drivers/src/tool_driver_1.py
diff --git a/griptape/drivers/__init__.py b/griptape/drivers/__init__.py
@@ -133,8 +133,6 @@
 from .assistant.base_assistant_driver import BaseAssistantDriver
 from .assistant.griptape_cloud_assistant_driver import GriptapeCloudAssistantDriver
 from .assistant.openai_assistant_driver import OpenAiAssistantDriver
-from .tool.base_tool_driver import BaseToolDriver
-from .tool.griptape_cloud_tool_driver import GriptapeCloudToolDriver
 
 __all__ = [
     "BasePromptDriver",
@@ -242,6 +240,4 @@
     "BaseAssistantDriver",
     "GriptapeCloudAssistantDriver",
     "OpenAiAssistantDriver",
-    "BaseToolDriver",
-    "GriptapeCloudToolDriver",
 ]
diff --git a/griptape/drivers/tool/__init__.py b/griptape/drivers/tool/__init__.py
diff --git a/griptape/drivers/tool/base_tool_driver.py b/griptape/drivers/tool/base_tool_driver.py
diff --git a/griptape/drivers/tool/griptape_cloud_tool_driver.py b/griptape/drivers/tool/griptape_cloud_tool_driver.py
diff --git a/griptape/schemas/base_schema.py b/griptape/schemas/base_schema.py
@@ -168,7 +168,6 @@ def _resolve_types(cls, attrs_cls: type) -> None:
             BasePromptDriver,
             BaseRulesetDriver,
             BaseTextToSpeechDriver,
-            BaseToolDriver,
             BaseVectorStoreDriver,
         )
         from griptape.events import EventListener
@@ -193,7 +192,6 @@ def _resolve_types(cls, attrs_cls: type) -> None:
                 "BaseConversationMemoryDriver": BaseConversationMemoryDriver,
                 "BaseRulesetDriver": BaseRulesetDriver,
                 "BaseImageGenerationDriver": BaseImageGenerationDriver,
-                "BaseToolDriver": BaseToolDriver,
                 "BaseArtifact": BaseArtifact,
                 "PromptStack": PromptStack,
                 "EventListener": EventListener,

diff --git a/griptape/tools/base_tool.py b/griptape/tools/base_tool.py
@@ -23,7 +23,6 @@
 
 if TYPE_CHECKING:
     from griptape.common import ToolAction
-    from griptape.drivers import BaseToolDriver
     from griptape.memory import TaskMemory
     from griptape.tasks import ActionsSubtask
 
@@ -57,7 +56,6 @@ class BaseTool(ActivityMixin, SerializableMixin, RunnableMixin["BaseTool"], ABC)
     dependencies_install_directory: Optional[str] = field(default=None, kw_only=True, metadata={"serializable": True})
     verbose: bool = field(default=False, kw_only=True, metadata={"serializable": True})
     off_prompt: bool = field(default=False, kw_only=True, metadata={"serializable": True})
-    tool_driver: BaseToolDriver = field(default=None, kw_only=True)
 
     def __attrs_post_init__(self) -> None:
         if (
@@ -67,9 +65,6 @@ def __attrs_post_init__(self) -> None:
         ):
             self.install_dependencies(os.environ.copy())
 
-        if self.tool_driver is not None:
-            self.tool_driver.initialize_tool(self)
-
     @output_memory.validator  # pyright: ignore[reportAttributeAccessIssue]
     def validate_output_memory(self, _: Attribute, output_memory: dict[str, Optional[list[TaskMemory]]]) -> None:
         if output_memory:

diff --git a/mkdocs.yml b/mkdocs.yml
@@ -131,7 +131,6 @@ nav:
           - Observability Drivers: "griptape-framework/drivers/observability-drivers.md"
           - Ruleset Drivers: "griptape-framework/drivers/ruleset-drivers.md"
           - File Manager Drivers: "griptape-framework/drivers/file-manager-drivers.md"
-          - Tool Drivers: "griptape-framework/drivers/tool-drivers.md"
       - Data:
           - Overview: "griptape-framework/data/index.md"
           - Artifacts: "griptape-framework/data/artifacts.md"

diff --git a/tests/unit/drivers/tool/__init__.py b/tests/unit/drivers/tool/__init__.py