omnia-tracing 0.1.0__tar.gz

omnia_tracing-0.1.0/.env

@@ -0,0 +1 @@
+PY_PY=pypi-AgEIcHlwaS5vcmcCJDZlMDMxNDM5LTU2NjQtNDA0OS1iMDBmLTg3YTEyN2I1MGY5ZQACKlszLCIwYzc3YjA4Zi0yMjM1LTRiYWEtODEzMC1kOGNjYTc2MDY0MDAiXQAABiC-pMRm8OFpi3aPTC8WcA6RYye-O9lYrXl4X16aViCdbw

omnia_tracing-0.1.0/.gitignore

@@ -0,0 +1,6 @@
+node_modules/
+dist/
+__pycache__/
+*.egg-info/
+.venv/
+.pytest_cache/

omnia_tracing-0.1.0/PKG-INFO

@@ -0,0 +1,26 @@
+Metadata-Version: 2.5
+Name: omnia-tracing
+Version: 0.1.0
+Summary: Omnia tracing — standard OpenTelemetry, curated. One install, one line; eject anytime, your spans don't change.
+Project-URL: Repository, https://github.com/omnia-v/omnia-tracing
+License-Expression: Apache-2.0
+Requires-Python: >=3.10
+Requires-Dist: opentelemetry-exporter-otlp-proto-http>=1.27
+Requires-Dist: opentelemetry-instrumentation-anthropic>=0.30
+Requires-Dist: opentelemetry-instrumentation-google-generativeai>=0.30
+Requires-Dist: opentelemetry-instrumentation-langchain>=0.30
+Requires-Dist: opentelemetry-instrumentation-openai>=0.30
+Requires-Dist: opentelemetry-sdk>=1.27
+Description-Content-Type: text/markdown
+
+# omnia-tracing
+
+Standard OpenTelemetry, curated. One install, one line; eject anytime — your spans don't change.
+
+```python
+from omnia_tracing import setup
+
+tracing = setup()  # reads OMNIA_API_KEY and OMNIA_TAG
+```
+
+See the [repository README](https://github.com/omnia-v/omnia-tracing) for the full story, and `docs/eject.md` for the identical setup in vanilla OpenTelemetry.

omnia_tracing-0.1.0/README.md

@@ -0,0 +1,11 @@
+# omnia-tracing
+
+Standard OpenTelemetry, curated. One install, one line; eject anytime — your spans don't change.
+
+```python
+from omnia_tracing import setup
+
+tracing = setup()  # reads OMNIA_API_KEY and OMNIA_TAG
+```
+
+See the [repository README](https://github.com/omnia-v/omnia-tracing) for the full story, and `docs/eject.md` for the identical setup in vanilla OpenTelemetry.

omnia_tracing-0.1.0/pyproject.toml

@@ -0,0 +1,28 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "omnia-tracing"
+version = "0.1.0"
+description = "Omnia tracing — standard OpenTelemetry, curated. One install, one line; eject anytime, your spans don't change."
+license = "Apache-2.0"
+readme = "README.md"
+requires-python = ">=3.10"
+dependencies = [
+  "opentelemetry-sdk>=1.27",
+  "opentelemetry-exporter-otlp-proto-http>=1.27",
+  "opentelemetry-instrumentation-openai>=0.30",
+  "opentelemetry-instrumentation-anthropic>=0.30",
+  "opentelemetry-instrumentation-google-generativeai>=0.30",
+  "opentelemetry-instrumentation-langchain>=0.30",
+]
+
+[project.urls]
+Repository = "https://github.com/omnia-v/omnia-tracing"
+
+[dependency-groups]
+dev = ["pytest>=8"]
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/omnia_tracing"]

omnia_tracing-0.1.0/src/omnia_tracing/__init__.py

@@ -0,0 +1,125 @@
+"""Omnia tracing — standard OpenTelemetry, curated.
+
+One install, one line; eject anytime, your spans don't change. This package
+contains NO instrumentation code of its own: it pins and configures standard,
+ecosystem-maintained OpenTelemetry pieces. The identical setup without this
+package is documented in docs/eject.md.
+"""
+
+from __future__ import annotations
+
+import os
+from dataclasses import dataclass, field
+
+DEFAULT_ENDPOINT = "https://gateway.omnia-voice.com/v1/traces"
+TAG_ATTRIBUTE = "omnia.tag"
+
+__all__ = ["setup", "resolve_config", "ResolvedConfig", "Tracing", "DEFAULT_ENDPOINT", "TAG_ATTRIBUTE"]
+
+
+@dataclass
+class ResolvedConfig:
+    endpoint: str
+    headers: dict[str, str]
+    service_name: str | None
+    resource_attributes: dict[str, str] = field(default_factory=dict)
+
+
+def resolve_config(
+    api_key: str | None = None,
+    tag: str | None = None,
+    service_name: str | None = None,
+    endpoint: str | None = None,
+    env: dict[str, str] | None = None,
+) -> ResolvedConfig:
+    """Pure configuration assembly — unit-testable without starting a pipeline."""
+    e = os.environ if env is None else env
+    key = api_key or e.get("OMNIA_API_KEY")
+    if not key:
+        raise ValueError(
+            "omnia-tracing: no API key. Pass setup(api_key=...) or set OMNIA_API_KEY. "
+            "Refusing to start a tracer that exports nowhere."
+        )
+    resource_attributes: dict[str, str] = {}
+    resolved_tag = tag or e.get("OMNIA_TAG")
+    if resolved_tag:
+        resource_attributes[TAG_ATTRIBUTE] = resolved_tag
+    return ResolvedConfig(
+        endpoint=endpoint or e.get("OMNIA_OTLP_ENDPOINT") or DEFAULT_ENDPOINT,
+        headers={"Authorization": f"Bearer {key}"},
+        service_name=service_name or e.get("OTEL_SERVICE_NAME"),
+        resource_attributes=resource_attributes,
+    )
+
+
+class Tracing:
+    """Handle returned by setup(): shutdown() flushes and stops.
+
+    `instrumented` names the libraries actually being traced — the ones both
+    installed in this environment and successfully instrumented."""
+
+    def __init__(self, provider, instrumented: list[str]) -> None:
+        self._provider = provider
+        self.instrumented = instrumented
+
+    def shutdown(self) -> None:
+        self._provider.shutdown()
+
+
+def setup(
+    api_key: str | None = None,
+    tag: str | None = None,
+    service_name: str | None = None,
+    endpoint: str | None = None,
+) -> Tracing:
+    """Start standard OpenTelemetry tracing, exporting to Omnia.
+
+    Call ONCE, at startup, before constructing LLM clients. Instruments
+    OpenAI, Anthropic, Gemini and LangChain via the ecosystem's standard
+    instrumentation packages.
+    """
+    config = resolve_config(api_key, tag, service_name, endpoint)
+
+    # Imports live here, not module top: `import omnia_tracing` must stay
+    # side-effect free so resolve_config is usable (and testable) alone.
+    from opentelemetry.sdk.resources import Resource
+    from opentelemetry.sdk.trace import TracerProvider
+    from opentelemetry.sdk.trace.export import BatchSpanProcessor
+    from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
+
+    attrs: dict[str, str] = dict(config.resource_attributes)
+    if config.service_name:
+        attrs["service.name"] = config.service_name
+    provider = TracerProvider(resource=Resource.create(attrs))
+    provider.add_span_processor(
+        BatchSpanProcessor(
+            OTLPSpanExporter(endpoint=config.endpoint, headers=config.headers)
+        )
+    )
+
+    # Each instrumentation package imports its TARGET library at import time,
+    # so instrument only what this environment actually has — an
+    # Anthropic-only app must not be forced to install `openai`.
+    candidates = (
+        ("openai", "opentelemetry.instrumentation.openai", "OpenAIInstrumentor"),
+        ("anthropic", "opentelemetry.instrumentation.anthropic", "AnthropicInstrumentor"),
+        (
+            "google-generativeai",
+            "opentelemetry.instrumentation.google_generativeai",
+            "GoogleGenerativeAiInstrumentor",
+        ),
+        ("langchain", "opentelemetry.instrumentation.langchain", "LangchainInstrumentor"),
+    )
+    instrumented: list[str] = []
+    import importlib
+
+    for name, module_path, class_name in candidates:
+        try:
+            instrumentor = getattr(importlib.import_module(module_path), class_name)()
+        except ImportError:
+            continue  # target library not installed — nothing to trace
+        if not instrumentor.is_instrumented_by_opentelemetry:
+            instrumentor.instrument(tracer_provider=provider)
+        instrumented.append(name)
+
+    return Tracing(provider, instrumented)

omnia_tracing-0.1.0/tests/test_config.py

@@ -0,0 +1,44 @@
+import pytest
+
+from omnia_tracing import DEFAULT_ENDPOINT, TAG_ATTRIBUTE, resolve_config
+
+
+def test_refuses_without_api_key():
+    with pytest.raises(ValueError, match="OMNIA_API_KEY"):
+        resolve_config(env={})
+
+
+def test_defaults_to_gateway_with_bearer():
+    c = resolve_config(api_key="sk_x", env={})
+    assert c.endpoint == DEFAULT_ENDPOINT
+    assert c.headers["Authorization"] == "Bearer sk_x"
+
+
+def test_reads_env():
+    c = resolve_config(
+        env={
+            "OMNIA_API_KEY": "sk_env",
+            "OMNIA_TAG": "checkout-agent",
+            "OMNIA_OTLP_ENDPOINT": "https://other.example/v1/traces",
+            "OTEL_SERVICE_NAME": "svc",
+        }
+    )
+    assert c.headers["Authorization"] == "Bearer sk_env"
+    assert c.resource_attributes[TAG_ATTRIBUTE] == "checkout-agent"
+    assert c.endpoint == "https://other.example/v1/traces"
+    assert c.service_name == "svc"
+
+
+def test_options_beat_env():
+    c = resolve_config(
+        api_key="sk_opt",
+        tag="t2",
+        env={"OMNIA_API_KEY": "sk_env", "OMNIA_TAG": "t1"},
+    )
+    assert c.headers["Authorization"] == "Bearer sk_opt"
+    assert c.resource_attributes[TAG_ATTRIBUTE] == "t2"
+
+
+def test_no_tag_means_no_attribute():
+    c = resolve_config(api_key="k", env={})
+    assert c.resource_attributes == {}