proxyagent 0.2.0__tar.gz → 0.2.1__tar.gz

{proxyagent-0.2.0 → proxyagent-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: proxyagent
-Version: 0.2.0
+Version: 0.2.1
 Summary: Run any agent (Claude, Codex, custom) on any machine — with no API key on the machine. A secure, self-hosted proxy for models and tools.
 Project-URL: Homepage, https://github.com/teddyoweh/proxyagent
 Author-email: Spawn Labs <teddy@spawnlabs.ai>
@@ -61,6 +61,15 @@ proxy and use the **machine token** as the "api key." The proxy authenticates th
 checks its scope, **swaps in the real key**, forwards upstream, and logs the call. The
 machine never sees a real credential.
 
+## Try it with zero keys (local)
+```bash
+pip install proxyagent && proxyagent serve        # prints an admin token
+proxyagent token new local --admin pa_admin_…     # mint a token
+# call the built-in `mock` model — full pipeline (auth, scope, usage, cost, log), no real key:
+curl -s localhost:8080/anthropic/v1/messages -H "x-api-key: pa_…" \
+  -d '{"model":"mock","max_tokens":50,"messages":[{"role":"user","content":"hi"}]}'
+```
+
 ## Quickstart
 
 **1. Run the proxy** (on a box you control — it holds the real keys):

{proxyagent-0.2.0 → proxyagent-0.2.1}/README.md

@@ -29,6 +29,15 @@ proxy and use the **machine token** as the "api key." The proxy authenticates th
 checks its scope, **swaps in the real key**, forwards upstream, and logs the call. The
 machine never sees a real credential.
 
+## Try it with zero keys (local)
+```bash
+pip install proxyagent && proxyagent serve        # prints an admin token
+proxyagent token new local --admin pa_admin_…     # mint a token
+# call the built-in `mock` model — full pipeline (auth, scope, usage, cost, log), no real key:
+curl -s localhost:8080/anthropic/v1/messages -H "x-api-key: pa_…" \
+  -d '{"model":"mock","max_tokens":50,"messages":[{"role":"user","content":"hi"}]}'
+```
+
 ## Quickstart
 
 **1. Run the proxy** (on a box you control — it holds the real keys):

{proxyagent-0.2.0 → proxyagent-0.2.1}/proxyagent/__init__.py

@@ -16,7 +16,7 @@ from typing import Optional
 
 from .harness import run  # noqa: F401  (the headline SDK call)
 
-__version__ = "0.2.0"
+__version__ = "0.2.1"
 __all__ = ["run", "serve", "create_app", "Config", "Admin", "__version__"]
 
 

{proxyagent-0.2.0 → proxyagent-0.2.1}/proxyagent/pricing.py

@@ -27,6 +27,8 @@ DEFAULT_PRICES: dict[str, tuple[float, float]] = {
     "o3-mini": (1.10, 4.40),
     "o3": (2.0, 8.0),
     "gpt-5": (1.25, 10.0),
+    # offline test provider — free
+    "mock": (0.0, 0.0),
 }
 
 

{proxyagent-0.2.0 → proxyagent-0.2.1}/proxyagent/providers.py

@@ -60,6 +60,23 @@ async def forward(
 ):
     """Forward a request upstream. Returns (status, headers, body_iter_or_dict, log_after)."""
     provider = PROVIDERS[provider_name]
+    model = body.get("model", "")
+    t0 = now_ms()
+
+    # Offline mock — exercise the full pipeline (auth, scope, log, cost) with NO real
+    # key. Use model "mock" (or "mock-…") anywhere a real model would go.
+    if model.startswith("mock"):
+        payload, (ptok, ctok) = _mock_payload(provider_name, body)
+        store.log_request(
+            token_id=token["id"], token_label=token.get("label"), provider=provider_name,
+            model=model, status=200, prompt_tokens=ptok, completion_tokens=ctok,
+            latency_ms=now_ms() - t0, streamed=1 if streaming else 0,
+            tools_used=json.dumps(tools_used or []), cost_usd=pricing.cost_usd(model, ptok, ctok),
+            error=None)
+        if streaming:
+            return 200, {"content-type": "text/event-stream"}, _mock_stream(provider_name, payload), None
+        return 200, {"content-type": "application/json"}, payload, None
+
     auth, ok = resolve_auth(provider, store)
     if not ok:
         return 502, {}, {"error": f"provider '{provider_name}' not configured on the proxy "
@@ -67,8 +84,6 @@ async def forward(
 
     url = provider.base_url + upstream_path
     headers = {"content-type": "application/json", **auth}
-    model = body.get("model", "")
-    t0 = now_ms()
 
     def _log(status, ptok, ctok, err=None):
         store.log_request(
@@ -115,3 +130,51 @@ async def forward(
     ptok, ctok = _extract_usage(provider_name, payload if isinstance(payload, dict) else {})
     _log(resp.status_code, ptok, ctok, None if resp.is_success else str(payload)[:300])
     return resp.status_code, {"content-type": "application/json"}, payload, None
+
+
+# ------------------------------------------------------------------ #
+# Offline mock — provider-shaped canned responses for local testing.
+# ------------------------------------------------------------------ #
+
+def _last_user_text(body: dict) -> str:
+    for m in reversed(body.get("messages", [])):
+        if m.get("role") == "user":
+            c = m.get("content")
+            if isinstance(c, str):
+                return c
+            if isinstance(c, list):
+                return " ".join(p.get("text", "") for p in c if isinstance(p, dict))
+    return ""
+
+
+def _mock_payload(provider: str, body: dict):
+    prompt = _last_user_text(body)[:200]
+    text = f"[proxyagent mock] received: {prompt!r}. No real key used — the pipeline works."
+    ptok, ctok = max(1, len(prompt) // 4), max(1, len(text) // 4)
+    if provider == "anthropic":
+        return ({
+            "id": "msg_mock", "type": "message", "role": "assistant", "model": body.get("model"),
+            "content": [{"type": "text", "text": text}], "stop_reason": "end_turn",
+            "usage": {"input_tokens": ptok, "output_tokens": ctok},
+        }, (ptok, ctok))
+    return ({
+        "id": "chatcmpl-mock", "object": "chat.completion", "model": body.get("model"),
+        "choices": [{"index": 0, "message": {"role": "assistant", "content": text},
+                     "finish_reason": "stop"}],
+        "usage": {"prompt_tokens": ptok, "completion_tokens": ctok, "total_tokens": ptok + ctok},
+    }, (ptok, ctok))
+
+
+async def _mock_stream(provider: str, payload: dict):
+    import json as _j
+    if provider == "anthropic":
+        text = payload["content"][0]["text"]
+        yield f"event: message_start\ndata: {_j.dumps({'type':'message_start','message':payload})}\n\n".encode()
+        yield (f"event: content_block_delta\ndata: "
+               f"{_j.dumps({'type':'content_block_delta','delta':{'type':'text_delta','text':text}})}\n\n").encode()
+        yield b"event: message_stop\ndata: {\"type\":\"message_stop\"}\n\n"
+    else:
+        text = payload["choices"][0]["message"]["content"]
+        chunk = {"choices": [{"delta": {"content": text}, "index": 0}]}
+        yield f"data: {_j.dumps(chunk)}\n\n".encode()
+        yield b"data: [DONE]\n\n"

{proxyagent-0.2.0 → proxyagent-0.2.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "proxyagent"
-version = "0.2.0"
+version = "0.2.1"
 description = "Run any agent (Claude, Codex, custom) on any machine — with no API key on the machine. A secure, self-hosted proxy for models and tools."
 readme = "README.md"
 requires-python = ">=3.10"

{proxyagent-0.2.0 → proxyagent-0.2.1}/tests/test_proxy.py

@@ -98,6 +98,25 @@ def test_credential_storage_and_resolution():
     assert s.remove_credential(cid)
 
 
+def test_mock_provider_offline():
+    """Full pipeline with no real key: mint → call model 'mock' → response + usage + log."""
+    c = _client()
+    tok = c.post("/admin/tokens", headers=ADMIN, json={"label": "m", "scope": ["*"]}).json()["token"]
+    r = c.post("/anthropic/v1/messages", headers={"x-api-key": tok},
+               json={"model": "mock", "max_tokens": 50, "messages": [{"role": "user", "content": "hello"}]})
+    assert r.status_code == 200
+    body = r.json()
+    assert body["content"][0]["text"].startswith("[proxyagent mock]")
+    assert body["usage"]["input_tokens"] >= 1
+    # it was logged (with $0 cost)
+    logs = c.get("/admin/logs", headers=ADMIN).json()["logs"]
+    assert logs[0]["model"] == "mock" and logs[0]["status"] == 200
+    # openai shape too
+    r2 = c.post("/openai/v1/chat/completions", headers={"authorization": f"Bearer {tok}"},
+                json={"model": "mock", "messages": [{"role": "user", "content": "hi"}]})
+    assert r2.json()["choices"][0]["message"]["content"].startswith("[proxyagent mock]")
+
+
 def test_provider_admin_endpoints():
     c = _client()
     r = c.post("/admin/providers", headers=ADMIN, json={"provider": "anthropic", "secret": "sk-ant-x"})