metergraph 0.1.0__tar.gz → 0.2.0__tar.gz

{metergraph-0.1.0 → metergraph-0.2.0}/PKG-INFO

@@ -1,8 +1,16 @@
 Metadata-Version: 2.4
 Name: metergraph
-Version: 0.1.0
+Version: 0.2.0
 Summary: Fire-and-forget LLM spend capture for Metergraph
+Author: Pioneer Square Labs
 License-Expression: Apache-2.0
+Project-URL: Homepage, https://www.metergraph.dev/
+Project-URL: Repository, https://github.com/PioneerSquareLabs/metergraphsdk
+Project-URL: Issues, https://github.com/PioneerSquareLabs/metergraphsdk/issues
+Keywords: llm,observability,openai,anthropic,gemini,cost-tracking
+Classifier: Intended Audience :: Developers
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: Typing :: Typed
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 Provides-Extra: dev

metergraph-0.2.0/pyproject.toml

@@ -0,0 +1,33 @@
+[project]
+name = "metergraph"
+version = "0.2.0"
+description = "Fire-and-forget LLM spend capture for Metergraph"
+readme = "README.md"
+requires-python = ">=3.10"
+license = "Apache-2.0"
+authors = [{ name = "Pioneer Square Labs" }]
+keywords = ["llm", "observability", "openai", "anthropic", "gemini", "cost-tracking"]
+classifiers = [
+    "Intended Audience :: Developers",
+    "Topic :: Software Development :: Libraries :: Python Modules",
+    "Typing :: Typed",
+]
+dependencies = []
+
+[project.urls]
+Homepage = "https://www.metergraph.dev/"
+Repository = "https://github.com/PioneerSquareLabs/metergraphsdk"
+Issues = "https://github.com/PioneerSquareLabs/metergraphsdk/issues"
+
+[project.optional-dependencies]
+dev = ["pytest>=8"]
+
+[build-system]
+requires = ["setuptools>=68"]
+build-backend = "setuptools.build_meta"
+
+[tool.setuptools.packages.find]
+where = ["src"]
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]

{metergraph-0.1.0 → metergraph-0.2.0}/src/metergraph/_capture.py

@@ -1111,6 +1111,10 @@ def wrap(client: Any, *, provider: str | None = None) -> Any:
     chat = getattr(getattr(client, "chat", None), "completions", None)
     if chat is not None:
         seams.append((chat, "create", "chat.completions"))
+        seams.append((chat, "parse", "chat.completions.parse"))
+    beta_chat = getattr(getattr(getattr(client, "beta", None), "chat", None), "completions", None)
+    if beta_chat is not None:
+        seams.append((beta_chat, "parse", "chat.completions.parse"))
     responses = getattr(client, "responses", None)
     if responses is not None:
         seams.extend(

{metergraph-0.1.0 → metergraph-0.2.0}/src/metergraph/_version.py

@@ -8,4 +8,4 @@ import importlib.metadata
 try:
     SDK_VERSION = importlib.metadata.version("metergraph")
 except Exception:
-    SDK_VERSION = "0.1.0"
+    SDK_VERSION = "0.2.0"

{metergraph-0.1.0 → metergraph-0.2.0}/src/metergraph.egg-info/PKG-INFO

@@ -1,8 +1,16 @@
 Metadata-Version: 2.4
 Name: metergraph
-Version: 0.1.0
+Version: 0.2.0
 Summary: Fire-and-forget LLM spend capture for Metergraph
+Author: Pioneer Square Labs
 License-Expression: Apache-2.0
+Project-URL: Homepage, https://www.metergraph.dev/
+Project-URL: Repository, https://github.com/PioneerSquareLabs/metergraphsdk
+Project-URL: Issues, https://github.com/PioneerSquareLabs/metergraphsdk/issues
+Keywords: llm,observability,openai,anthropic,gemini,cost-tracking
+Classifier: Intended Audience :: Developers
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: Typing :: Typed
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 Provides-Extra: dev

{metergraph-0.1.0 → metergraph-0.2.0}/tests/test_sdk.py

@@ -112,6 +112,38 @@ def test_wrap_sync_records_usage_context_and_preserves_response(tmp_path):
     _capture.set_runtime(None)
 
 
+def test_wrap_patches_create_and_parse_on_both_chat_and_beta_chat(tmp_path):
+    rows = Rows()
+    _capture.set_runtime(
+        Runtime(rows, Options(app_root=str(Path(__file__).parents[1])))
+    )
+
+    class Completions:
+        def create(self, **kwargs):
+            return response()
+
+        def parse(self, **kwargs):
+            return response()
+
+    client = SimpleNamespace(
+        chat=SimpleNamespace(completions=Completions()),
+        beta=SimpleNamespace(chat=SimpleNamespace(completions=Completions())),
+        responses=None,
+    )
+    metergraph.wrap(client, provider="openai")
+
+    client.chat.completions.create(model="gpt-test", messages=[])
+    client.chat.completions.parse(model="gpt-test", messages=[])
+    client.beta.chat.completions.parse(model="gpt-test", messages=[])
+
+    assert [row["endpoint"] for row in rows.rows] == [
+        "chat.completions",
+        "chat.completions.parse",
+        "chat.completions.parse",
+    ]
+    _capture.set_runtime(None)
+
+
 def gemini_response(text="gemini done"):
     return SimpleNamespace(
         text=text,

metergraph-0.1.0/pyproject.toml

@@ -1,21 +0,0 @@
-[project]
-name = "metergraph"
-version = "0.1.0"
-description = "Fire-and-forget LLM spend capture for Metergraph"
-readme = "README.md"
-requires-python = ">=3.10"
-license = "Apache-2.0"
-dependencies = []
-
-[project.optional-dependencies]
-dev = ["pytest>=8"]
-
-[build-system]
-requires = ["setuptools>=68"]
-build-backend = "setuptools.build_meta"
-
-[tool.setuptools.packages.find]
-where = ["src"]
-
-[tool.pytest.ini_options]
-testpaths = ["tests"]