antz-studio 0.1.0__tar.gz

antz_studio-0.1.0/PKG-INFO

@@ -0,0 +1,34 @@
+Metadata-Version: 2.4
+Name: antz-studio
+Version: 0.1.0
+Summary: Antz Studio -- Sovereign Agentic OS
+Author-email: Antz Studio <gabriel@antz.studio>
+License: AGPL-3.0
+Project-URL: Homepage, https://antz.studio
+Project-URL: Repository, https://github.com/Gabbsx7/antz-studio-aos
+Keywords: agents,sovereign-ai,agentic-os,llm,observability,nest,colony,hive
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: httpx>=0.27
+Requires-Dist: typer>=0.12
+Requires-Dist: pyyaml>=6.0
+Requires-Dist: opentelemetry-api>=1.24
+Requires-Dist: opentelemetry-sdk>=1.24
+Requires-Dist: opentelemetry-exporter-otlp-proto-http>=1.24
+Requires-Dist: rich>=13.0
+Requires-Dist: pydantic>=2.0
+Requires-Dist: sqlalchemy>=2.0
+Provides-Extra: dev
+Requires-Dist: pytest>=8.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.23; extra == "dev"
+Requires-Dist: pytest-mock>=3.12; extra == "dev"
+Provides-Extra: cloud
+Requires-Dist: azure-mgmt-compute>=30.0.0; extra == "cloud"
+Requires-Dist: azure-mgmt-network>=25.0.0; extra == "cloud"
+Requires-Dist: azure-mgmt-resource>=23.0.0; extra == "cloud"
+Requires-Dist: azure-identity>=1.15.0; extra == "cloud"

antz_studio-0.1.0/antz/__init__.py

@@ -0,0 +1,29 @@
+# Antz Studio -- Colony SDK
+# Sovereign Agentic OS - antz.studio
+
+from antz.decorators import agent, tool, memory
+from antz.client import AntzClient
+from antz.config import AntzConfig
+from antz.hive_mode import HiveMode
+from antz.db_connector import db_connector
+from antz.privacy import add_gaussian_noise
+from antz.exceptions import (
+    AntzConnectionError,
+    ConstitutionViolationError,
+    AntzAuthError,
+)
+
+__version__ = "0.1.0"
+__all__ = [
+    "agent",
+    "tool",
+    "memory",
+    "HiveMode",
+    "db_connector",
+    "add_gaussian_noise",
+    "AntzClient",
+    "AntzConfig",
+    "AntzConnectionError",
+    "ConstitutionViolationError",
+    "AntzAuthError",
+]

antz_studio-0.1.0/antz/cli.py

@@ -0,0 +1,545 @@
+from __future__ import annotations
+
+import subprocess
+import sys
+from pathlib import Path
+from typing import Optional
+
+import typer
+import yaml
+from rich.console import Console
+from rich.panel import Panel
+from rich.table import Table
+
+app     = typer.Typer(help="Ant'z Studio — Sovereign Agentic OS CLI", add_completion=False)
+console = Console()
+
+
+# ── antz login ────────────────────────────────────────────────────────────────
+
+@app.command()
+def login(
+    nest: str = typer.Option(..., "--nest", help="Nest URL (e.g. http://nest.local:8001)"),
+    api_key: str = typer.Option(..., "--api-key", help="Nest API key"),
+    mode: str = typer.Option("isolated", "--mode", help="isolated | shared | sovereign"),
+):
+    """Authenticate with a Nest and save credentials locally."""
+    from antz.config import AntzConfig
+    from antz.client import AntzClient
+
+    console.print(f"[dim]Connecting to Nest at {nest}...[/dim]")
+
+    try:
+        cfg    = AntzConfig(nest_url=nest, api_key=api_key, mode=mode)
+        client = AntzClient(cfg)
+        health = client.health()
+
+        AntzConfig.save_global(nest_url=nest, api_key=api_key, mode=mode)
+
+        console.print(Panel.fit(
+            f"[green]✓ Connected[/green]\n\n"
+            f"  Nest:    [cyan]{nest}[/cyan]\n"
+            f"  Mode:    [yellow]{mode}[/yellow]\n"
+            f"  Version: {health.get('version', 'unknown')}\n\n"
+            f"Credentials saved to [dim]~/.antz/config.yaml[/dim]",
+            title="Ant'z Studio",
+            border_style="green",
+        ))
+
+    except Exception as e:
+        console.print(f"[red]✗ Connection failed:[/red] {e}")
+        console.print("[dim]Hint: make sure the Nest is running and the API key is correct.[/dim]")
+        raise typer.Exit(1)
+
+
+# ── antz status ───────────────────────────────────────────────────────────────
+
+@app.command()
+def status():
+    """Check connection to the configured Nest."""
+    from antz.config import AntzConfig
+    from antz.client import AntzClient
+
+    try:
+        cfg    = AntzConfig.load()
+        client = AntzClient(cfg)
+        health = client.health()
+
+        table = Table(show_header=False, box=None, padding=(0, 2))
+        table.add_row("[dim]Nest URL[/dim]",  f"[cyan]{cfg.nest_url}[/cyan]")
+        table.add_row("[dim]Mode[/dim]",      f"[yellow]{cfg.mode}[/yellow]")
+        table.add_row("[dim]Status[/dim]",    "[green]● online[/green]")
+        table.add_row("[dim]Version[/dim]",   health.get("version", "unknown"))
+
+        console.print(Panel(table, title="Ant'z Nest Status", border_style="green"))
+
+    except Exception as e:
+        console.print(f"[red]✗ Cannot reach Nest:[/red] {e}")
+        console.print("[dim]Run: antz login --nest <url> --api-key <key>[/dim]")
+        raise typer.Exit(1)
+
+
+# ── antz init colony ──────────────────────────────────────────────────────────
+
+@app.command()
+def init(
+    resource: str = typer.Argument(..., help="Resource type: colony"),
+    name: str = typer.Argument(..., help="Colony name (e.g. financial-credit-analysis)"),
+    framework: str = typer.Option("langgraph", "--framework", "-f",
+                                   help="langgraph | crewai | mixed"),
+):
+    """Scaffold a new Colony with the chosen framework template."""
+    if resource != "colony":
+        console.print(f"[red]Unknown resource '{resource}'. Currently supported: colony[/red]")
+        raise typer.Exit(1)
+
+    target = Path.cwd() / f"colony-{name}"
+
+    if target.exists():
+        console.print(f"[red]✗ Directory '{target.name}' already exists.[/red]")
+        raise typer.Exit(1)
+
+    _scaffold_colony(name=name, framework=framework, target=target)
+
+    console.print(Panel.fit(
+        f"[green]✓ Colony scaffolded[/green]\n\n"
+        f"  Name:      [cyan]{name}[/cyan]\n"
+        f"  Framework: [yellow]{framework}[/yellow]\n"
+        f"  Directory: [dim]{target}[/dim]\n\n"
+        f"Next steps:\n"
+        f"  [dim]cd colony-{name}[/dim]\n"
+        f"  [dim]code .[/dim]                  # open in VSCode\n"
+        f"  [dim]antz run --demo[/dim]         # run the example agent",
+        title="Colony created",
+        border_style="cyan",
+    ))
+
+
+# ── antz run ──────────────────────────────────────────────────────────────────
+
+@app.command()
+def run(
+    input_file: Optional[Path] = typer.Option(None, "--input", "-i",
+                                               help="JSON input file for the agent"),
+    demo: bool = typer.Option(False, "--demo", help="Run the built-in demo agent"),
+    colony_dir: Optional[Path] = typer.Option(None, "--colony", "-c",
+                                               help="Colony directory (default: cwd)"),
+):
+    """Execute the Colony agent."""
+    cwd = colony_dir or Path.cwd()
+    cfg_file = cwd / "antz.config.yaml"
+
+    if not cfg_file.exists():
+        console.print(
+            "[red]✗ No antz.config.yaml found.[/red]\n"
+            "[dim]Run from inside a Colony directory, or use: antz init colony <name>[/dim]"
+        )
+        raise typer.Exit(1)
+
+    with open(cfg_file) as f:
+        colony_cfg = yaml.safe_load(f) or {}
+
+    colony_id  = colony_cfg.get("colony_id", "unknown")
+    entrypoint = colony_cfg.get("entrypoint", "agents/main.py")
+
+    console.print(f"[dim]Running Colony [cyan]{colony_id}[/cyan]...[/dim]")
+
+    if demo:
+        _run_demo(cwd, colony_cfg)
+        return
+
+    entry_path = cwd / entrypoint
+    if not entry_path.exists():
+        console.print(f"[red]✗ Entrypoint not found: {entry_path}[/red]")
+        raise typer.Exit(1)
+
+    cmd = [sys.executable, str(entry_path)]
+    if input_file:
+        cmd += ["--input", str(input_file)]
+
+    result = subprocess.run(cmd, cwd=str(cwd))
+    raise typer.Exit(result.returncode)
+
+
+# ── antz version ─────────────────────────────────────────────────────────────
+
+@app.command()
+def version():
+    """Show SDK version."""
+    from antz import __version__
+    console.print(f"antz-sdk [cyan]{__version__}[/cyan]")
+
+
+# ── Internal helpers ──────────────────────────────────────────────────────────
+
+def _scaffold_colony(name: str, framework: str, target: Path) -> None:
+    """Create Colony directory structure from template."""
+    target.mkdir(parents=True)
+
+    (target / "agents").mkdir()
+    (target / "tools").mkdir()
+    (target / "workflows").mkdir()
+    (target / "tests").mkdir()
+
+    # antz.config.yaml
+    config = {
+        "colony_id":  name,
+        "framework":  framework,
+        "entrypoint": "agents/main.py",
+        "nest_url":   "http://localhost:8001",
+        "mode":       "isolated",
+    }
+    with open(target / "antz.config.yaml", "w", encoding="utf-8") as f:
+        yaml.dump(config, f, default_flow_style=False)
+
+    # constitution.yaml
+    constitution = {
+        "agent_id": f"{name}-v1",
+        "version":  "1.0",
+        "allowed_tools": [],
+        "prohibited_actions": ["delete_record", "export_data", "modify_audit_log"],
+        "spend_limits": {"max_per_action_usd": 5.0, "max_per_day_usd": 500.0},
+        "data_policy": {
+            "can_send_outside_nest": False,
+            "can_log_pii": False,
+            "audit_all_tool_calls": True,
+        },
+    }
+    with open(target / "constitution.yaml", "w", encoding="utf-8") as f:
+        yaml.dump(constitution, f, default_flow_style=False)
+
+    # agents/main.py
+    agent_code = _agent_template(name=name, framework=framework)
+    with open(target / "agents" / "main.py", "w", encoding="utf-8") as f:
+        f.write(agent_code)
+
+    # README
+    with open(target / "README.md", "w", encoding="utf-8") as f:
+        f.write(_readme_template(name=name, framework=framework))
+
+
+def _agent_template(name: str, framework: str) -> str:
+    safe_name = name.replace("-", "_")
+
+    if framework == "langgraph":
+        return f'''# Colony: {name}
+# Framework: LangGraph
+# Ant'z SDK injects: observability, audit, constitution validation, memory
+
+from antz import agent, tool, memory
+from typing import TypedDict
+
+
+# ── State ─────────────────────────────────────────────────────────────────────
+
+class AgentState(TypedDict):
+    input: dict
+    result: dict
+    status: str
+
+
+# ── Tools ─────────────────────────────────────────────────────────────────────
+
+@tool("{safe_name}-lookup")
+def lookup_data(query: str) -> dict:
+    """
+    Replace with your actual tool logic.
+    Ant'z automatically logs this call and validates against constitution.
+    """
+    return {{"data": f"mock result for: {{query}}", "confidence": 0.9}}
+
+
+# ── Main agent ────────────────────────────────────────────────────────────────
+
+@agent("{safe_name}-v1")
+@memory(namespace="{safe_name}")
+def run_agent(input_data: dict, memory_context: list = None) -> dict:
+    """
+    Main agent entrypoint.
+    memory_context is auto-populated with relevant past runs from Hive.
+    Replace the logic below with your LangGraph workflow.
+    """
+    past    = memory_context or []
+    context = past[0]["content"] if past else "No prior context."
+
+    result = lookup_data(str(input_data))
+
+    return {{
+        "status":       "success",
+        "colony":       "{name}",
+        "result":       result,
+        "memory_hits":  len(past),
+        "context_used": context,
+    }}
+
+
+# ── Demo entrypoint ───────────────────────────────────────────────────────────
+
+if __name__ == "__main__":
+    import json, sys
+
+    sample_input = {{"query": "demo run", "source": "antz-cli"}}
+
+    if "--input" in sys.argv:
+        idx = sys.argv.index("--input")
+        with open(sys.argv[idx + 1]) as f:
+            sample_input = json.load(f)
+
+    output = run_agent(sample_input)
+    print(json.dumps(output, indent=2))
+'''
+
+    elif framework == "crewai":
+        return f'''# Colony: {name}
+# Framework: CrewAI
+# Ant'z SDK injects: observability, audit, constitution validation, memory
+
+from antz import agent, tool, memory
+from crewai import Agent, Task, Crew
+
+
+# ── Tools ─────────────────────────────────────────────────────────────────────
+
+@tool("{safe_name}-lookup")
+def lookup_data(query: str) -> str:
+    """Replace with your actual tool logic."""
+    return f"mock result for: {{query}}"
+
+
+# ── Crew definition ───────────────────────────────────────────────────────────
+
+def build_crew():
+    analyst = Agent(
+        role="Data Analyst",
+        goal="Analyze input data and produce a structured report",
+        backstory="Expert analyst with deep domain knowledge",
+        verbose=False,
+    )
+    task = Task(
+        description="Analyze the provided data and return a structured summary.",
+        expected_output="A JSON-formatted analysis report",
+        agent=analyst,
+    )
+    return Crew(agents=[analyst], tasks=[task], verbose=False)
+
+
+# ── Main agent ────────────────────────────────────────────────────────────────
+
+@agent("{safe_name}-v1")
+@memory(namespace="{safe_name}")
+def run_agent(input_data: dict, memory_context: list = None) -> dict:
+    """
+    Main agent entrypoint.
+    memory_context is auto-populated with relevant past runs from Hive.
+    """
+    past   = memory_context or []
+    crew   = build_crew()
+    result = crew.kickoff(inputs=input_data)
+    return {{
+        "status":      "success",
+        "colony":      "{name}",
+        "result":      str(result),
+        "memory_hits": len(past),
+    }}
+
+
+# ── Demo entrypoint ───────────────────────────────────────────────────────────
+
+if __name__ == "__main__":
+    import json, sys
+
+    sample_input = {{"query": "demo run", "source": "antz-cli"}}
+
+    if "--input" in sys.argv:
+        idx = sys.argv.index("--input")
+        with open(sys.argv[idx + 1]) as f:
+            sample_input = json.load(f)
+
+    output = run_agent(sample_input)
+    print(json.dumps(output, indent=2))
+'''
+
+    else:  # mixed
+        return f'''# Colony: {name}
+# Framework: Mixed (LangGraph + raw LiteLLM)
+# Ant'z SDK injects: observability, audit, constitution validation, memory
+
+from antz import agent, tool, memory
+import httpx
+import os
+
+LITELLM_URL = os.getenv("ANTZ_LITELLM_URL", "http://localhost:4000")
+
+
+# ── Tools ─────────────────────────────────────────────────────────────────────
+
+@tool("{safe_name}-llm-call")
+def call_llm(prompt: str) -> str:
+    """Direct LiteLLM call through the Nest — stays sovereign."""
+    r = httpx.post(
+        f"{{LITELLM_URL}}/chat/completions",
+        json={{
+            "model": "antz-default",
+            "messages": [{{"role": "user", "content": prompt}}],
+            "max_tokens": 512,
+        }},
+        timeout=30,
+    )
+    r.raise_for_status()
+    return r.json()["choices"][0]["message"]["content"]
+
+
+# ── Main agent ────────────────────────────────────────────────────────────────
+
+@agent("{safe_name}-v1")
+@memory(namespace="{safe_name}")
+def run_agent(input_data: dict, memory_context: list = None) -> dict:
+    """
+    Main agent with memory context from Hive.
+    memory_context is auto-populated with relevant past runs.
+    """
+    past        = memory_context or []
+    context_str = "\\n".join(str(m) for m in past[:3]) if past else "No prior context."
+
+    prompt = f"""
+Context from previous runs:
+{{context_str}}
+
+Current input: {{input_data}}
+
+Analyze and respond with a structured result.
+"""
+
+    response = call_llm(prompt)
+
+    return {{
+        "status":      "success",
+        "colony":      "{name}",
+        "result":      response,
+        "memory_hits": len(past),
+    }}
+
+
+# ── Demo entrypoint ───────────────────────────────────────────────────────────
+
+if __name__ == "__main__":
+    import json, sys
+
+    sample_input = {{"query": "demo run", "source": "antz-cli"}}
+
+    if "--input" in sys.argv:
+        idx = sys.argv.index("--input")
+        with open(sys.argv[idx + 1]) as f:
+            sample_input = json.load(f)
+
+    output = run_agent(sample_input)
+    print(json.dumps(output, indent=2))
+'''
+
+
+def _readme_template(name: str, framework: str) -> str:
+    return f"""# Colony: {name}
+
+Framework: `{framework}`
+
+## Run
+
+```bash
+# Demo (no input required)
+antz run --demo
+
+# With input file
+antz run --input data/sample.json
+```
+
+## Structure
+
+```
+agents/main.py       # Main agent logic — edit this
+tools/               # Add your tools here
+workflows/           # Workflow definitions
+constitution.yaml    # Agent behavioral constraints
+antz.config.yaml     # Colony configuration
+```
+
+## Constitution
+
+Edit `constitution.yaml` to define what this agent can and cannot do.
+The Ant'z Hive validates every tool call against these rules at runtime.
+
+## Observability
+
+All executions are automatically traced. View them in the Nest Grafana dashboard.
+"""
+
+
+def _run_demo(cwd: Path, colony_cfg: dict) -> None:
+    """Run the demo mode — executes agents/main.py with sample data."""
+    entry = cwd / "agents" / "main.py"
+    if not entry.exists():
+        console.print("[red]✗ agents/main.py not found.[/red]")
+        raise typer.Exit(1)
+
+    console.print("[dim]Running demo agent...[/dim]\n")
+    result = subprocess.run([sys.executable, str(entry)], cwd=str(cwd))
+    raise typer.Exit(result.returncode)
+
+
+
+
+# -- antz nest ----------------------------------------------------------------
+
+@app.command("nest")
+def nest_cmd(
+    action: str = typer.Argument(..., help="status"),
+    tenant: str = typer.Option(..., "--tenant", "-t", help="Tenant ID"),
+    watch: bool = typer.Option(False, "--watch", "-w"),
+    api_key: str = typer.Option(None, "--api-key"),
+):
+    """Manage Nest provisioning and status."""
+    import httpx, os, time
+    cp_url = os.getenv("ANTZ_CONTROL_PLANE_URL", "https://control.antz.studio")
+    if not api_key:
+        api_key = typer.prompt("Control Plane API Key")
+    STATUS = {
+        "pending": "Aguardando...", "provisioning": "Iniciando...",
+        "creating_vm": "Criando VM...", "vm_created": "VM criada...",
+        "wireguard_ready": "WireGuard pronto...", "provisioning_nest": "Instalando Nest...",
+        "sending_email": "Enviando email...", "active": "Nest ativo!", "failed": "Falhou",
+    }
+    def get():
+        try:
+            r = httpx.get(f"{cp_url}/tenants/{tenant}", headers={"x-api-key": api_key}, timeout=10)
+            return r.json() if r.status_code == 200 else None
+        except Exception:
+            return None
+    data = get()
+    if not data:
+        console.print(f"[red]Tenant {tenant!r} nao encontrado.[/red]")
+        raise typer.Exit(1)
+    if not watch:
+        console.print(f"Tenant: [cyan]{tenant}[/cyan]")
+        console.print(f"Status: {STATUS.get(data.get('status','?'), data.get('status','?'))}")
+        if data.get("nest_url"):
+            console.print(f"URL: [green]{data['nest_url']}[/green]")
+        return
+    last = None
+    while True:
+        data = get()
+        if not data:
+            time.sleep(10); continue
+        s = data.get("status", "")
+        if s != last:
+            console.print(STATUS.get(s, s)); last = s
+        if s == "active":
+            url = data.get("nest_url", "")
+            console.print(f"[green]Nest ativo em {url}[/green]")
+            console.print(f"[dim]antz login --nest {url} --api-key SUA_CHAVE[/dim]")
+            raise typer.Exit(0)
+        if s == "failed":
+            console.print("[red]Falhou.[/red]"); raise typer.Exit(1)
+        time.sleep(15)
+
+
+if __name__ == "__main__":
+    app()