opentelemetry-overmind 0.53.0__tar.gz

opentelemetry_overmind-0.53.0/PKG-INFO

@@ -0,0 +1,16 @@
+Metadata-Version: 2.4
+Name: opentelemetry-overmind
+Version: 0.53.0
+Summary: 
+Author: pritamsoni-hsr
+Author-email: 23050213+pritamsoni-hsr@users.noreply.github.com
+Requires-Python: >=3.10,<4
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Description-Content-Type: text/markdown
+
+

opentelemetry_overmind-0.53.0/opentelemetry/overmind/attrs.py

@@ -0,0 +1,35 @@
+class OvermindAttributes:
+    # Prompt attributes
+    GEN_AI_PROMPT_ID = "overmind.prompt.id"
+    GEN_AI_PROMPT_HASH = "overmind.prompt.hash"
+    GEN_AI_PROMPT_VERSION = "overmind.prompt.version"
+    GEN_AI_PROMPT_KWARGS = "overmind.prompt.kwargs"
+    GEN_AI_PROMPT_TEMPLATE = "overmind.prompt.template"
+    GEN_AI_REFERENCE_OUTPUT = "overmind.reference.output"
+    GEN_AI_REASONING_EFFORT = "overmind.reasoning.effort"
+
+    # Request attributes
+    GEN_AI_REQUEST_REASONING_EFFORT = "gen_ai.request.reasoning_effort"
+    GEN_AI_REQUEST_MAX_COMPLETION_TOKENS = "gen_ai.request.max_completion_tokens"
+    GEN_AI_REQUEST_SEED = "gen_ai.request.seed"
+    GEN_AI_REQUEST_STOP_SEQUENCES = "gen_ai.request.stop_sequences"
+    GEN_AI_REQUEST_N = "gen_ai.request.n"
+    GEN_AI_REQUEST_RESPONSE_FORMAT = "gen_ai.request.response_format"
+    GEN_AI_REQUEST_SERVICE_TIER = "gen_ai.request.service_tier"
+    GEN_AI_REQUEST_PARALLEL_TOOL_CALLS = "gen_ai.request.parallel_tool_calls"
+    GEN_AI_REQUEST_TOOL_CHOICE = "gen_ai.request.tool_choice"
+    GEN_AI_REQUEST_LOGPROBS = "gen_ai.request.logprobs"
+    GEN_AI_REQUEST_TOP_LOGPROBS = "gen_ai.request.top_logprobs"
+    GEN_AI_REQUEST_MODALITIES = "gen_ai.request.modalities"
+
+    # Response attributes
+    GEN_AI_RESPONSE_ID = "gen_ai.response.id"
+    GEN_AI_RESPONSE_SYSTEM_FINGERPRINT = "gen_ai.response.system_fingerprint"
+    GEN_AI_RESPONSE_SERVICE_TIER = "gen_ai.response.service_tier"
+
+    # Usage detail attributes
+    GEN_AI_USAGE_REASONING_TOKENS = "gen_ai.usage.reasoning_tokens"
+    GEN_AI_USAGE_CACHED_TOKENS = "gen_ai.usage.cached_tokens"
+    GEN_AI_USAGE_AUDIO_TOKENS = "gen_ai.usage.audio_tokens"
+    GEN_AI_USAGE_ACCEPTED_PREDICTION_TOKENS = "gen_ai.usage.accepted_prediction_tokens"
+    GEN_AI_USAGE_REJECTED_PREDICTION_TOKENS = "gen_ai.usage.rejected_prediction_tokens"

opentelemetry_overmind-0.53.0/opentelemetry/overmind/processor.py

@@ -0,0 +1,106 @@
+import json
+from collections.abc import Mapping, Sequence
+
+from opentelemetry.trace import Span
+from opentelemetry.overmind.prompt import PromptString
+from opentelemetry.overmind.attrs import OvermindAttributes
+
+
+def _store_prompt_attributes(span: Span, content: PromptString | str) -> None:
+    if isinstance(content, PromptString):
+        span.set_attribute(OvermindAttributes.GEN_AI_PROMPT_ID, content.id)
+        span.set_attribute(OvermindAttributes.GEN_AI_PROMPT_HASH, content.hash)
+        span.set_attribute(OvermindAttributes.GEN_AI_PROMPT_KWARGS, json.dumps(content.kwargs))
+        span.set_attribute(OvermindAttributes.GEN_AI_PROMPT_TEMPLATE, content.template)
+
+
+MAX_PROMPT_COUNT = 1
+
+
+def _walk_and_store_prompts(span: Span, value, counter: list[int]) -> None:
+    """
+    Recursively walk an arbitrary value (PromptString, str, list, dict, etc.)
+    and store prompt attributes for any PromptString instances found.
+    """
+    if isinstance(value, PromptString):
+        _store_prompt_attributes(span, value)
+        counter[0] += 1
+        return
+
+    # Plain strings are not traced unless wrapped in PromptString
+    if isinstance(value, str):
+        return
+
+    if isinstance(value, Mapping):
+        for v in value.values():
+            _walk_and_store_prompts(span, v, counter)
+        return
+
+    if isinstance(value, Sequence) and not isinstance(value, (str, bytes, bytearray)):
+        for item in value:
+            _walk_and_store_prompts(span, item, counter)
+        return
+
+
+def clean_kwargs(kwargs: dict) -> dict:
+    """remove additional keys from kwargs, so that the default methods can be used, but returns a copy of the original args so tracing can work"""
+
+    reference_output = kwargs.pop("reference_output", None)
+    return {"reference_output": reference_output, **kwargs}
+
+
+def request_processor(span: Span, kwargs: dict, type: str = "openai.chat") -> None:
+    """
+    Normalize prompt capture across providers.
+
+    Supported types:
+      - openai.chat
+      - openai.responses
+      - anthropic.messages
+      - google.genai.responses
+      - agno.agent.run
+    """
+    if reference_output := kwargs.pop("reference_output", None):
+        span.set_attribute(OvermindAttributes.GEN_AI_REFERENCE_OUTPUT, reference_output)
+
+    prompt_count = [0]
+
+    if type == "openai.chat":
+        # OpenAI chat: classic messages and newer content blocks
+        for message in kwargs.get("messages", []):
+            content = message.get("content")
+            _walk_and_store_prompts(span, content, prompt_count)
+
+        if prompt_count[0] > MAX_PROMPT_COUNT:
+            raise ValueError("Too many prompts")
+
+    elif type == "openai.response":
+        # OpenAI Responses API: instructions + input (and potentially messages/system)
+        if "instructions" in kwargs:
+            _walk_and_store_prompts(span, kwargs["instructions"], prompt_count)
+        if "input" in kwargs:
+            _walk_and_store_prompts(span, kwargs["input"], prompt_count)
+
+    elif type == "openai.embeddings":
+        if "input" in kwargs:
+            _store_prompt_attributes(span, kwargs["input"])
+
+    elif type == "anthropic.messages":
+        # Anthropic messages API (including beta and Bedrock)
+        if "system" in kwargs:
+            _walk_and_store_prompts(span, kwargs["system"], prompt_count)
+        for message in kwargs.get("messages", []):
+            _walk_and_store_prompts(span, message.get("content"), prompt_count)
+
+    elif type == "google.genai.responses":
+        # Google Generative AI / Gemini generate_content
+        if "system_instruction" in kwargs:
+            _walk_and_store_prompts(span, kwargs["system_instruction"], prompt_count)
+        if "contents" in kwargs:
+            _walk_and_store_prompts(span, kwargs["contents"], prompt_count)
+
+    elif type == "agno.agent.run":
+        # Agno Agent / Team run entrypoints
+        for key in ("prompt", "system_prompt", "messages", "input"):
+            if key in kwargs:
+                _walk_and_store_prompts(span, kwargs[key], prompt_count)

opentelemetry_overmind-0.53.0/opentelemetry/overmind/prompt.py

@@ -0,0 +1,53 @@
+import hashlib
+
+
+class PromptString(str):
+    id: str
+    template: str
+    kwargs: dict
+    hash: str
+    traceable: bool
+
+    def __new__(
+        cls,
+        *args,
+        id: str | None = None,
+        template: str | None = None,
+        kwargs: dict | None = None,
+        **_extra,
+    ):
+        """
+        Supports two construction patterns:
+        1) Our explicit usage:
+           PromptString(id=\"...\", template=\"...\", kwargs={...})
+        2) Library/internal usage (e.g. Agno) that may call:
+           PromptString(\"prompt text\", ...)
+        """
+        if id is not None and template is not None:
+            # Explicit, structured construction with template formatting.
+            kwargs = kwargs or {}
+            prompt = template.format(**kwargs)
+            traceable = True
+        elif args:
+            # Fallback: treat first positional arg as the already-formatted prompt.
+            base_prompt = str(args[0])
+            prompt = base_prompt
+            id = id or "prompt"
+            template = template or base_prompt
+            kwargs = kwargs or {}
+            traceable = False
+        else:
+            # Extremely defensive fallback – empty prompt.
+            prompt = ""
+            id = id or "prompt"
+            template = template or ""
+            kwargs = kwargs or {}
+            traceable = False
+
+        obj = super().__new__(cls, prompt)
+        obj.id = id
+        obj.template = template
+        obj.kwargs = kwargs
+        obj.hash = hashlib.md5(template.encode()).hexdigest() if template else ""
+        obj.traceable = traceable
+        return obj

opentelemetry_overmind-0.53.0/pyproject.toml

@@ -0,0 +1,15 @@
+[project]
+authors = [
+    { name = "pritamsoni-hsr", email = "23050213+pritamsoni-hsr@users.noreply.github.com" },
+]
+name = "opentelemetry-overmind"
+readme = "README.md"
+requires-python = ">=3.10,<4"
+version = "0.53.0"
+
+[tool.poetry]
+packages = [{ include = "opentelemetry/overmind" }]
+
+[build-system]
+build-backend = "poetry.core.masonry.api"
+requires      = ["poetry-core>=2.0.0,<3.0.0"]