strands support in python toolkit

python/examples/strands/.env.template

@@ -0,0 +1,2 @@
+OPENAI_API_KEY=""
+STRIPE_SECRET_KEY=""

python/examples/strands/main.py

@@ -0,0 +1,15 @@
+from strands import Agent
+from strands_tools import calculator, file_read, shell
+
+# Add tools to our agent
+agent = Agent(
+    tools=[calculator, file_read, shell]
+)
+
+# Agent will automatically determine when to use the calculator tool
+agent("What is 42 ^ 9")
+
+print("\n\n")  # Print new lines
+
+# Agent will use the shell and file reader tool when appropriate
+agent("Show me the contents of a single file in this directory")

python/requirements.txt

@@ -12,3 +12,5 @@ ruff==0.4.4
 stripe==11.0.0
 openai==1.66.0
 openai-agents==0.0.2
+strands-agent>=0.1.0
+strands-agents-tools

python/stripe_agent_toolkit/strands/tool.py

@@ -0,0 +1,18 @@
+from strands.tools.tools import PythonAgentTool as StrandTool
+import json
+
+def StripeTool(api, tool) -> StrandTool:
+    parameters = tool["args_schema"].model_json_schema()
+    parameters["additionalProperties"] = False
+    parameters["type"] = "object"
+    return StrandTool(
+        tool_name=tool["method"],
+        tool_spec={
+            "name": tool["method"],
+            "description": tool["description"],
+            "inputSchema": {
+                "json": parameters
+            }
+        },
+        callback=lambda x: api.run(tool["method"], **json.loads(x))
+    )

python/stripe_agent_toolkit/strands/toolkit.py

@@ -0,0 +1,43 @@
+"""Stripe Agent Toolkit."""
+
+from typing import List, Optional
+from pydantic import PrivateAttr
+import json
+
+from strands.tools.tools import PythonAgentTool as StrandTool
+
+
+from ..api import StripeAPI
+from ..tools import tools
+from ..configuration import Configuration, is_tool_allowed
+from .tool import StripeTool
+from .hooks import BillingHooks
+
+class StripeAgentToolkit:
+    _tools: List[FunctionTool] = PrivateAttr(default=[])
+    _stripe_api: StripeAPI = PrivateAttr(default=None)
+
+    def __init__(
+        self, secret_key: str, configuration: Optional[Configuration] = None
+    ):
+        super().__init__()
+
+        context = configuration.get("context") if configuration else None
+
+        self._stripe_api = StripeAPI(secret_key=secret_key, context=context)
+
+        filtered_tools = [
+            tool for tool in tools if is_tool_allowed(tool, configuration)
+        ]
+
+        self._tools = [
+            StripeTool(self._stripe_api, tool)
+            for tool in filtered_tools
+        ]
+
+    def get_tools(self) -> List[FunctionTool]:
+        """Get the tools in the toolkit."""
+        return self._tools
+
+    def billing_hook(self, type: Optional[str] = None, customer: Optional[str] = None, meter: Optional[str] = None, meters: Optional[dict[str, str]] = None) -> BillingHooks:
+        return BillingHooks(self._stripe_api, type, customer, meter, meters)