bluejay-sdk 0.1.0__py3-none-any.whl

bluejay_sdk/__init__.py

@@ -0,0 +1,3 @@
+from bluejay_sdk.tracing import BluejayTracing
+
+__all__ = ["BluejayTracing"]

bluejay_sdk/tracing.py

@@ -0,0 +1,135 @@
+import logging
+import os
+
+import httpx
+import opentelemetry.trace as otel_trace
+from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
+from opentelemetry.sdk import trace as trace_sdk
+from opentelemetry.sdk.resources import SERVICE_NAME, Resource
+from opentelemetry.sdk.trace.export import SimpleSpanProcessor
+
+logger = logging.getLogger("bluejay_sdk")
+
+DEFAULT_OTLP_ENDPOINT = "https://otlp.getbluejay.ai/v1/traces"
+DEFAULT_API_URL = "https://api.getbluejay.ai/v1"
+
+
+class BluejayTracing:
+    def __init__(
+        self,
+        api_key: str,
+        *,
+        service_name: str = "bluejay-agent",
+        otlp_endpoint: str = DEFAULT_OTLP_ENDPOINT,
+        api_url: str = DEFAULT_API_URL,
+    ):
+        self.api_key = api_key
+        self.api_url = api_url
+        self.simulation_result_id: str | None = None
+        self.trace_id: str | None = None
+
+        resource = Resource.create({SERVICE_NAME: service_name})
+        self.tracer_provider = trace_sdk.TracerProvider(resource=resource)
+        self.tracer_provider.add_span_processor(
+            SimpleSpanProcessor(
+                OTLPSpanExporter(otlp_endpoint, headers={"X-API-KEY": api_key})
+            )
+        )
+
+        try:
+            from livekit.agents.telemetry import set_tracer_provider
+
+            set_tracer_provider(self.tracer_provider)
+        except ImportError:
+            pass
+
+        otel_trace.set_tracer_provider(self.tracer_provider)
+
+    @classmethod
+    def init(
+        cls,
+        api_key: str | None = None,
+        *,
+        service_name: str | None = None,
+        otlp_endpoint: str | None = None,
+        api_url: str | None = None,
+    ) -> "BluejayTracing":
+        key = api_key or os.getenv("BLUEJAY_API_KEY")
+        if not key:
+            raise ValueError("BLUEJAY_API_KEY not set and no api_key provided")
+
+        return cls(
+            api_key=key,
+            service_name=service_name or os.getenv("BLUEJAY_SERVICE_NAME", "bluejay-agent"),
+            otlp_endpoint=otlp_endpoint or os.getenv("BLUEJAY_OTLP_ENDPOINT", DEFAULT_OTLP_ENDPOINT),
+            api_url=api_url or os.getenv("BLUEJAY_API_URL", DEFAULT_API_URL),
+        )
+
+    def link_to_session(self, ctx) -> None:
+        """Register participant listener and shutdown callback on a LiveKit JobContext."""
+        from livekit import rtc
+
+        @ctx.room.on("participant_connected")
+        def on_participant_connected(participant: rtc.RemoteParticipant):
+            if self.simulation_result_id:
+                return
+            result_id = (participant.attributes or {}).get("X-Simulation-Result-Id")
+            if result_id:
+                self.simulation_result_id = str(result_id)
+                logger.info(f"captured simulation_result_id={result_id}")
+
+        for p in ctx.room.remote_participants.values():
+            result_id = (p.attributes or {}).get("X-Simulation-Result-Id")
+            if result_id:
+                self.simulation_result_id = str(result_id)
+                break
+
+        ctx.add_shutdown_callback(self._on_shutdown)
+
+    def capture_trace(self) -> str | None:
+        """Capture trace_id from the current OTel span (call after session.start())."""
+        span = otel_trace.get_current_span()
+        span_ctx = span.get_span_context()
+        if span_ctx.is_valid:
+            self.trace_id = format(span_ctx.trace_id, "032x")
+            logger.info(f"agent_session trace_id={self.trace_id}")
+        return self.trace_id
+
+    async def _on_shutdown(self) -> None:
+        logger.info(f"shutdown — simulation_result_id={self.simulation_result_id} trace_id={self.trace_id}")
+
+        try:
+            self.tracer_provider.force_flush()
+        except Exception as e:
+            logger.error(f"force_flush failed: {e}")
+
+        if self.simulation_result_id and self.trace_id:
+            await self._post_result()
+        else:
+            logger.warning(f"skipping update-simulation-result — simulation_result_id={self.simulation_result_id} trace_id={self.trace_id}")
+
+        try:
+            self.tracer_provider.shutdown()
+        except Exception as e:
+            logger.error(f"tracer shutdown failed: {e}")
+
+    async def _post_result(self) -> None:
+        payload = {
+            "simulation_result_id": self.simulation_result_id,
+            "trace_ids": [self.trace_id],
+        }
+        headers = {"X-API-Key": self.api_key, "Content-Type": "application/json"}
+
+        async with httpx.AsyncClient(timeout=10) as client:
+            try:
+                resp = await client.post(
+                    f"{self.api_url}/update-simulation-result",
+                    json=payload,
+                    headers=headers,
+                )
+                resp.raise_for_status()
+                logger.info("update-simulation-result SUCCESS")
+            except httpx.HTTPStatusError as e:
+                logger.error(f"update-simulation-result FAILED: {e.response.status_code} {e.response.text}")
+            except Exception as e:
+                logger.error(f"update-simulation-result FAILED: {e}")

bluejay_sdk-0.1.0.dist-info/METADATA

@@ -0,0 +1,10 @@
+Metadata-Version: 2.4
+Name: bluejay-sdk
+Version: 0.1.0
+Summary: Bluejay SDK for LiveKit voice agents
+Requires-Python: >=3.10
+Requires-Dist: httpx>=0.25
+Requires-Dist: opentelemetry-exporter-otlp-proto-http
+Requires-Dist: opentelemetry-sdk
+Provides-Extra: livekit
+Requires-Dist: livekit-agents>=1.0; extra == 'livekit'

bluejay_sdk-0.1.0.dist-info/RECORD

@@ -0,0 +1,5 @@
+bluejay_sdk/__init__.py,sha256=vNAhrGWEJmZsCscA6NrWdS7U5JInuChqUxDDuFlTT8Y,77
+bluejay_sdk/tracing.py,sha256=zPc-PZPjUOV0OYx5TGLaV3eOx91-FXOE36e2_ZhmFLc,5070
+bluejay_sdk-0.1.0.dist-info/METADATA,sha256=E7IRPCdXebHtC_aMI8kWlamim1hwLuVl4T6UY0drTAo,318
+bluejay_sdk-0.1.0.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+bluejay_sdk-0.1.0.dist-info/RECORD,,

bluejay_sdk-0.1.0.dist-info/WHEEL

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