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

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/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.dist-info/METADATA

@@ -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.dist-info/RECORD

@@ -0,0 +1,5 @@
+provision_stack_langchain/__init__.py,sha256=yEBsPuGnANVGkle48hmluoaxgjsu_phQaG3pPxjw2VA,398
+provision_stack_langchain/tools.py,sha256=vAE3YWQXdvVDnOw2SklBKUpliOOp3GJ8tpXhMac6tFs,7062
+provision_stack_langchain-0.1.0.dist-info/METADATA,sha256=3-vy0Tp_cJHGoc4fuek2CWodkbzRs2e6426ZF8rPVU0,2666
+provision_stack_langchain-0.1.0.dist-info/WHEEL,sha256=lCkmxWfQsSc9CfIClYeavTdQeEX2toPqufh9gI35EQA,87
+provision_stack_langchain-0.1.0.dist-info/RECORD,,

provision_stack_langchain-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