provision-stack-crewai 0.1.0__py3-none-any.whl

provision_stack_crewai/__init__.py

@@ -0,0 +1,18 @@
+"""
+Provision Stack tools for CrewAI.
+
+Usage::
+
+    from provision_stack_crewai import get_provision_stack_tools
+
+    tools = get_provision_stack_tools(api_key="your-token")
+    agent = Agent(
+        role="Infrastructure Engineer",
+        tools=tools,
+        ...
+    )
+"""
+
+from provision_stack_crewai.tools import get_provision_stack_tools
+
+__all__ = ["get_provision_stack_tools"]

provision_stack_crewai/tools.py

@@ -0,0 +1,252 @@
+"""
+CrewAI tool wrappers for Provision Stack.
+
+Each tool is a CrewAI ``BaseTool`` subclass that wraps the
+corresponding Provision Stack API endpoint.
+"""
+
+from __future__ import annotations
+
+from typing import Any, Type
+
+from crewai.tools import BaseTool
+from pydantic import BaseModel, Field
+
+from provision_stack.client import ProvisionStackClient
+from provision_stack.config import PRODUCT_NAME
+
+
+# ── Pydantic schemas for tool inputs ──────────────────────────────
+
+
+class SuggestInput(BaseModel):
+    """Input for getting deployment suggestions."""
+
+    outcome: str = Field(
+        ...,
+        description="Natural language description of desired infrastructure "
+        "(e.g. 'deploy a production web application').",
+    )
+    region: str | None = Field(
+        None,
+        description="Cloud region (e.g. 'us-east-1', 'europe-west1').",
+    )
+    provider: str | None = Field(
+        None,
+        description="Restrict to one provider: 'aws', 'gcp', 'azure', 'oracle'.",
+    )
+    max_monthly_cost_usd: float | None = Field(
+        None,
+        description="Monthly cost cap in USD.",
+    )
+
+
+class DeployInput(BaseModel):
+    """Input for deploying infrastructure."""
+
+    outcome: str = Field(
+        ...,
+        description="Natural language description of what to deploy.",
+    )
+    tier: str | None = Field(
+        None,
+        description="Deployment tier: 'Starter', 'MVP', 'Startup', 'Enterprise'.",
+    )
+    region: str | None = Field(
+        None,
+        description="Target cloud region.",
+    )
+    max_monthly_cost_usd: float | None = Field(
+        None,
+        description="Monthly cost cap in USD.",
+    )
+
+
+class StatusInput(BaseModel):
+    """Input for checking deployment status."""
+
+    deployment_id: str = Field(
+        ...,
+        description="The deployment ID to check.",
+    )
+
+
+class DestroyInput(BaseModel):
+    """Input for destroying a deployment."""
+
+    deployment_id: str = Field(
+        ...,
+        description="The deployment ID to destroy.",
+    )
+
+
+class TopupInput(BaseModel):
+    """Input for adding credits."""
+
+    amount_usd: float = Field(
+        ...,
+        description="Amount to add in USD.",
+    )
+    rail: str = Field(
+        "stripe",
+        description="Payment rail: 'stripe' (card) or 'xrp' (crypto).",
+    )
+
+
+# ── CrewAI tools ──────────────────────────────────────────────────
+
+
+class ProvisionStackSuggestTool(BaseTool):
+    """Get deployment suggestions for a desired outcome."""
+
+    name: str = "provision_stack_suggest"
+    description: str = (
+        f"Get deployment suggestions from {PRODUCT_NAME}. "
+        "Returns tier options with monthly cost estimates."
+    )
+    args_schema: Type[BaseModel] = SuggestInput
+    client: ProvisionStackClient
+
+    def _run(
+        self,
+        outcome: str,
+        region: str | None = None,
+        provider: str | None = None,
+        max_monthly_cost_usd: float | None = None,
+    ) -> str:
+        import json
+
+        result = self.client.suggest(
+            outcome=outcome,
+            region=region,
+            provider=provider,
+            max_monthly_cost_usd=max_monthly_cost_usd,
+        )
+        return json.dumps(result, indent=2)
+
+
+class ProvisionStackDeployTool(BaseTool):
+    """Deploy infrastructure for a selected suggestion."""
+
+    name: str = "provision_stack_deploy"
+    description: str = (
+        f"Deploy infrastructure via {PRODUCT_NAME}. "
+        "Fee held from balance, captured after verification (failed = free)."
+    )
+    args_schema: Type[BaseModel] = DeployInput
+    client: ProvisionStackClient
+
+    def _run(
+        self,
+        outcome: str,
+        tier: str | None = None,
+        region: str | None = None,
+        max_monthly_cost_usd: float | None = None,
+    ) -> str:
+        import json
+
+        result = self.client.deploy(
+            outcome=outcome,
+            tier=tier,
+            region=region,
+            max_monthly_cost_usd=max_monthly_cost_usd,
+        )
+        return json.dumps(result, indent=2)
+
+
+class ProvisionStackStatusTool(BaseTool):
+    """Check deployment status and verification results."""
+
+    name: str = "provision_stack_status"
+    description: str = f"Check status and verification of a {PRODUCT_NAME} deployment."
+    args_schema: Type[BaseModel] = StatusInput
+    client: ProvisionStackClient
+
+    def _run(self, deployment_id: str) -> str:
+        import json
+
+        result = self.client.status(deployment_id)
+        return json.dumps(result, indent=2)
+
+
+class ProvisionStackDestroyTool(BaseTool):
+    """Destroy all resources owned by a deployment."""
+
+    name: str = "provision_stack_destroy"
+    description: str = f"Destroy all cloud resources owned by a {PRODUCT_NAME} deployment."
+    args_schema: Type[BaseModel] = DestroyInput
+    client: ProvisionStackClient
+
+    def _run(self, deployment_id: str) -> str:
+        import json
+
+        result = self.client.destroy(deployment_id)
+        return json.dumps(result, indent=2)
+
+
+class ProvisionStackBalanceTool(BaseTool):
+    """Check account credit balance."""
+
+    name: str = "provision_stack_balance"
+    description: str = f"Check your {PRODUCT_NAME} account credit balance."
+    client: ProvisionStackClient
+
+    def _run(self) -> str:
+        import json
+
+        result = self.client.balance()
+        return json.dumps(result, indent=2)
+
+
+class ProvisionStackTopupTool(BaseTool):
+    """Add credits to your account."""
+
+    name: str = "provision_stack_topup"
+    description: str = f"Add credits to your {PRODUCT_NAME} account."
+    args_schema: Type[BaseModel] = TopupInput
+    client: ProvisionStackClient
+
+    def _run(self, amount_usd: float, rail: str = "stripe") -> str:
+        import json
+
+        if rail == "xrp":
+            result = self.client.topup_xrp(amount_usd)
+        else:
+            result = self.client.topup_stripe(amount_usd)
+        return json.dumps(result, indent=2)
+
+
+def get_provision_stack_tools(
+    api_key: str | None = None,
+    base_url: str | None = None,
+    sandbox: bool = False,
+) -> list[BaseTool]:
+    """
+    Get all Provision Stack tools for CrewAI.
+
+    Parameters
+    ----------
+    api_key : str, optional
+        API token. Falls back to PROVISION_STACK_API_KEY env var.
+    base_url : str, optional
+        API base URL. Falls back to env var or default.
+    sandbox : bool
+        If True, use sandbox (mock Terraform, no real deploys).
+
+    Returns
+    -------
+    list[BaseTool]
+        CrewAI Tool instances ready to assign to an agent.
+    """
+    client = ProvisionStackClient(
+        api_key=api_key, base_url=base_url, sandbox=sandbox
+    )
+
+    return [
+        ProvisionStackSuggestTool(client=client),
+        ProvisionStackDeployTool(client=client),
+        ProvisionStackStatusTool(client=client),
+        ProvisionStackDestroyTool(client=client),
+        ProvisionStackBalanceTool(client=client),
+        ProvisionStackTopupTool(client=client),
+    ]

provision_stack_crewai-0.1.0.dist-info/METADATA

@@ -0,0 +1,85 @@
+Metadata-Version: 2.4
+Name: provision-stack-crewai
+Version: 0.1.0
+Summary: CrewAI tools for Provision Stack — deploy cloud infrastructure via AI agents.
+Project-URL: Homepage, https://provision-stack.com
+Project-URL: Documentation, https://docs.provision-stack.com
+Project-URL: Repository, https://github.com/TravisLinkey/provision-stack
+Author: Provision Stack
+License-Expression: MIT
+Keywords: ai-agents,aws,crewai,gcp,infrastructure,mcp
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.10
+Requires-Dist: crewai>=0.100.0
+Requires-Dist: provision-stack>=0.1.0
+Provides-Extra: test
+Requires-Dist: pytest-mock>=3.12; extra == 'test'
+Requires-Dist: pytest>=8.0; extra == 'test'
+Description-Content-Type: text/markdown
+
+# provision-stack-crewai
+
+> CrewAI tools for Provision Stack — deploy cloud infrastructure via AI agents.
+
+## Install
+
+```bash
+pip install provision-stack-crewai
+```
+
+## Quick Start
+
+```python
+from provision_stack_crewai import get_provision_stack_tools
+from crewai import Agent, Task, Crew
+
+tools = get_provision_stack_tools(api_key="your-token", sandbox=True)
+
+engineer = Agent(
+    role="Infrastructure Engineer",
+    goal="Deploy production infrastructure",
+    tools=tools,
+    llm="gpt-4o",
+)
+
+task = Task(
+    description="Deploy a production web application with a managed database",
+    agent=engineer,
+)
+
+crew = Crew(agents=[engineer], tasks=[task])
+result = crew.kickoff()
+```
+
+## Tools
+
+| Tool | Description |
+|------|-------------|
+| `provision_stack_suggest` | Get deployment suggestions with tier options and costs |
+| `provision_stack_deploy` | Deploy infrastructure (fee held, captured after verification) |
+| `provision_stack_status` | Check deployment status and verification results |
+| `provision_stack_destroy` | Destroy all resources owned by a deployment |
+| `provision_stack_balance` | Check account credit balance |
+| `provision_stack_topup` | Add credits via Stripe or XRP |
+
+## Configuration
+
+```python
+tools = get_provision_stack_tools(
+    api_key="your-token",         # or PROVISION_STACK_API_KEY env var
+    base_url="https://...",       # or PROVISION_STACK_API_URL env var
+    sandbox=True,                 # use sandbox (mock Terraform)
+)
+```
+
+## License
+
+MIT

provision_stack_crewai-0.1.0.dist-info/RECORD

@@ -0,0 +1,5 @@
+provision_stack_crewai/__init__.py,sha256=58MX8owPMDaOVz2gnkLblcMkAJUC7HVXdOQEPFKhMIU,385
+provision_stack_crewai/tools.py,sha256=3OvkaZv5uiBA2f_l84BxAaNgYl1-2gxfNOZGkpVDdbo,7087
+provision_stack_crewai-0.1.0.dist-info/METADATA,sha256=ytYnpXn74fk9K9J9WfUkmY6fv_emRtq3fXhoqpeQ0n0,2652
+provision_stack_crewai-0.1.0.dist-info/WHEEL,sha256=lCkmxWfQsSc9CfIClYeavTdQeEX2toPqufh9gI35EQA,87
+provision_stack_crewai-0.1.0.dist-info/RECORD,,

provision_stack_crewai-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.31.0
+Root-Is-Purelib: true
+Tag: py3-none-any