dagster-openai 0.26.13__py3-none-any.whl → 0.26.15__py3-none-any.whl

dagster_openai/resources.py

@@ -38,7 +38,9 @@ context_to_counters = WeakKeyDictionary()
 
 
 def _add_to_asset_metadata(
-    context: AssetExecutionContext, usage_metadata: dict, output_name: Optional[str]
+    context: AssetExecutionContext,
+    usage_metadata: dict[str, int],
+    output_name: Optional[str],
 ):
     if context not in context_to_counters:
         context_to_counters[context] = defaultdict(lambda: 0)
@@ -46,7 +48,11 @@ def _add_to_asset_metadata(
 
     for metadata_key, delta in usage_metadata.items():
         counters[metadata_key] += delta
-    context.add_output_metadata(dict(counters), output_name)
+
+    context.add_output_metadata(
+        metadata=dict(counters),
+        output_name=output_name,
+    )
 
 
 @public
@@ -128,15 +134,24 @@ def with_usage_metadata(
     @wraps(func)
     def wrapper(*args, **kwargs):
         response = func(*args, **kwargs)
+        calls_key = f"openai.{response.model}.calls"
+        total_tokens_key = f"openai.{response.model}.total_tokens"
+        prompt_tokens_key = f"openai.{response.model}.prompt_tokens"
+        completion_tokens_key = f"openai.{response.model}.completion_tokens"
+
         usage = response.usage
         usage_metadata = {
-            "openai.calls": 1,
-            "openai.total_tokens": usage.total_tokens,
-            "openai.prompt_tokens": usage.prompt_tokens,
+            calls_key: 1,
+            total_tokens_key: usage.total_tokens,
+            prompt_tokens_key: usage.prompt_tokens,
         }
         if hasattr(usage, "completion_tokens"):
-            usage_metadata["openai.completion_tokens"] = usage.completion_tokens
-        _add_to_asset_metadata(context, usage_metadata, output_name)
+            usage_metadata[completion_tokens_key] = usage.completion_tokens
+        _add_to_asset_metadata(
+            context=context,
+            usage_metadata=usage_metadata,
+            output_name=output_name,
+        )
 
         return response
 

dagster_openai/version.py

@@ -1 +1 @@
-__version__ = "0.26.13"
+__version__ = "0.26.15"

{dagster_openai-0.26.13.dist-info → dagster_openai-0.26.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dagster-openai
-Version: 0.26.13
+Version: 0.26.15
 Summary: An OpenAI client resource for interacting with OpenAI API.
 Home-page: https://github.com/dagster-io/dagster/tree/master/python_modules/libraries/dagster-openai
 Author: Dagster Labs
@@ -12,7 +12,7 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 License-File: LICENSE
-Requires-Dist: dagster==1.10.13
+Requires-Dist: dagster==1.10.15
 Requires-Dist: openai
 Dynamic: author
 Dynamic: author-email

dagster_openai-0.26.15.dist-info/RECORD

@@ -0,0 +1,9 @@
+dagster_openai/__init__.py,sha256=7_AS1TCgl9chTzaFNTsCpt3G-aVQnzUAz-RpT9OC-Xk,299
+dagster_openai/py.typed,sha256=la67KBlbjXN-_-DfGNcdOcjYumVpKG_Tkw-8n5dnGB4,8
+dagster_openai/resources.py,sha256=Z3sknKI37Ru72GuqHwmUuzzC51j4QBMNCG_57ro5t8g,15832
+dagster_openai/version.py,sha256=IIhGitgHey00De1EI8U4OzQYe2QXMJMKhPNhRMkJeas,24
+dagster_openai-0.26.15.dist-info/licenses/LICENSE,sha256=4lsMW-RCvfVD4_F57wrmpe3vX1xwUk_OAKKmV_XT7Z0,11348
+dagster_openai-0.26.15.dist-info/METADATA,sha256=Y7OBoDVQ2Lcyk7-_7WuUy6memJPsInbtgD2Wp7YeUfQ,800
+dagster_openai-0.26.15.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
+dagster_openai-0.26.15.dist-info/top_level.txt,sha256=-66JR-jWfULrYSXUk2AfOdXl8cQyyldzBvBmEQdKn9o,15
+dagster_openai-0.26.15.dist-info/RECORD,,

{dagster_openai-0.26.13.dist-info → dagster_openai-0.26.15.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.1.0)
+Generator: setuptools (80.7.1)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

{dagster_openai-0.26.13.dist-info → dagster_openai-0.26.15.dist-info}/licenses/LICENSE

@@ -186,7 +186,7 @@
       same "printed page" as the copyright notice for easier
       identification within third-party archives.
 
-   Copyright 2024 Dagster Labs, Inc".
+   Copyright 2025 Dagster Labs, Inc.
 
    Licensed under the Apache License, Version 2.0 (the "License");
    you may not use this file except in compliance with the License.

dagster_openai-0.26.13.dist-info/RECORD

@@ -1,9 +0,0 @@
-dagster_openai/__init__.py,sha256=7_AS1TCgl9chTzaFNTsCpt3G-aVQnzUAz-RpT9OC-Xk,299
-dagster_openai/py.typed,sha256=la67KBlbjXN-_-DfGNcdOcjYumVpKG_Tkw-8n5dnGB4,8
-dagster_openai/resources.py,sha256=lpvQeDKfNG9IPdYafoGTBPGwyjD_Kkf8A1EvC7qNxqw,15439
-dagster_openai/version.py,sha256=EDf5CcPxrnkRq2ViErwz2GB5gPrihrLuUdZnXlor9gM,24
-dagster_openai-0.26.13.dist-info/licenses/LICENSE,sha256=HTQpxmWIpPec-pUSMBAj0wR44OnVdRmrWygQ-GUuVzQ,11349
-dagster_openai-0.26.13.dist-info/METADATA,sha256=Plei-a5LK1VXDaSi4cEv7qiXc_6tktlOb1XkCwm4bb4,800
-dagster_openai-0.26.13.dist-info/WHEEL,sha256=wXxTzcEDnjrTwFYjLPcsW_7_XihufBwmpiBeiXNBGEA,91
-dagster_openai-0.26.13.dist-info/top_level.txt,sha256=-66JR-jWfULrYSXUk2AfOdXl8cQyyldzBvBmEQdKn9o,15
-dagster_openai-0.26.13.dist-info/RECORD,,