probiai 0.1.0__tar.gz

probiai-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 ProbiAI
+
+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.

probiai-0.1.0/PKG-INFO

@@ -0,0 +1,69 @@
+Metadata-Version: 2.5
+Name: probiai
+Version: 0.1.0
+Summary: Sends ProbiAI a copy of each action your agent takes. Report-only, no dependencies.
+Project-URL: Homepage, https://probiai-personalagent.vercel.app
+License-Expression: MIT
+License-File: LICENSE
+Keywords: agents,claude-agent-sdk,openai-agents,probiai,telemetry
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+
+# probiai
+
+Sends ProbiAI a copy of each action your agent takes, so your ProbiAI page shows what it did. For agents you build
+with the Claude Agent SDK (Python or TypeScript) or the OpenAI Agents SDK (Python).
+
+- Report-only: it never blocks, delays or changes your agent. If ProbiAI can't be reached, the copy is dropped.
+- No dependencies.
+- Contents and secrets in tool arguments (message bodies, file contents, passwords, card numbers, API keys) are
+  dropped before anything leaves your machine.
+
+## Set up
+
+1. On your ProbiAI page, choose your framework. It shows your key.
+2. Put the key in your agent's environment, not in your code:
+
+   ```bash
+   export PROBIAI_KEY=pk....
+   ```
+
+3. Pass the hooks to your agent (merge them with any hooks of your own).
+
+### Python
+
+```bash
+pip install probiai
+```
+
+```python
+import probiai
+
+# Claude Agent SDK
+options = ClaudeAgentOptions(hooks=probiai.claude_hooks())
+
+# OpenAI Agents SDK
+result = await Runner.run(agent, prompt, hooks=probiai.OpenAIHooks())
+```
+
+### TypeScript (Claude Agent SDK)
+
+```bash
+npm install probiai
+```
+
+```ts
+import { claudeHooks } from "probiai";
+
+const options = { hooks: claudeHooks() };
+```
+
+## What is sent
+
+For each tool call: the tool name, its arguments without contents or secrets, whether it worked, the first 500
+characters of the result, and the model when known. Sub-agents and hand-offs are sent by name. Copies go to
+`PROBIAI_URL` (default https://probiai-personalagent.vercel.app) with your key. Without a key, nothing is sent.
+
+Using OpenClaw or Pi? Your ProbiAI page has a one-line install with your key built in.
+
+ProbiAI is a pilot: it is being tested and is not yet an insurance product.

probiai-0.1.0/README.md

@@ -0,0 +1,58 @@
+# probiai
+
+Sends ProbiAI a copy of each action your agent takes, so your ProbiAI page shows what it did. For agents you build
+with the Claude Agent SDK (Python or TypeScript) or the OpenAI Agents SDK (Python).
+
+- Report-only: it never blocks, delays or changes your agent. If ProbiAI can't be reached, the copy is dropped.
+- No dependencies.
+- Contents and secrets in tool arguments (message bodies, file contents, passwords, card numbers, API keys) are
+  dropped before anything leaves your machine.
+
+## Set up
+
+1. On your ProbiAI page, choose your framework. It shows your key.
+2. Put the key in your agent's environment, not in your code:
+
+   ```bash
+   export PROBIAI_KEY=pk....
+   ```
+
+3. Pass the hooks to your agent (merge them with any hooks of your own).
+
+### Python
+
+```bash
+pip install probiai
+```
+
+```python
+import probiai
+
+# Claude Agent SDK
+options = ClaudeAgentOptions(hooks=probiai.claude_hooks())
+
+# OpenAI Agents SDK
+result = await Runner.run(agent, prompt, hooks=probiai.OpenAIHooks())
+```
+
+### TypeScript (Claude Agent SDK)
+
+```bash
+npm install probiai
+```
+
+```ts
+import { claudeHooks } from "probiai";
+
+const options = { hooks: claudeHooks() };
+```
+
+## What is sent
+
+For each tool call: the tool name, its arguments without contents or secrets, whether it worked, the first 500
+characters of the result, and the model when known. Sub-agents and hand-offs are sent by name. Copies go to
+`PROBIAI_URL` (default https://probiai-personalagent.vercel.app) with your key. Without a key, nothing is sent.
+
+Using OpenClaw or Pi? Your ProbiAI page has a one-line install with your key built in.
+
+ProbiAI is a pilot: it is being tested and is not yet an insurance product.

probiai-0.1.0/probiai.py

@@ -0,0 +1,117 @@
+"""ProbiAI add-on for agents you build in Python: a copy of each action your agent takes goes to ProbiAI automatically.
+
+    Claude Agent SDK:    options = ClaudeAgentOptions(hooks=probiai.claude_hooks())   # merge with hooks of your own
+    OpenAI Agents SDK:   result = await Runner.run(agent, prompt, hooks=probiai.OpenAIHooks())
+
+Set PROBIAI_KEY (from your ProbiAI page). Standard library only. Report-only: it never blocks, delays or changes your
+agent. Copies go out on a background thread (your program waits at most 5 seconds for the last ones when it exits);
+failures are ignored. Contents and secrets in tool arguments are dropped before anything leaves your machine.
+"""
+from __future__ import annotations
+
+import json
+import os
+import re
+import threading
+import urllib.request
+from datetime import datetime, timezone
+
+URL = os.environ.get("PROBIAI_URL", "https://probiai-personalagent.vercel.app").rstrip("/")
+DROP = re.compile(r"^(body|text|content|contents|message|html|markdown|attachments?|file_?content|data|base64|password|"
+                  r"passcode|pin|otp|token|secret|api_?key|card(_?number)?|cvv|cvc|ssn)$", re.I)
+
+
+def facts(value, depth: int = 0):
+    """The arguments with contents and secrets removed, nested at most four levels, long strings cut."""
+    if depth > 3:
+        return "…"
+    if isinstance(value, (list, tuple)):
+        return [facts(v, depth + 1) for v in list(value)[:50]]
+    if isinstance(value, dict):
+        return {k: facts(v, depth + 1) for k, v in list(value.items())[:40] if not DROP.match(str(k))}
+    return value[:200] if isinstance(value, str) else value
+
+
+def short(result) -> str | None:
+    """A tool result, short, so ProbiAI can read an amount or a booking reference from it."""
+    if result is None:
+        return None
+    try:
+        text = result if isinstance(result, str) else json.dumps(result, default=str)
+    except (TypeError, ValueError):
+        text = str(result)
+    return text[:500]
+
+
+def send(copy: dict, key: str | None = None) -> None:
+    """Send one copy in the background. Does nothing without a key."""
+    key = key or os.environ.get("PROBIAI_KEY", "")
+    if key:
+        body = json.dumps({"v": 1, "ts": datetime.now(timezone.utc).isoformat(), **copy}, default=str).encode()
+        threading.Thread(target=_post, args=(body, key), name="probiai-copy").start()
+
+
+def _post(body: bytes, key: str) -> None:
+    req = urllib.request.Request(f"{URL}/v1/copies", data=body, method="POST",
+                                 headers={"content-type": "application/json", "authorization": f"Bearer {key}"})
+    try:
+        urllib.request.urlopen(req, timeout=5).close()
+    except Exception:   # noqa: BLE001 - a copy that can't be sent must never affect the agent
+        pass
+
+
+def claude_hooks(key: str | None = None, model: str | None = None) -> dict:
+    """Claude Agent SDK: every tool call, failed tool call and finished sub-agent."""
+    from claude_agent_sdk import HookMatcher
+
+    def copy(i: dict, status: str) -> dict:
+        return {"harness": "claude-agent-sdk", "model": model, "tool": i.get("tool_name"),
+                "input": facts(i.get("tool_input") or {}), "status": status, "result": short(i.get("tool_response"))}
+
+    async def done(i, tool_use_id, context):
+        send(copy(i, "ok"), key)
+        return {}
+
+    async def failed(i, tool_use_id, context):
+        send(copy(i, "error"), key)
+        return {}
+
+    async def subagent(i, tool_use_id, context):
+        send({"harness": "claude-agent-sdk", "model": model,
+              "handoff": {"name": i.get("agent_type") or "sub-agent", "via": "own sub-agent"}}, key)
+        return {}
+
+    return {"PostToolUse": [HookMatcher(hooks=[done])], "PostToolUseFailure": [HookMatcher(hooks=[failed])],
+            "SubagentStop": [HookMatcher(hooks=[subagent])]}
+
+
+try:
+    from agents import RunHooks as _RunHooks
+except ImportError:          # the OpenAI Agents SDK isn't installed: OpenAIHooks is never used then
+    _RunHooks = object
+
+
+class OpenAIHooks(_RunHooks):
+    """OpenAI Agents SDK: every tool call and every hand-off to another agent."""
+
+    def __init__(self, key: str | None = None):
+        super().__init__()
+        self.key = key
+
+    async def on_tool_end(self, context, agent, tool, result) -> None:
+        args = getattr(context, "tool_arguments", None)
+        try:
+            args = json.loads(args) if isinstance(args, str) else {}
+        except ValueError:
+            args = {}
+        send({"harness": "openai-agents", "model": _model(agent), "tool": getattr(context, "tool_name", None)
+              or getattr(tool, "name", "tool"), "input": facts(args), "status": "ok", "result": short(result)}, self.key)
+
+    async def on_handoff(self, context, from_agent, to_agent) -> None:
+        send({"harness": "openai-agents", "model": _model(from_agent),
+              "handoff": {"name": getattr(to_agent, "name", "agent"), "via": "handoff"}}, self.key)
+
+
+def _model(agent) -> str | None:
+    m = getattr(agent, "model", None)
+    return m if isinstance(m, str) else getattr(m, "model", None)

probiai-0.1.0/pyproject.toml

@@ -0,0 +1,19 @@
+[build-system]
+requires = ["hatchling>=1.27"]
+build-backend = "hatchling.build"
+
+[project]
+name = "probiai"
+version = "0.1.0"
+description = "Sends ProbiAI a copy of each action your agent takes. Report-only, no dependencies."
+readme = "README.md"
+license = "MIT"
+license-files = ["LICENSE"]
+requires-python = ">=3.9"
+keywords = ["claude-agent-sdk", "openai-agents", "agents", "telemetry", "probiai"]
+
+[project.urls]
+Homepage = "https://probiai-personalagent.vercel.app"
+
+[tool.hatch.build.targets.wheel]
+only-include = ["probiai.py"]