lmnr 0.7.7__tar.gz → 0.7.8__tar.gz

{lmnr-0.7.7 → lmnr-0.7.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lmnr
-Version: 0.7.7
+Version: 0.7.8
 Summary: Python SDK for Laminar
 Author: lmnr.ai
 Author-email: lmnr.ai <founders@lmnr.ai>

{lmnr-0.7.7 → lmnr-0.7.8}/pyproject.toml

@@ -6,7 +6,7 @@
 
 [project]
 name = "lmnr"
-version = "0.7.7"
+version = "0.7.8"
 description = "Python SDK for Laminar"
 authors = [
   { name = "lmnr.ai", email = "founders@lmnr.ai" }
@@ -121,21 +121,21 @@ dev = [
   "flake8>=7.2.0",
   "pytest>=8.3.5",
   "pytest-sugar>=1.0.0",
-  "pytest-asyncio>=0.26.0",
+  "pytest-asyncio>=1.1.0",
   "playwright>=1.52.0",
   "vcrpy>=7.0.0",
   # litellm breaks with openai>=1.100.0 (as of litellm 1.75.8)
   "openai>=1.99.7,<1.100.0",
   "pytest-recording>=0.13.4",
   "patchright>=1.52.3",
-  "google-genai>=1.19.0",
+  "google-genai>=1.31.0",
   "langgraph>=0.4.8",
-  "langchain-core>=0.3.64",
-  "langchain>=0.3.25",
+  "langchain-core>=0.3.75",
+  "langchain>=0.3.27",
   "litellm>=1.75.8",
   "groq>=0.30.0",
   "anthropic>=0.60.0",
-  "langchain-openai>=0.3.28",
+  "langchain-openai>=0.3.32",
 ]
 
 [build-system]

{lmnr-0.7.7 → lmnr-0.7.8}/src/lmnr/opentelemetry_lib/opentelemetry/instrumentation/google_genai/__init__.py

@@ -375,6 +375,7 @@ def _build_from_streaming_response(
             # total token count in every chunk is greater by prompt token count than it should be,
             # thus this awkward logic here
             if aggregated_usage_metadata.get("prompt_token_count") is None:
+                # or 0, not .get(key, 0), because sometimes the value is explicitly None
                 aggregated_usage_metadata["prompt_token_count"] = (
                     usage_dict.get("prompt_token_count") or 0
                 )
@@ -417,14 +418,15 @@ async def _abuild_from_streaming_response(
     aggregated_usage_metadata = defaultdict(int)
     model_version = None
     async for chunk in response:
+        if chunk.model_version:
+            model_version = chunk.model_version
+
         if chunk.candidates:
             # Currently gemini throws an error if you pass more than one candidate
             # with streaming
             if chunk.candidates and len(chunk.candidates) > 0:
                 final_parts += chunk.candidates[0].content.parts or []
                 role = chunk.candidates[0].content.role or role
-                if chunk.model_version:
-                    model_version = chunk.model_version
         if chunk.usage_metadata:
             usage_dict = to_dict(chunk.usage_metadata)
             # prompt token count is sent in every chunk
@@ -432,11 +434,12 @@ async def _abuild_from_streaming_response(
             # total token count in every chunk is greater by prompt token count than it should be,
             # thus this awkward logic here
             if aggregated_usage_metadata.get("prompt_token_count") is None:
-                aggregated_usage_metadata["prompt_token_count"] = usage_dict.get(
-                    "prompt_token_count"
+                # or 0, not .get(key, 0), because sometimes the value is explicitly None
+                aggregated_usage_metadata["prompt_token_count"] = (
+                    usage_dict.get("prompt_token_count") or 0
                 )
-                aggregated_usage_metadata["total_token_count"] = usage_dict.get(
-                    "total_token_count"
+                aggregated_usage_metadata["total_token_count"] = (
+                    usage_dict.get("total_token_count") or 0
                 )
             aggregated_usage_metadata["candidates_token_count"] += (
                 usage_dict.get("candidates_token_count") or 0

{lmnr-0.7.7 → lmnr-0.7.8}/src/lmnr/version.py

@@ -3,7 +3,7 @@ import httpx
 from packaging import version
 
 
-__version__ = "0.7.7"
+__version__ = "0.7.8"
 PYTHON_VERSION = f"{sys.version_info.major}.{sys.version_info.minor}"