kalibr 1.2.2__py3-none-any.whl → 1.2.3__py3-none-any.whl

{kalibr-1.2.2.dist-info → kalibr-1.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: kalibr
-Version: 1.2.2
+Version: 1.2.3
 Summary: Unified LLM Observability & Multi-Model AI Integration Framework - Deploy to GPT, Claude, Gemini, Copilot with full telemetry.
 Author-email: Kalibr Team <support@kalibr.systems>
 License: Apache-2.0

{kalibr-1.2.2.dist-info → kalibr-1.2.3.dist-info}/RECORD

@@ -35,15 +35,15 @@ kalibr/middleware/__init__.py,sha256=qyDUn_irAX67MS-IkuDVxg4RmFnJHDf_BfIT3qfGoBI
 kalibr/middleware/auto_tracer.py,sha256=ZBSBM0O3a6rwVzfik1n5NUmQDah8_iaf86rU64aPYT4,13037
 kalibr_crewai/__init__.py,sha256=b0HFTiE80eArtSMBOIEKu1JM6KU0tCjEylKCVVVF29Q,1796
 kalibr_crewai/callbacks.py,sha256=_d1M4J-6XfKqrVIxnOgOQu57jpFKVv-VIsmPV0HNgZ4,20419
-kalibr_crewai/instrumentor.py,sha256=x26v0RcriImkPiC8KB1Hmez1XOYLcDa9o-g35BMu5Ek,24420
+kalibr_crewai/instrumentor.py,sha256=-G_-xaqE3Op70MSEIaZjPYioGDxKRagwLbZmcmmvzFg,26793
 kalibr_langchain/__init__.py,sha256=O4XYVyhLp1v-Y1kGZw3zD-tUK9wp0UX8Jt6oN0QTHN4,1373
 kalibr_langchain/async_callback.py,sha256=_Mj_YrKbULNtfxixZ7iwiHyWEV9l178ZA5Oy5A5Pakk,27748
 kalibr_langchain/callback.py,sha256=SNM1aHOXdG55grHmGyTwbXOeM6hjZTub2REiZD2H-d8,35216
 kalibr_openai_agents/__init__.py,sha256=wL59LzGstptKigfQDrKKt_7hcMO1JGVQtVAsE0lz-Zw,1367
 kalibr_openai_agents/processor.py,sha256=F550sdRf3rpguP1yOlgAUQWDLPBy4hSACV3-zOyCpOU,18257
-kalibr-1.2.2.dist-info/LICENSE,sha256=5mwAnB38l3_PjmOQn6_L6cZnJvus143DUjMBPIH1yso,10768
-kalibr-1.2.2.dist-info/METADATA,sha256=XqhSOZekE5D5WQVT0pH2h1HCZWXiruGM9CqOelAZnf0,11201
-kalibr-1.2.2.dist-info/WHEEL,sha256=beeZ86-EfXScwlR_HKu4SllMC9wUEj_8Z_4FJ3egI2w,91
-kalibr-1.2.2.dist-info/entry_points.txt,sha256=Kojlc6WRX8V1qS9lOMdDPZpTUVHCtzGtHqXusErgmLY,47
-kalibr-1.2.2.dist-info/top_level.txt,sha256=dIfBOWUnnHGFDwgz5zfIx5_0bU3wOUgAbYr4JcFHZmo,59
-kalibr-1.2.2.dist-info/RECORD,,
+kalibr-1.2.3.dist-info/LICENSE,sha256=5mwAnB38l3_PjmOQn6_L6cZnJvus143DUjMBPIH1yso,10768
+kalibr-1.2.3.dist-info/METADATA,sha256=x6_p6pRthFdGcHYRMXeKCmf0WRwMU5xE9FT142ce4ts,11201
+kalibr-1.2.3.dist-info/WHEEL,sha256=beeZ86-EfXScwlR_HKu4SllMC9wUEj_8Z_4FJ3egI2w,91
+kalibr-1.2.3.dist-info/entry_points.txt,sha256=Kojlc6WRX8V1qS9lOMdDPZpTUVHCtzGtHqXusErgmLY,47
+kalibr-1.2.3.dist-info/top_level.txt,sha256=dIfBOWUnnHGFDwgz5zfIx5_0bU3wOUgAbYr4JcFHZmo,59
+kalibr-1.2.3.dist-info/RECORD,,

kalibr_crewai/instrumentor.py

@@ -12,6 +12,8 @@ from datetime import datetime, timezone
 from functools import wraps
 from typing import Any, Callable, Dict, List, Optional
 
+from opentelemetry import trace as otel_trace
+
 from .callbacks import EventBatcher, _count_tokens, _get_provider_from_model
 
 # Import Kalibr cost adapters
@@ -275,6 +277,21 @@ class KalibrCrewAIInstrumentor:
                 duration_ms = int((time.time() - start_time) * 1000)
                 ts_end = datetime.now(timezone.utc)
 
+                # Enrich CrewAI's OTel span with Kalibr telemetry
+                try:
+                    current_span = otel_trace.get_current_span()
+                    if current_span and current_span.is_recording():
+                        current_span.set_attribute("kalibr.cost_usd", instrumentor._accumulated_cost)
+                        current_span.set_attribute("kalibr.input_tokens", instrumentor._accumulated_tokens["input"])
+                        current_span.set_attribute("kalibr.output_tokens", instrumentor._accumulated_tokens["output"])
+                        current_span.set_attribute("kalibr.total_tokens", instrumentor._accumulated_tokens["input"] + instrumentor._accumulated_tokens["output"])
+                        current_span.set_attribute("kalibr.model_id", model_name)
+                        current_span.set_attribute("kalibr.provider", provider)
+                        current_span.set_attribute("kalibr.duration_ms", duration_ms)
+                        current_span.set_attribute("kalibr.tenant_id", instrumentor.tenant_id)
+                except Exception:
+                    pass  # Don't fail if span enrichment fails
+
                 # Build output info
                 output_preview = None
                 if instrumentor.capture_output and result is not None:
@@ -376,6 +393,21 @@ class KalibrCrewAIInstrumentor:
                 duration_ms = int((time.time() - start_time) * 1000)
                 ts_end = datetime.now(timezone.utc)
 
+                # Enrich CrewAI's OTel span with Kalibr telemetry
+                try:
+                    current_span = otel_trace.get_current_span()
+                    if current_span and current_span.is_recording():
+                        current_span.set_attribute("kalibr.cost_usd", instrumentor._accumulated_cost)
+                        current_span.set_attribute("kalibr.input_tokens", instrumentor._accumulated_tokens["input"])
+                        current_span.set_attribute("kalibr.output_tokens", instrumentor._accumulated_tokens["output"])
+                        current_span.set_attribute("kalibr.total_tokens", instrumentor._accumulated_tokens["input"] + instrumentor._accumulated_tokens["output"])
+                        current_span.set_attribute("kalibr.model_id", model_name)
+                        current_span.set_attribute("kalibr.provider", provider)
+                        current_span.set_attribute("kalibr.duration_ms", duration_ms)
+                        current_span.set_attribute("kalibr.tenant_id", instrumentor.tenant_id)
+                except Exception:
+                    pass  # Don't fail if span enrichment fails
+
                 output_preview = None
                 if instrumentor.capture_output and result is not None:
                     output_preview = str(result)[:500]