argosvix 0.4.0a1__tar.gz

argosvix-0.4.0a1/.gitignore

@@ -0,0 +1,10 @@
+__pycache__/
+*.py[cod]
+*.egg-info/
+.pytest_cache/
+.coverage
+htmlcov/
+dist/
+build/
+.venv/
+*.egg

argosvix-0.4.0a1/CHANGELOG.md

@@ -0,0 +1,46 @@
+# Changelog
+
+All notable changes to `argosvix` (Python SDK) are documented in this file.
+
+The format follows [Keep a Changelog](https://keepachangelog.com/en/1.1.0/) and the project adheres to [PEP 440](https://peps.python.org/pep-0440/) versioning.
+
+## [0.4.0a1] - 2026-05-29
+
+### Added
+- **Async wrap for Gemini and Mistral**: complete the async matrix for all four providers.
+  - Gemini = `client.aio.models.generate_content` (google-genai) is now detected and wrapped alongside the sync API. A single `wrap()` call covers both code paths.
+  - Mistral = `client.chat.complete_async` (mistralai 1.x) is wrapped in addition to `chat.complete`. The same Mistral client object exposes both, so both get patched.
+- Provider-table in README updated to reflect full async support; Phase 5 backlog now lists streaming + Responses API only.
+
+## [0.3.0a1] - 2026-05-29
+
+### Added
+- **Async wrap for OpenAI and Anthropic**: `inspect.iscoroutinefunction` detection auto-dispatches to async or sync wrapper. AsyncOpenAI / AsyncAnthropic clients are wrapped seamlessly.
+- Stream-mode safety check on async path: `stream=True` calls warn once per provider and skip recording (instead of silently posting 0-token records).
+- `__version__` and `User-Agent` header bumped accordingly.
+
+### Fixed
+- Dashboard MCP onboarding hint clarifies that some MCP clients prefix tool names with `argosvix::` instead of hardcoding `argosvix__` (cross-client convention difference).
+
+## [0.2.0a1] - 2026-05-29
+
+### Added
+- **Phase 2 providers**: Anthropic, Google Gemini, and Mistral sync wrap.
+  - Anthropic = `client.messages.create`
+  - Gemini = `client.models.generate_content` (google-genai), with fallback to legacy `GenerativeModel.generate_content`
+  - Mistral = `client.chat.complete`
+- Shape-based provider detection fallback for adapter or proxy clients whose constructor name doesn't include the provider keyword.
+
+### Fixed
+- `wrap()` is now genuinely idempotent: each `_wrap_<provider>` returns a boolean indicating whether monkeypatching succeeded, and the wrap marker is only set on success. Previous behavior could mark a client as wrapped even after a no-op fall-through, blocking a re-wrap attempt.
+- All providers warn once and skip recording when called with `stream=True` (previously silent 0-token records).
+
+## [0.1.0a1] - 2026-05-29
+
+### Added
+- Initial Phase 1 alpha release.
+- OpenAI sync `client.chat.completions.create` wrap.
+- Buffered ingest to `https://ingest.argosvix.com/v1/ingest` with periodic flush, retry on 5xx, and `atexit` flush for short-lived processes.
+- Pricing table mirroring `@argosvix/sdk` (JS) across all four providers.
+- `LlmCallRecord` dataclass with camelCase wire-format serialization that matches the backend ingest contract.
+- 27 pytest cases covering pricing, type serialization, and wrap behavior.

argosvix-0.4.0a1/LICENSE

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

argosvix-0.4.0a1/PKG-INFO

@@ -0,0 +1,224 @@
+Metadata-Version: 2.4
+Name: argosvix
+Version: 0.4.0a1
+Summary: Argosvix Python SDK = AI agent observability (cost / latency / tokens / errors) for OpenAI / Anthropic / Gemini / Mistral
+Project-URL: Homepage, https://argosvix.com
+Project-URL: Documentation, https://github.com/argosvix/Argosvix/tree/main/packages/sdk-python
+Project-URL: Repository, https://github.com/argosvix/Argosvix
+Project-URL: Issues, https://github.com/argosvix/Argosvix/issues
+Author-email: Yuto Makihara <hello@argosvix.com>
+License: MIT License
+        
+        Copyright (c) 2026 Yuto Makihara (Argosvix)
+        
+        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.
+License-File: LICENSE
+Keywords: ai-agent,anthropic,gemini,llm,mistral,observability,openai
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+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
+Classifier: Topic :: System :: Monitoring
+Requires-Python: >=3.10
+Provides-Extra: all
+Requires-Dist: anthropic>=0.30; extra == 'all'
+Requires-Dist: google-genai>=0.3; extra == 'all'
+Requires-Dist: mistralai>=1.0; extra == 'all'
+Requires-Dist: openai>=1.0; extra == 'all'
+Provides-Extra: anthropic
+Requires-Dist: anthropic>=0.30; extra == 'anthropic'
+Provides-Extra: dev
+Requires-Dist: anthropic>=0.30; extra == 'dev'
+Requires-Dist: google-genai>=0.3; extra == 'dev'
+Requires-Dist: mistralai>=1.0; extra == 'dev'
+Requires-Dist: openai>=1.0; extra == 'dev'
+Requires-Dist: pytest-cov>=5.0; extra == 'dev'
+Requires-Dist: pytest>=8.0; extra == 'dev'
+Requires-Dist: ruff>=0.6; extra == 'dev'
+Provides-Extra: gemini
+Requires-Dist: google-genai>=0.3; extra == 'gemini'
+Provides-Extra: mistral
+Requires-Dist: mistralai>=1.0; extra == 'mistral'
+Provides-Extra: openai
+Requires-Dist: openai>=1.0; extra == 'openai'
+Provides-Extra: test
+Requires-Dist: pytest-cov>=5.0; extra == 'test'
+Requires-Dist: pytest>=8.0; extra == 'test'
+Description-Content-Type: text/markdown
+
+# Argosvix Python SDK
+
+AI agent observability (cost / latency / tokens / errors) for OpenAI / Anthropic / Gemini / Mistral. **Phase 4 alpha — sync + async wrap for all 4 providers.** Streaming + Responses API are Phase 5 backlog.
+
+[![PyPI version](https://img.shields.io/pypi/v/argosvix)](https://pypi.org/project/argosvix/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+
+## Install
+
+```bash
+pip install argosvix
+# OR include a specific provider SDK as extra
+pip install "argosvix[openai]"
+pip install "argosvix[anthropic]"
+pip install "argosvix[gemini]"
+pip install "argosvix[mistral]"
+# all 4 at once
+pip install "argosvix[all]"
+```
+
+## Quickstart
+
+```python
+from openai import OpenAI
+from argosvix import wrap, ArgosvixConfig
+
+client = wrap(
+    OpenAI(),
+    ArgosvixConfig(
+        api_key="argosvix_live_...",  # get from https://dashboard.argosvix.com/api-keys
+        tags={"service": "my-app", "env": "prod"},
+    ),
+)
+
+resp = client.chat.completions.create(
+    model="gpt-5.5",
+    messages=[{"role": "user", "content": "Hello"}],
+)
+# The call is automatically recorded (cost / tokens / latency / model) and
+# batched to https://ingest.argosvix.com/v1/ingest within 5 seconds.
+```
+
+Visit https://dashboard.argosvix.com after a few seconds to see the call appear.
+
+## Configuration
+
+`ArgosvixConfig` accepts:
+
+| Field | Default | Description |
+|---|---|---|
+| `api_key` | `None` | Argosvix API key. **Required** for record submission. |
+| `endpoint` | `https://ingest.argosvix.com/v1/ingest` | Ingest endpoint. |
+| `tags` | `{}` | Tags attached to every record (e.g. `{"service": "bot"}`). |
+| `disabled` | `False` | Disable record submission entirely (e.g. local dev). |
+| `flush_interval_ms` | `5000` | Buffer flush interval. |
+| `buffer_max_size` | `100` | Max records before auto-flush. |
+| `flush_retry_attempts` | `2` | Total retry attempts including the initial try. |
+| `provider` | `None` | Explicit provider override (`"openai"` / etc). Auto-detected from client class name. |
+| `trace_id` | `None` | OTel-subset trace ID. Attached to all records from this client. |
+| `span_id` | `None` | OTel-subset span ID. |
+| `parent_span_id` | `None` | OTel-subset parent span ID. |
+
+## Short-lived processes (Lambda / Cron / CLI)
+
+The SDK auto-registers `atexit` to flush remaining records when the process exits. But for Lambda / Edge Functions / Workers-style short-lived runtimes where `atexit` may not fire, explicitly flush:
+
+```python
+from argosvix import get_recorder
+
+rec = get_recorder(client)
+if rec is not None:
+    rec.flush_blocking()  # blocks until all buffered records are POSTed
+```
+
+## Supported providers (Phase 4)
+
+| Provider | Sync | Async | Streaming | Notes |
+|---|---|---|---|---|
+| **OpenAI** | ✅ | ✅ | 🔄 P5 | `client.chat.completions.create` (sync + AsyncOpenAI) |
+| **Anthropic** | ✅ | ✅ | 🔄 P5 | `client.messages.create` (sync + AsyncAnthropic) |
+| **Google Gemini** | ✅ | ✅ | 🔄 P5 | `client.models.generate_content` + `client.aio.models.generate_content` (google-genai) |
+| **Mistral** | ✅ | ✅ | 🔄 P5 | `client.chat.complete` + `client.chat.complete_async` (mistralai 1.x) |
+
+OpenAI Responses API support is also Phase 3 backlog. Need a provider sooner? File an issue at https://github.com/argosvix/Argosvix/issues.
+
+### Multi-provider example
+
+```python
+from openai import OpenAI
+from anthropic import Anthropic
+from google import genai
+from mistralai import Mistral
+from argosvix import wrap, ArgosvixConfig
+
+cfg = ArgosvixConfig(api_key="argosvix_live_...", tags={"app": "comparison-bot"})
+oa = wrap(OpenAI(), cfg)
+an = wrap(Anthropic(), cfg)
+gm = wrap(genai.Client(), cfg)
+ms = wrap(Mistral(api_key="..."), cfg)
+
+# All calls are recorded to the same Argosvix account, distinguishable by provider.
+oa.chat.completions.create(model="gpt-5.5", messages=[{"role": "user", "content": "Hi"}])
+an.messages.create(model="claude-opus-4", messages=[{"role": "user", "content": "Hi"}], max_tokens=512)
+gm.models.generate_content(model="gemini-2.5-flash", contents="Hi")
+ms.chat.complete(model="mistral-large-latest", messages=[{"role": "user", "content": "Hi"}])
+```
+
+## Trace correlation
+
+Group related LLM calls into a single trace by passing `trace_id` to `wrap()`:
+
+```python
+import uuid
+
+trace_id = uuid.uuid4().hex
+client = wrap(
+    OpenAI(),
+    ArgosvixConfig(api_key="...", trace_id=trace_id),
+)
+# All calls from this client share trace_id and appear together in the
+# dashboard's traces waterfall view.
+```
+
+## Privacy
+
+The SDK records **metadata only** (= tokens, cost, latency, model name, error info, your tags). **Prompts and completions are NOT recorded** by default. Opt-in plain-text storage (with PII redaction + AES-256 encryption + 7-30 day retention + 1-click delete) is planned for v1.5 — see https://argosvix.com/privacy for details.
+
+## Pricing table
+
+`PRICING` is a snapshot updated quarterly from each provider's official pricing page. Unknown models return `0.0` cost + a warning. To verify a model is known:
+
+```python
+from argosvix import calculate_cost
+
+cost = calculate_cost("openai", "gpt-5.5", prompt_tokens=1000, completion_tokens=500)
+print(cost)  # 0.0125 USD
+```
+
+## Development
+
+```bash
+# install with dev deps
+pip install -e ".[dev]"
+
+# run tests
+pytest
+
+# lint
+ruff check argosvix tests
+```
+
+## License
+
+MIT © Yuto Makihara (Argosvix). See [LICENSE](LICENSE).

argosvix-0.4.0a1/README.md

@@ -0,0 +1,154 @@
+# Argosvix Python SDK
+
+AI agent observability (cost / latency / tokens / errors) for OpenAI / Anthropic / Gemini / Mistral. **Phase 4 alpha — sync + async wrap for all 4 providers.** Streaming + Responses API are Phase 5 backlog.
+
+[![PyPI version](https://img.shields.io/pypi/v/argosvix)](https://pypi.org/project/argosvix/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+
+## Install
+
+```bash
+pip install argosvix
+# OR include a specific provider SDK as extra
+pip install "argosvix[openai]"
+pip install "argosvix[anthropic]"
+pip install "argosvix[gemini]"
+pip install "argosvix[mistral]"
+# all 4 at once
+pip install "argosvix[all]"
+```
+
+## Quickstart
+
+```python
+from openai import OpenAI
+from argosvix import wrap, ArgosvixConfig
+
+client = wrap(
+    OpenAI(),
+    ArgosvixConfig(
+        api_key="argosvix_live_...",  # get from https://dashboard.argosvix.com/api-keys
+        tags={"service": "my-app", "env": "prod"},
+    ),
+)
+
+resp = client.chat.completions.create(
+    model="gpt-5.5",
+    messages=[{"role": "user", "content": "Hello"}],
+)
+# The call is automatically recorded (cost / tokens / latency / model) and
+# batched to https://ingest.argosvix.com/v1/ingest within 5 seconds.
+```
+
+Visit https://dashboard.argosvix.com after a few seconds to see the call appear.
+
+## Configuration
+
+`ArgosvixConfig` accepts:
+
+| Field | Default | Description |
+|---|---|---|
+| `api_key` | `None` | Argosvix API key. **Required** for record submission. |
+| `endpoint` | `https://ingest.argosvix.com/v1/ingest` | Ingest endpoint. |
+| `tags` | `{}` | Tags attached to every record (e.g. `{"service": "bot"}`). |
+| `disabled` | `False` | Disable record submission entirely (e.g. local dev). |
+| `flush_interval_ms` | `5000` | Buffer flush interval. |
+| `buffer_max_size` | `100` | Max records before auto-flush. |
+| `flush_retry_attempts` | `2` | Total retry attempts including the initial try. |
+| `provider` | `None` | Explicit provider override (`"openai"` / etc). Auto-detected from client class name. |
+| `trace_id` | `None` | OTel-subset trace ID. Attached to all records from this client. |
+| `span_id` | `None` | OTel-subset span ID. |
+| `parent_span_id` | `None` | OTel-subset parent span ID. |
+
+## Short-lived processes (Lambda / Cron / CLI)
+
+The SDK auto-registers `atexit` to flush remaining records when the process exits. But for Lambda / Edge Functions / Workers-style short-lived runtimes where `atexit` may not fire, explicitly flush:
+
+```python
+from argosvix import get_recorder
+
+rec = get_recorder(client)
+if rec is not None:
+    rec.flush_blocking()  # blocks until all buffered records are POSTed
+```
+
+## Supported providers (Phase 4)
+
+| Provider | Sync | Async | Streaming | Notes |
+|---|---|---|---|---|
+| **OpenAI** | ✅ | ✅ | 🔄 P5 | `client.chat.completions.create` (sync + AsyncOpenAI) |
+| **Anthropic** | ✅ | ✅ | 🔄 P5 | `client.messages.create` (sync + AsyncAnthropic) |
+| **Google Gemini** | ✅ | ✅ | 🔄 P5 | `client.models.generate_content` + `client.aio.models.generate_content` (google-genai) |
+| **Mistral** | ✅ | ✅ | 🔄 P5 | `client.chat.complete` + `client.chat.complete_async` (mistralai 1.x) |
+
+OpenAI Responses API support is also Phase 3 backlog. Need a provider sooner? File an issue at https://github.com/argosvix/Argosvix/issues.
+
+### Multi-provider example
+
+```python
+from openai import OpenAI
+from anthropic import Anthropic
+from google import genai
+from mistralai import Mistral
+from argosvix import wrap, ArgosvixConfig
+
+cfg = ArgosvixConfig(api_key="argosvix_live_...", tags={"app": "comparison-bot"})
+oa = wrap(OpenAI(), cfg)
+an = wrap(Anthropic(), cfg)
+gm = wrap(genai.Client(), cfg)
+ms = wrap(Mistral(api_key="..."), cfg)
+
+# All calls are recorded to the same Argosvix account, distinguishable by provider.
+oa.chat.completions.create(model="gpt-5.5", messages=[{"role": "user", "content": "Hi"}])
+an.messages.create(model="claude-opus-4", messages=[{"role": "user", "content": "Hi"}], max_tokens=512)
+gm.models.generate_content(model="gemini-2.5-flash", contents="Hi")
+ms.chat.complete(model="mistral-large-latest", messages=[{"role": "user", "content": "Hi"}])
+```
+
+## Trace correlation
+
+Group related LLM calls into a single trace by passing `trace_id` to `wrap()`:
+
+```python
+import uuid
+
+trace_id = uuid.uuid4().hex
+client = wrap(
+    OpenAI(),
+    ArgosvixConfig(api_key="...", trace_id=trace_id),
+)
+# All calls from this client share trace_id and appear together in the
+# dashboard's traces waterfall view.
+```
+
+## Privacy
+
+The SDK records **metadata only** (= tokens, cost, latency, model name, error info, your tags). **Prompts and completions are NOT recorded** by default. Opt-in plain-text storage (with PII redaction + AES-256 encryption + 7-30 day retention + 1-click delete) is planned for v1.5 — see https://argosvix.com/privacy for details.
+
+## Pricing table
+
+`PRICING` is a snapshot updated quarterly from each provider's official pricing page. Unknown models return `0.0` cost + a warning. To verify a model is known:
+
+```python
+from argosvix import calculate_cost
+
+cost = calculate_cost("openai", "gpt-5.5", prompt_tokens=1000, completion_tokens=500)
+print(cost)  # 0.0125 USD
+```
+
+## Development
+
+```bash
+# install with dev deps
+pip install -e ".[dev]"
+
+# run tests
+pytest
+
+# lint
+ruff check argosvix tests
+```
+
+## License
+
+MIT © Yuto Makihara (Argosvix). See [LICENSE](LICENSE).

argosvix-0.4.0a1/argosvix/__init__.py

@@ -0,0 +1,46 @@
+"""Argosvix Python SDK = AI agent observability for OpenAI / Anthropic / Gemini / Mistral.
+
+Phase 4 (= 2026-05): sync + async wrap for all 4 providers (OpenAI / Anthropic / Gemini / Mistral)。
+streaming + Responses API は Phase 5 backlog。
+
+Quickstart:
+    from openai import OpenAI
+    from argosvix import wrap, ArgosvixConfig
+
+    client = wrap(
+        OpenAI(),
+        ArgosvixConfig(
+            api_key="argosvix_live_...",
+            tags={"service": "my-app", "env": "prod"},
+        ),
+    )
+    resp = client.chat.completions.create(
+        model="gpt-5.5",
+        messages=[{"role": "user", "content": "Hello"}],
+    )
+    # The call is automatically recorded (cost / tokens / latency / model) and
+    # batched to https://ingest.argosvix.com/v1/ingest within 5 seconds.
+
+Visit https://argosvix.com for dashboards + alerts.
+"""
+
+from argosvix.client import get_recorder, wrap
+from argosvix.pricing import PRICING, calculate_cost
+from argosvix.recorder import Recorder
+from argosvix.types import ArgosvixConfig, ErrorDetails, LlmCallRecord, Provider, RequestMeta
+
+__version__ = "0.4.0a1"
+
+__all__ = [
+    "ArgosvixConfig",
+    "ErrorDetails",
+    "LlmCallRecord",
+    "PRICING",
+    "Provider",
+    "Recorder",
+    "RequestMeta",
+    "__version__",
+    "calculate_cost",
+    "get_recorder",
+    "wrap",
+]