provision-stack-crewai 0.1.0__tar.gz

provision_stack_crewai-0.1.0/.gitignore

@@ -0,0 +1,34 @@
+node_modules
+dist
+build
+coverage
+
+.env
+.env.*
+!.env.example
+
+.turbo
+
+.DS_Store
+
+.vscode/settings.json
+
+*.log
+
+# Generated Terraform deployment artifacts (written to user's cwd, not repo)
+terraform/
+
+# Local SQLite state (DATABASE_PATH default location)
+data/
+
+# Local deploy secrets (NEVER commit — deploy-time env lives here)
+.env.local
+deploy/.env
+
+# Host SSH keys + terraform state (local artifacts)
+deploy/hosting/provision-stack-host-key
+deploy/hosting/provision-stack-host-key.pub
+deploy/hosting/.terraform/
+deploy/hosting/terraform.tfstate*
+deploy/hosting/.terraform.lock.hcl
+__pycache__/

provision_stack_crewai-0.1.0/PKG-INFO

@@ -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/README.md

@@ -0,0 +1,58 @@
+# 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/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-0.1.0/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/pyproject.toml

@@ -0,0 +1,51 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "provision-stack-crewai"
+version = "0.1.0"
+description = "CrewAI tools for Provision Stack — deploy cloud infrastructure via AI agents."
+readme = "README.md"
+license = "MIT"
+requires-python = ">=3.10"
+authors = [
+    { name = "Provision Stack" },
+]
+keywords = [
+    "crewai",
+    "ai-agents",
+    "infrastructure",
+    "aws",
+    "gcp",
+    "mcp",
+]
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Topic :: Software Development :: Libraries :: Python Modules",
+]
+dependencies = [
+    "provision-stack>=0.1.0",
+    "crewai>=0.100.0",
+]
+
+[project.urls]
+Homepage = "https://provision-stack.com"
+Documentation = "https://docs.provision-stack.com"
+Repository = "https://github.com/TravisLinkey/provision-stack"
+
+[project.optional-dependencies]
+test = [
+    "pytest>=8.0",
+    "pytest-mock>=3.12",
+]
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]

provision_stack_crewai-0.1.0/tests/test_tools.py

@@ -0,0 +1,115 @@
+"""Tests for provision_stack_crewai.tools — CrewAI tool wrappers."""
+
+from __future__ import annotations
+
+import sys
+import types
+from typing import Any, Type
+import pytest
+
+
+# ── Mock setup ────────────────────────────────────────────────────
+# CrewAI's BaseTool uses Pydantic metaclass magic; we mock it
+# to avoid pulling in crewai while keeping class inheritance.
+
+
+class MockBaseModel:
+    """Stand-in for pydantic.BaseModel."""
+    model_config: dict[str, Any] = {}
+
+    def __init__(self, **kwargs: Any) -> None:
+        for k, v in kwargs.items():
+            setattr(self, k, v)
+
+
+def mock_field(*args: Any, **kwargs: Any) -> Any:
+    return kwargs.get("default", args[0] if args else None)
+
+
+class MockBaseTool:
+    """Stand-in for crewai.tools.BaseTool — supports subclassing."""
+
+    name: str = ""
+    description: str = ""
+    args_schema: Type[MockBaseModel] | None = None
+    client: Any = None
+
+    def __init__(self, **kwargs: Any) -> None:
+        for k, v in kwargs.items():
+            setattr(self, k, v)
+
+    def _run(self, **kwargs: Any) -> Any:
+        raise NotImplementedError
+
+
+# Install mocks before importing our modules
+crewai = types.ModuleType("crewai")
+crewai_tools = types.ModuleType("crewai.tools")
+crewai_tools.BaseTool = MockBaseTool
+
+pydantic = types.ModuleType("pydantic")
+pydantic.BaseModel = MockBaseModel
+pydantic.Field = mock_field
+
+sys.modules.setdefault("crewai", crewai)
+sys.modules.setdefault("crewai.tools", crewai_tools)
+sys.modules.setdefault("pydantic", pydantic)
+
+from provision_stack_crewai.tools import get_provision_stack_tools
+
+
+# ── Tool factory ──────────────────────────────────────────────────
+
+
+class TestGetProvisionStackTools:
+    def test_returns_list(self):
+        tools = get_provision_stack_tools(api_key="test")
+        assert isinstance(tools, list)
+
+    def test_returns_six_tools(self):
+        tools = get_provision_stack_tools(api_key="test")
+        assert len(tools) == 6
+
+    def test_tool_names(self):
+        tools = get_provision_stack_tools(api_key="test")
+        names = {t.name for t in tools}
+        assert names == {
+            "provision_stack_suggest",
+            "provision_stack_deploy",
+            "provision_stack_status",
+            "provision_stack_destroy",
+            "provision_stack_balance",
+            "provision_stack_topup",
+        }
+
+    def test_tools_have_descriptions(self):
+        tools = get_provision_stack_tools(api_key="test")
+        for tool in tools:
+            assert tool.description
+            assert len(tool.description) > 10
+
+    def test_tools_have_args_schema(self):
+        tools = get_provision_stack_tools(api_key="test")
+        # All tools except balance have an args_schema (balance takes no input)
+        for tool in tools:
+            if tool.name == "provision_stack_balance":
+                assert tool.args_schema is None
+            else:
+                assert tool.args_schema is not None
+
+    def test_tools_share_client(self):
+        tools = get_provision_stack_tools(api_key="test")
+        clients = {id(t.client) for t in tools}
+        assert len(clients) == 1
+
+    def test_sandbox_mode(self):
+        from provision_stack.config import SANDBOX_URL
+
+        tools = get_provision_stack_tools(api_key="test", sandbox=True)
+        assert tools[0].client.base_url == SANDBOX_URL
+
+    def test_custom_base_url(self):
+        tools = get_provision_stack_tools(
+            api_key="test", base_url="https://custom.api.com"
+        )
+        assert tools[0].client.base_url == "https://custom.api.com"