aitracer 0.1.0__py3-none-any.whl

aitracer-0.1.0.dist-info/METADATA

@@ -0,0 +1,234 @@
+Metadata-Version: 2.4
+Name: aitracer
+Version: 0.1.0
+Summary: AITracer SDK - Monitor your AI/LLM applications
+Project-URL: Homepage, https://aitracer.co
+Project-URL: Documentation, https://docs.aitracer.co
+Project-URL: Repository, https://github.com/haro-inc/aitracer
+Author-email: "HARO Inc." <info@haro.co.jp>
+License: Proprietary
+Keywords: ai,anthropic,llm,monitoring,observability,openai
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: Other/Proprietary License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.9
+Requires-Dist: anthropic>=0.18.0
+Requires-Dist: httpx>=0.25.0
+Requires-Dist: openai>=1.0.0
+Provides-Extra: dev
+Requires-Dist: mypy>=1.0.0; extra == 'dev'
+Requires-Dist: pytest-asyncio>=0.21.0; extra == 'dev'
+Requires-Dist: pytest-cov>=4.0.0; extra == 'dev'
+Requires-Dist: pytest>=7.0.0; extra == 'dev'
+Requires-Dist: ruff>=0.1.0; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# AITracer Python SDK
+
+AIエージェント/LLMアプリの実行ログ・コスト・パフォーマンスを監視するためのPython SDK。
+
+## インストール
+
+```bash
+pip install aitracer
+```
+
+## クイックスタート
+
+### 基本的な使い方
+
+```python
+from aitracer import AITracer
+from openai import OpenAI
+
+# AITracerを初期化
+tracer = AITracer(
+    api_key="at-xxxxxxxx",
+    project="my-chatbot"
+)
+
+# OpenAIクライアントをラップ
+client = tracer.wrap_openai(OpenAI())
+
+# 普通に使う（自動でログが送信される）
+response = client.chat.completions.create(
+    model="gpt-4o",
+    messages=[{"role": "user", "content": "Hello!"}]
+)
+```
+
+### Anthropic対応
+
+```python
+from aitracer import AITracer
+from anthropic import Anthropic
+
+tracer = AITracer(
+    api_key="at-xxxxxxxx",
+    project="my-chatbot"
+)
+
+# Anthropicクライアントをラップ
+client = tracer.wrap_anthropic(Anthropic())
+
+response = client.messages.create(
+    model="claude-3-5-sonnet-20241022",
+    max_tokens=1024,
+    messages=[{"role": "user", "content": "Hello!"}]
+)
+```
+
+### Google Gemini対応
+
+```python
+from aitracer import AITracer
+import google.generativeai as genai
+
+tracer = AITracer(
+    api_key="at-xxxxxxxx",
+    project="my-chatbot"
+)
+
+# Geminiを設定
+genai.configure(api_key="your-google-api-key")
+model = genai.GenerativeModel("gemini-1.5-flash")
+
+# Geminiモデルをラップ
+model = tracer.wrap_gemini(model)
+
+response = model.generate_content("Hello!")
+print(response.text)
+```
+
+ストリーミングも対応しています：
+
+```python
+response = model.generate_content("Write a story...", stream=True)
+for chunk in response:
+    print(chunk.text, end="")
+```
+
+### トレース機能
+
+複数のAPI呼び出しをグループ化して追跡できます。
+
+```python
+with tracer.trace("user-query-123") as trace:
+    # この中のAPI呼び出しは同じtrace_idでグループ化される
+    response1 = client.chat.completions.create(
+        model="gpt-4o",
+        messages=[{"role": "user", "content": "Summarize this..."}]
+    )
+
+    response2 = client.chat.completions.create(
+        model="gpt-4o",
+        messages=[{"role": "user", "content": "Translate to Japanese..."}]
+    )
+
+    # メタデータを追加
+    trace.set_metadata({
+        "user_id": "user-456",
+        "feature": "summarization"
+    })
+
+    # タグを追加
+    trace.add_tag("production")
+```
+
+### ストリーミング対応
+
+ストリーミングレスポンスも自動的にログされます。
+
+```python
+stream = client.chat.completions.create(
+    model="gpt-4o",
+    messages=[{"role": "user", "content": "Write a story..."}],
+    stream=True
+)
+
+for chunk in stream:
+    print(chunk.choices[0].delta.content or "", end="")
+
+# ストリーム完了後に自動的にログが送信される
+```
+
+## 設定オプション
+
+```python
+tracer = AITracer(
+    # 必須
+    api_key="at-xxxxxxxx",      # または AITRACER_API_KEY 環境変数
+    project="my-chatbot",        # または AITRACER_PROJECT 環境変数
+
+    # オプション
+    base_url="https://api.aitracer.co",  # APIエンドポイント
+    sync=False,                  # True: 同期送信（Lambda用）
+    flush_on_exit=True,          # 終了時に未送信ログをフラッシュ
+    batch_size=10,               # バッチサイズ
+    flush_interval=5.0,          # 自動フラッシュ間隔（秒）
+    enabled=True,                # False: ログ無効化（テスト用）
+)
+```
+
+### Lambda/サーバーレス環境
+
+サーバーレス環境では `sync=True` を使用してください。
+
+```python
+tracer = AITracer(
+    api_key="at-xxxxxxxx",
+    project="my-lambda",
+    sync=True  # 同期送信
+)
+```
+
+### 環境変数
+
+```bash
+export AITRACER_API_KEY=at-xxxxxxxx
+export AITRACER_PROJECT=my-chatbot
+```
+
+```python
+# 環境変数から自動で読み込み
+tracer = AITracer()
+```
+
+## 手動ログ
+
+自動ラッパーを使わずに手動でログを送信することもできます。
+
+```python
+tracer.log(
+    model="gpt-4o",
+    provider="openai",
+    input_data={"messages": [{"role": "user", "content": "Hello"}]},
+    output_data={"content": "Hi there!"},
+    input_tokens=10,
+    output_tokens=5,
+    latency_ms=150,
+    status="success",
+    metadata={"user_id": "user-123"},
+    tags=["production"]
+)
+```
+
+## フラッシュとシャットダウン
+
+```python
+# 未送信のログを即座に送信
+tracer.flush()
+
+# シャットダウン（フラッシュ + リソース解放）
+tracer.shutdown()
+```
+
+## ライセンス
+
+Copyright (c) 株式会社HARO. All rights reserved.

aitracer-0.1.0.dist-info/RECORD

@@ -0,0 +1,15 @@
+python/__init__.py,sha256=ufrrGDz44Nv6tCjZqAyvCpN3Sk843rDu5F7z_OmiELg,866
+python/client.py,sha256=yvH6f_RpTo9Bl0VXubuvNxDHVE1i_KafVBjKQ0fFOtA,14432
+python/pii.py,sha256=Tgbq1NwM5L0I03aESNQvzbWd8WYlF5JLcYOxd933vE0,7412
+python/queue.py,sha256=jYZNFFdTbVNZNiPQ-CCZ7y-KiQ5QLs0zIQG6-23d43I,6756
+python/session.py,sha256=UtKruvKNWQW7qcFTmWrwEXCA6CutNzh4g5n4oT9E5xg,4927
+python/trace.py,sha256=D7cfeBQxOmkJl7W9VCrYL0MXLMLKOc4XVpIq7sL6p1M,1517
+python/integrations/__init__.py,sha256=hrYDWS9PVUfRurkszRvna3_2DB2nKYABF9jX1X79-Vg,144
+python/integrations/langchain.py,sha256=qJAlpZ6y-gBZ_xNwEls1eO4ikwCPmqwjZzB4PlnMjzE,14567
+python/wrappers/__init__.py,sha256=j8pwozRAMY633wFz-sWQtIajZvnK_4updOxQnfwRm-8,305
+python/wrappers/anthropic_wrapper.py,sha256=WVvmWU4eKc_Qjgj4_ImqBnGddV5MfVgMmOKvQjBkhB0,6561
+python/wrappers/gemini_wrapper.py,sha256=GdIZ5FlGI6Wrdr5WXUgyd5l2rPlAYnBUc4nI-v1fb0c,10976
+python/wrappers/openai_wrapper.py,sha256=sziE6ClwwEKxpgCP_FxAhuXXoia3H9wZkSNIt-EMZRc,5859
+aitracer-0.1.0.dist-info/METADATA,sha256=vBNOOuRH_9ek5_fso_QV-ugjeecfwbliXBJ2wg9EUvU,5875
+aitracer-0.1.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+aitracer-0.1.0.dist-info/RECORD,,

aitracer-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.28.0
+Root-Is-Purelib: true
+Tag: py3-none-any

python/__init__.py

@@ -0,0 +1,30 @@
+"""
+AITracer SDK - Monitor your AI/LLM applications
+
+Usage:
+    from aitracer import AITracer
+    from openai import OpenAI
+
+    tracer = AITracer(api_key="at-xxxxxxxx", project="my-chatbot")
+    client = tracer.wrap_openai(OpenAI())
+
+    response = client.chat.completions.create(
+        model="gpt-4o",
+        messages=[{"role": "user", "content": "Hello!"}]
+    )
+
+LangChain Integration:
+    from aitracer.integrations.langchain import AITracerCallbackHandler
+    from langchain_openai import ChatOpenAI
+
+    handler = AITracerCallbackHandler(api_key="at-xxx", project="my-langchain-app")
+    llm = ChatOpenAI(callbacks=[handler])
+"""
+
+from aitracer.client import AITracer
+from aitracer.pii import PIIDetector
+from aitracer.session import Session
+from aitracer.trace import Trace
+
+__version__ = "0.1.0"
+__all__ = ["AITracer", "Trace", "Session", "PIIDetector"]