provision-stack-langchain 0.1.0__tar.gz

provision_stack_langchain-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_langchain-0.1.0/PKG-INFO

@@ -0,0 +1,78 @@
+Metadata-Version: 2.4
+Name: provision-stack-langchain
+Version: 0.1.0
+Summary: LangChain 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,gcp,infrastructure,langchain,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: langchain-core>=0.3.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-langchain
+
+> LangChain tools for Provision Stack — deploy cloud infrastructure via AI agents.
+
+## Install
+
+```bash
+pip install provision-stack-langchain
+```
+
+## Quick Start
+
+```python
+from provision_stack_langchain import get_provision_stack_tools
+
+tools = get_provision_stack_tools(api_key="your-token", sandbox=True)
+
+# Use with any LangChain agent
+from langchain_openai import ChatOpenAI
+from langchain.agents import initialize_agent, AgentType
+
+llm = ChatOpenAI(model="gpt-4o")
+agent = initialize_agent(tools, llm, agent=AgentType.STRUCTURED_CHAT_ZERO_SHOT_REACT_DESCRIPTION)
+
+result = agent.invoke("Deploy a production web application on AWS under $50/month")
+```
+
+## 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_langchain-0.1.0/README.md

@@ -0,0 +1,51 @@
+# provision-stack-langchain
+
+> LangChain tools for Provision Stack — deploy cloud infrastructure via AI agents.
+
+## Install
+
+```bash
+pip install provision-stack-langchain
+```
+
+## Quick Start
+
+```python
+from provision_stack_langchain import get_provision_stack_tools
+
+tools = get_provision_stack_tools(api_key="your-token", sandbox=True)
+
+# Use with any LangChain agent
+from langchain_openai import ChatOpenAI
+from langchain.agents import initialize_agent, AgentType
+
+llm = ChatOpenAI(model="gpt-4o")
+agent = initialize_agent(tools, llm, agent=AgentType.STRUCTURED_CHAT_ZERO_SHOT_REACT_DESCRIPTION)
+
+result = agent.invoke("Deploy a production web application on AWS under $50/month")
+```
+
+## 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_langchain-0.1.0/provision_stack_langchain/__init__.py

@@ -0,0 +1,14 @@
+"""
+Provision Stack tools for LangChain.
+
+Usage::
+
+    from provision_stack_langchain import get_provision_stack_tools
+
+    tools = get_provision_stack_tools(api_key="your-token")
+    agent = initialize_agent(tools, llm, agent=AgentType.STRUCTURED_CHAT_ZERO_SHOT_REACT_DESCRIPTION)
+"""
+
+from provision_stack_langchain.tools import get_provision_stack_tools
+
+__all__ = ["get_provision_stack_tools"]

provision_stack_langchain-0.1.0/provision_stack_langchain/tools.py

@@ -0,0 +1,241 @@
+"""
+LangChain tool wrappers for Provision Stack.
+
+Each tool is a LangChain ``Tool`` instance that wraps the
+corresponding Provision Stack API endpoint.
+"""
+
+from __future__ import annotations
+
+from typing import Any, Optional, Type
+
+from langchain_core.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: Optional[str] = Field(
+        None,
+        description="Cloud region (e.g. 'us-east-1', 'europe-west1').",
+    )
+    provider: Optional[str] = Field(
+        None,
+        description="Restrict to one provider: 'aws', 'gcp', 'azure', 'oracle'.",
+    )
+    max_monthly_cost_usd: Optional[float] = Field(
+        None,
+        description="Monthly cost cap in USD. Options above this are filtered out.",
+    )
+
+
+class DeployInput(BaseModel):
+    """Input for deploying infrastructure."""
+
+    outcome: str = Field(
+        ...,
+        description="Natural language description of what to deploy.",
+    )
+    tier: Optional[str] = Field(
+        None,
+        description="Deployment tier: 'Starter', 'MVP', 'Startup', 'Enterprise'.",
+    )
+    region: Optional[str] = Field(
+        None,
+        description="Target cloud region.",
+    )
+    max_monthly_cost_usd: Optional[float] = 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 (e.g. 'd-1234567890-abc123').",
+    )
+
+
+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).",
+    )
+
+
+# ── LangChain 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 (Starter/MVP/Startup/Enterprise) with "
+        "monthly cost estimates across multiple cloud providers."
+    )
+    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,
+    ) -> dict[str, Any]:
+        return self.client.suggest(
+            outcome=outcome,
+            region=region,
+            provider=provider,
+            max_monthly_cost_usd=max_monthly_cost_usd,
+        )
+
+
+class ProvisionStackDeployTool(BaseTool):
+    """Deploy infrastructure for a selected suggestion."""
+
+    name: str = "provision_stack_deploy"
+    description: str = (
+        f"Deploy infrastructure via {PRODUCT_NAME}. "
+        "The deploy fee is held from your balance and captured only "
+        "after verification passes (failed deployments are 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,
+    ) -> dict[str, Any]:
+        return self.client.deploy(
+            outcome=outcome,
+            tier=tier,
+            region=region,
+            max_monthly_cost_usd=max_monthly_cost_usd,
+        )
+
+
+class ProvisionStackStatusTool(BaseTool):
+    """Check deployment status and verification results."""
+
+    name: str = "provision_stack_status"
+    description: str = (
+        f"Check the status and verification results of a {PRODUCT_NAME} deployment."
+    )
+    args_schema: Type[BaseModel] = StatusInput
+    client: ProvisionStackClient
+
+    def _run(self, deployment_id: str) -> dict[str, Any]:
+        return self.client.status(deployment_id)
+
+
+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) -> dict[str, Any]:
+        return self.client.destroy(deployment_id)
+
+
+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) -> dict[str, Any]:
+        return self.client.balance()
+
+
+class ProvisionStackTopupTool(BaseTool):
+    """Add credits to your account."""
+
+    name: str = "provision_stack_topup"
+    description: str = (
+        f"Add credits to your {PRODUCT_NAME} account via Stripe (card) or XRP (crypto)."
+    )
+    args_schema: Type[BaseModel] = TopupInput
+    client: ProvisionStackClient
+
+    def _run(self, amount_usd: float, rail: str = "stripe") -> dict[str, Any]:
+        if rail == "xrp":
+            return self.client.topup_xrp(amount_usd)
+        return self.client.topup_stripe(amount_usd)
+
+
+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 LangChain.
+
+    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]
+        LangChain Tool instances ready to pass 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_langchain-0.1.0/pyproject.toml

@@ -0,0 +1,51 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "provision-stack-langchain"
+version = "0.1.0"
+description = "LangChain tools for Provision Stack — deploy cloud infrastructure via AI agents."
+readme = "README.md"
+license = "MIT"
+requires-python = ">=3.10"
+authors = [
+    { name = "Provision Stack" },
+]
+keywords = [
+    "langchain",
+    "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",
+    "langchain-core>=0.3.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_langchain-0.1.0/tests/test_tools.py

@@ -0,0 +1,117 @@
+"""Tests for provision_stack_langchain.tools — LangChain tool wrappers."""
+
+from __future__ import annotations
+
+import sys
+import types
+from typing import Any, Type
+import pytest
+
+
+# ── Mock setup ────────────────────────────────────────────────────
+# LangChain's BaseTool uses Pydantic metaclass magic; we mock it
+# to avoid pulling in langchain-core 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 langchain_core.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
+lc_tools = types.ModuleType("langchain_core.tools")
+lc_tools.BaseTool = MockBaseTool
+
+lc = types.ModuleType("langchain_core")
+lc.tools = lc_tools
+
+pydantic = types.ModuleType("pydantic")
+pydantic.BaseModel = MockBaseModel
+pydantic.Field = mock_field
+
+sys.modules.setdefault("langchain_core", lc)
+sys.modules.setdefault("langchain_core.tools", lc_tools)
+sys.modules.setdefault("pydantic", pydantic)
+
+from provision_stack_langchain.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"