aicc-sdk 0.1.0__py3-none-any.whl

aicc/__init__.py

@@ -0,0 +1,158 @@
+"""aicc - thin client for AI Command Center.
+
+The gateway does all the real work; this helper just points the official
+provider SDKs at it via their standard environment variables.
+
+    import aicc
+    aicc.init(project="invoice-bot")          # BEFORE creating any clients
+
+    from openai import OpenAI
+    client = OpenAI()                          # now flows through the gateway
+
+Works with: openai (OPENAI_BASE_URL), anthropic (ANTHROPIC_BASE_URL),
+google-genai (GOOGLE_GEMINI_BASE_URL), and anything else via aicc.url("provider").
+Your API keys are untouched - the gateway passes them through.
+"""
+
+from __future__ import annotations
+
+import json
+import os
+import sys
+import time
+import urllib.request
+from dataclasses import dataclass, field
+from typing import Any, Dict, Optional
+from urllib.parse import quote
+
+__all__ = ["init", "url", "track", "AiccSession"]
+__version__ = "0.1.0"
+
+DEFAULT_GATEWAY = "http://localhost:4321"
+
+_session: Optional["AiccSession"] = None
+
+
+@dataclass
+class AiccSession:
+    gateway: str
+    project: str
+    key: Optional[str] = None
+    env: Dict[str, str] = field(default_factory=dict)
+
+    def _base(self) -> str:
+        # With auth enabled the gateway key both authenticates and attributes the
+        # call; without it, the project name in the path does the grouping.
+        return (
+            f"{self.gateway}/k/{quote(self.key, safe='')}"
+            if self.key
+            else f"{self.gateway}/p/{quote(self.project, safe='')}"
+        )
+
+    def url(self, provider: str) -> str:
+        """Gateway base URL for any provider, e.g. url('openai') or url('ollama')."""
+        base = f"{self._base()}/{provider}"
+        return base + "/v1" if provider not in ("anthropic", "gemini") else base
+
+
+def init(
+    project: str = "default",
+    key: Optional[str] = None,
+    gateway: Optional[str] = None,
+    check: bool = True,
+) -> AiccSession:
+    """Route this process's AI SDK traffic through AI Command Center.
+
+    Call BEFORE constructing OpenAI()/Anthropic()/genai.Client() - the SDKs
+    read their base-URL environment variables at construction time.
+
+    project  - how calls are grouped on the dashboard.
+    key      - project gateway key (required when the gateway has auth enabled;
+               defaults to $AICC_KEY). Get it from the dashboard → settings.
+    gateway  - gateway origin; defaults to $AICC_GATEWAY or http://localhost:4321.
+    check    - ping the gateway and print a warning if it is unreachable.
+    """
+    global _session
+    gw = (gateway or os.environ.get("AICC_GATEWAY") or DEFAULT_GATEWAY).rstrip("/")
+    key = key or os.environ.get("AICC_KEY") or None
+    base = f"{gw}/k/{quote(key, safe='')}" if key else f"{gw}/p/{quote(project, safe='')}"
+
+    env = {
+        "OPENAI_BASE_URL": f"{base}/openai/v1",
+        "OPENAI_API_BASE": f"{base}/openai/v1",  # legacy libs (openai<1, some frameworks)
+        "ANTHROPIC_BASE_URL": f"{base}/anthropic",
+        "GOOGLE_GEMINI_BASE_URL": f"{base}/gemini",
+    }
+    os.environ.update(env)
+    _session = AiccSession(gateway=gw, project=project, key=key, env=env)
+
+    if check:
+        try:
+            with urllib.request.urlopen(f"{gw}/health", timeout=1.5) as res:
+                json.loads(res.read().decode("utf-8"))
+        except Exception:
+            print(
+                f"[aicc] warning: no AI Command Center gateway at {gw} - "
+                f"calls will fail until you run: npx ai-command-center start",
+                file=sys.stderr,
+            )
+    return _session
+
+
+def url(provider: str) -> str:
+    """Gateway base URL for a provider (requires init() first)."""
+    if _session is None:
+        raise RuntimeError("aicc.init() has not been called")
+    return _session.url(provider)
+
+
+def track(
+    *,
+    project: Optional[str] = None,
+    provider: str = "custom",
+    model: Optional[str] = None,
+    tokens_in: Optional[int] = None,
+    tokens_out: Optional[int] = None,
+    cost_usd: Optional[float] = None,
+    latency_ms: Optional[float] = None,
+    ok: bool = True,
+    error: Optional[str] = None,
+    **extra: Any,
+) -> bool:
+    """Report usage for calls the gateway can't see (batch jobs, exotic providers).
+
+    Returns True when the record was accepted, False when the gateway was
+    unreachable (never raises - telemetry must not break the host app).
+    """
+    gw = _session.gateway if _session else (os.environ.get("AICC_GATEWAY") or DEFAULT_GATEWAY).rstrip("/")
+    gw_key = (_session.key if _session else None) or os.environ.get("AICC_KEY")
+    record = {
+        "project": project or (_session.project if _session else "default"),
+        "provider": provider,
+        "model": model,
+        "tokensIn": tokens_in,
+        "tokensOut": tokens_out,
+        "costUsd": cost_usd,
+        "latencyMs": latency_ms,
+        "ok": ok,
+        "error": error,
+        "ts": int(time.time() * 1000),
+        **extra,
+    }
+    # Drop None-valued keys so the gateway treats them as absent (and, e.g.,
+    # prices the record itself) rather than coercing null to 0.
+    record = {k: v for k, v in record.items() if v is not None}
+    headers = {"Content-Type": "application/json"}
+    if gw_key:
+        headers["x-aicc-key"] = gw_key
+    req = urllib.request.Request(
+        f"{gw}/api/track",
+        data=json.dumps(record).encode("utf-8"),
+        headers=headers,
+        method="POST",
+    )
+    try:
+        with urllib.request.urlopen(req, timeout=3) as res:
+            return 200 <= res.status < 300
+    except Exception:
+        return False

aicc_sdk-0.1.0.dist-info/METADATA

@@ -0,0 +1,45 @@
+Metadata-Version: 2.4
+Name: aicc-sdk
+Version: 0.1.0
+Summary: Thin helper that routes your OpenAI/Anthropic/Gemini SDK calls through an AI Command Center gateway (one line: aicc.init(project=...)).
+License: MIT
+Keywords: llm,observability,cost-tracking,openai,anthropic,gemini
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Dynamic: license-file
+
+# aicc-sdk (Python)
+
+Thin helper for [AI Command Center](../../README.md). One line routes every
+OpenAI / Anthropic / Gemini SDK call in your process through the gateway, so
+usage and cost land on the consolidated dashboard.
+
+```python
+import aicc
+aicc.init(project="invoice-bot")   # BEFORE creating clients
+
+from openai import OpenAI
+client = OpenAI()                   # unchanged code, now fully tracked
+```
+
+If the gateway has login enabled, pass the project's gateway key (from the
+dashboard → settings, or `$AICC_KEY`):
+
+```python
+aicc.init(project="invoice-bot", key="aicc_…")   # or set AICC_KEY in the env
+```
+
+The SDK only sets standard environment variables (`OPENAI_BASE_URL`,
+`ANTHROPIC_BASE_URL`, `GOOGLE_GEMINI_BASE_URL`) - you can also set those by
+hand and skip this package entirely. Your API keys never change hands.
+
+Extras:
+
+```python
+aicc.url("mistral")                       # gateway base URL for any provider
+aicc.track(model="gpt-4o-mini",           # report usage the gateway can't see
+           tokens_in=1200, tokens_out=300)
+```
+
+Install (local, from this repo): `pip install -e packages/sdk-python`

aicc_sdk-0.1.0.dist-info/RECORD

@@ -0,0 +1,6 @@
+aicc/__init__.py,sha256=bkVj6uSCjHZSareDb60X7xTKYw1XGLSH60QXnL3nkTs,5532
+aicc_sdk-0.1.0.dist-info/licenses/LICENSE,sha256=nzNIbi_fwN6f21P-YoP0cLIbnvc3hdG79Api20dwqQU,1070
+aicc_sdk-0.1.0.dist-info/METADATA,sha256=OQ18GtPmymTNa3kPEeehwr6UmmsRRfXFrlBAjhbZGa4,1520
+aicc_sdk-0.1.0.dist-info/WHEEL,sha256=K260EYznzXsJYBQGqmI8VTxEdiZYNvDZwW9cBh9-_MA,91
+aicc_sdk-0.1.0.dist-info/top_level.txt,sha256=9nYQKRw37Ozu-MAr-Wsr25o0E6dlFbQPxbsTO5peW2s,5
+aicc_sdk-0.1.0.dist-info/RECORD,,

aicc_sdk-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (83.0.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

aicc_sdk-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Aditya Sarade
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

aicc_sdk-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+aicc