infinity-observability-sdk 0.1.0__py2.py3-none-any.whl

infinity_observability_sdk/__init__.py

@@ -0,0 +1,16 @@
+"""
+Infinity Observability SDK
+
+Provides standardized observability configuration for Infinity services,
+including OpenTelemetry tracing via logfire with routing to the central
+observability collector.
+
+Required environment variables:
+    INFINITY_OBSERVABILITY_ENDPOINT: The OTLP collector endpoint
+        (e.g., https://observability.staging.infinityconstellation-labs.com)
+    SECRET_INFINITY_OBSERVABILITY_API_KEY: API key for authentication
+"""
+
+from .sdk import configure_observability, agent_context, ObservabilityConfig
+
+__all__ = ["configure_observability", "agent_context", "ObservabilityConfig"]

infinity_observability_sdk/sdk.py

@@ -0,0 +1,115 @@
+import os
+from contextlib import contextmanager
+from dataclasses import dataclass
+
+import logfire
+from opentelemetry import trace
+from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
+from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.trace.export import BatchSpanProcessor
+
+
+@dataclass
+class ObservabilityConfig:
+    business_unit_name: str
+    service_name: str
+    instrument_pydantic_ai: bool = True
+    instrument_httpx: bool = True
+
+
+_config: ObservabilityConfig | None = None
+
+
+@contextmanager
+def agent_context(
+    agent_employee_equivalent: str,
+    hourly_rate: float,
+    agent_name: str,
+    task_description: str,
+    task_instance_identifier: str,
+    approximate_person_hours: float,
+    **metadata,
+):
+    global _config
+
+    if _config is None:
+        raise Exception(
+            "It looks like the infinity constellation monitoring SDK is not yet configured. Please run configure_observability before adding agent_context"
+        )
+
+    attributes = {
+        "infinity_observability_sdk.agent_employee_equivalent": agent_employee_equivalent,
+        "infinity_observability_sdk.hourly_rate": hourly_rate,
+        "infinity_observability_sdk.business_unit_name": _config.business_unit_name,
+        "infinity_observability_sdk.service_name": _config.service_name,
+        "infinity_observability_sdk.agent_name": agent_name,
+        "infinity_observability_sdk.task_description": task_description,
+        "infinity_observability_sdk.task_instance_identifier": task_instance_identifier,
+        "infinity_observability_sdk.approximate_person_hours": approximate_person_hours,
+        "infinity_observability_sdk.ai_wattage": approximate_person_hours * hourly_rate,
+        "infinity_observability_sdk.is_tracked_agent": True,
+        "infinity_observability_sdk.version": "0.1",
+        **metadata,
+    }
+
+    baggage = {
+        "infinity_observability_sdk.baggage.business_unit_name": _config.business_unit_name,
+        "infinity_observability_sdk.baggage.service_name": _config.service_name,
+        "infinity_observability_sdk.baggage.agent_name": agent_name,
+        "infinity_observability_sdk.baggage.is_tracked_agent": "true",
+    }
+
+    with (
+        logfire.set_baggage(**baggage),
+        logfire.span("agent_context", **attributes) as span,
+    ):
+        yield span
+
+
+def configure_observability(config: ObservabilityConfig) -> None:
+    global _config
+
+    endpoint = os.environ.get("INFINITY_OBSERVABILITY_ENDPOINT")
+    api_key = os.environ.get("INFINITY_OBSERVABILITY_API_KEY")
+
+    if not endpoint:
+        raise RuntimeError(
+            "INFINITY_OBSERVABILITY_ENDPOINT environment variable is required. "
+            "Set it to the observability collector endpoint "
+        )
+
+    if not api_key:
+        raise RuntimeError(
+            "INFINITY_OBSERVABILITY_API_KEY environment variable is required. "
+            "Set it to your observability API key."
+        )
+
+    otlp_exporter = OTLPSpanExporter(
+        endpoint=f"{endpoint}/v1/traces",
+        headers={"X-Api-Key": api_key},
+    )
+    span_processor = BatchSpanProcessor(otlp_exporter)
+
+    provider = trace.get_tracer_provider()
+
+    if isinstance(provider, TracerProvider):
+        provider.add_span_processor(span_processor)
+    else:
+        logfire.configure(
+            send_to_logfire=False,
+            service_name=config.business_unit_name + "." + config.service_name,
+            additional_span_processors=[span_processor],
+        )
+
+    if config.instrument_httpx:
+        logfire.instrument_httpx(capture_all=True)
+
+    if config.instrument_pydantic_ai:
+        logfire.instrument_pydantic_ai()
+
+    _config = config
+
+    print(
+        "Successfully initialized Infinity Monitoring. OTLP traffic is being forwarded to:",
+        endpoint,
+    )

infinity_observability_sdk-0.1.0.dist-info/METADATA

@@ -0,0 +1,12 @@
+Metadata-Version: 2.4
+Name: infinity-observability-sdk
+Version: 0.1.0
+Summary: Infinity Observability SDK
+License-File: LICENSE
+Requires-Dist: logfire>=4.18.0
+Requires-Dist: opentelemetry-exporter-otlp-proto-http>=1.39.1
+Requires-Dist: pytest>=9.0.2
+Requires-Dist: ruff>=0.14.13
+Description-Content-Type: text/markdown
+
+# Infinity Observability SDK

infinity_observability_sdk-0.1.0.dist-info/RECORD

@@ -0,0 +1,6 @@
+infinity_observability_sdk/__init__.py,sha256=ncC7P_l2z8eZUo64fNtBX2pEekcoqCAOndXpqDQ36go,608
+infinity_observability_sdk/sdk.py,sha256=YFVAq8x3vMJPk2qcHnH-X3_rEKHHKRpSPyu5P82RZkU,3823
+infinity_observability_sdk-0.1.0.dist-info/METADATA,sha256=_azkfq3iYtxV_YFpAMwUANmiKUeFG4vMZhaVNumiQig,349
+infinity_observability_sdk-0.1.0.dist-info/WHEEL,sha256=aha0VrrYvgDJ3Xxl3db_g_MDIW-ZexDdrc_m-Hk8YY4,105
+infinity_observability_sdk-0.1.0.dist-info/licenses/LICENSE,sha256=ACwmltkrXIz5VsEQcrqljq-fat6ZXAMepjXGoe40KtE,1069
+infinity_observability_sdk-0.1.0.dist-info/RECORD,,

infinity_observability_sdk-0.1.0.dist-info/WHEEL

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

infinity_observability_sdk-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) [year] [fullname]
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.