openlit 1.34.31__py3-none-any.whl → 1.34.32__py3-none-any.whl

openlit/instrumentation/ag2/utils.py

@@ -147,7 +147,7 @@ def common_agent_logic(
     scope._span.set_status(Status(StatusCode.OK))
 
     # Metrics
-    if not disable_metrics and hasattr(scope, "_input_tokens"):
+    if not disable_metrics and metrics is not None and hasattr(scope, "_input_tokens"):
         record_completion_metrics(
             metrics,
             operation_type,
@@ -196,7 +196,7 @@ def process_agent_creation(
     scope._end_time = time.time()
     scope._span = span
     scope._agent_name = agent_name
-    scope._request_model = llm_config.get("model", "gpt-4o")
+    scope._request_model = llm_config.get("model", "unknown")
     scope._response_model = scope._request_model
     scope._system_message = system_message
     scope._server_address, scope._server_port = server_address, server_port
@@ -276,3 +276,344 @@ def process_agent_run(
     )
 
     return response
+
+
+def process_agent_generate_reply(
+    response,
+    agent_name,
+    request_model,
+    messages,
+    sender,
+    pricing_info,
+    server_port,
+    server_address,
+    environment,
+    application_name,
+    metrics,
+    start_time,
+    span,
+    capture_message_content=False,
+    disable_metrics=False,
+    version="1.0.0",
+    **kwargs,
+):
+    """
+    Process agent generate_reply and generate Telemetry
+    """
+
+    # Create scope object
+    scope = type("GenericScope", (), {})()
+
+    scope._start_time = start_time
+    scope._end_time = time.time()
+    scope._span = span
+    scope._agent_name = agent_name
+    scope._request_model = request_model
+    scope._response_model = request_model
+    scope._server_address, scope._server_port = server_address, server_port
+    scope._messages = messages
+    scope._sender_name = getattr(sender, "name", "Unknown") if sender else "Unknown"
+
+    # Set agent-specific attributes
+    span.set_attribute(SemanticConvention.GEN_AI_AGENT_MESSAGE_TYPE, "generate_reply")
+    span.set_attribute(SemanticConvention.GEN_AI_AGENT_SENDER, scope._sender_name)
+
+    # Process response content
+    if response and isinstance(response, str):
+        scope._response_content = response
+    elif response and hasattr(response, "content"):
+        scope._response_content = response.content
+    else:
+        scope._response_content = str(response) if response else ""
+
+    # Try to extract token information if available
+    try:
+        # Mock token calculation for generate_reply
+        scope._input_tokens = len(str(messages)) // 4 if messages else 0
+        scope._output_tokens = len(scope._response_content) // 4
+        scope._cost = get_chat_model_cost(
+            request_model, pricing_info, scope._input_tokens, scope._output_tokens
+        )
+    except Exception:
+        scope._input_tokens = 0
+        scope._output_tokens = 0
+        scope._cost = 0.0
+
+    common_agent_logic(
+        scope,
+        pricing_info,
+        environment,
+        application_name,
+        metrics,
+        capture_message_content,
+        disable_metrics,
+        version,
+        SemanticConvention.GEN_AI_OPERATION_TYPE_CHAT,
+    )
+
+    return response
+
+
+def process_agent_receive(
+    message,
+    agent_name,
+    sender_name,
+    agent_instance,
+    pricing_info,
+    server_port,
+    server_address,
+    environment,
+    application_name,
+    metrics,
+    start_time,
+    span,
+    capture_message_content=False,
+    disable_metrics=False,
+    version="1.0.0",
+    **kwargs,
+):
+    """
+    Process agent receive and generate Telemetry
+    """
+
+    # Create scope object
+    scope = type("GenericScope", (), {})()
+
+    scope._start_time = start_time
+    scope._end_time = time.time()
+    scope._span = span
+    scope._agent_name = agent_name
+    scope._sender_name = sender_name
+    scope._server_address, scope._server_port = server_address, server_port
+    scope._message = message
+
+    # Extract model from agent instance
+    if hasattr(agent_instance, "llm_config") and isinstance(
+        agent_instance.llm_config, dict
+    ):
+        scope._request_model = agent_instance.llm_config.get("model", "unknown")
+    else:
+        scope._request_model = "unknown"
+    scope._response_model = scope._request_model
+
+    # Set agent-specific attributes
+    span.set_attribute(SemanticConvention.GEN_AI_AGENT_MESSAGE_TYPE, "receive")
+    span.set_attribute(SemanticConvention.GEN_AI_AGENT_SENDER, sender_name)
+
+    # Content capture for received message
+    if capture_message_content:
+        span.set_attribute(SemanticConvention.GEN_AI_CONTENT_PROMPT, str(message))
+
+    common_agent_logic(
+        scope,
+        pricing_info,
+        environment,
+        application_name,
+        metrics,
+        capture_message_content,
+        disable_metrics,
+        version,
+        SemanticConvention.GEN_AI_OPERATION_TYPE_CHAT,
+    )
+
+
+def process_agent_send(
+    message,
+    agent_name,
+    recipient_name,
+    agent_instance,
+    pricing_info,
+    server_port,
+    server_address,
+    environment,
+    application_name,
+    metrics,
+    start_time,
+    span,
+    capture_message_content=False,
+    disable_metrics=False,
+    version="1.0.0",
+    **kwargs,
+):
+    """
+    Process agent send and generate Telemetry
+    """
+
+    # Create scope object
+    scope = type("GenericScope", (), {})()
+
+    scope._start_time = start_time
+    scope._end_time = time.time()
+    scope._span = span
+    scope._agent_name = agent_name
+    scope._recipient_name = recipient_name
+    scope._server_address, scope._server_port = server_address, server_port
+    scope._message = message
+
+    # Extract model from agent instance
+    if hasattr(agent_instance, "llm_config") and isinstance(
+        agent_instance.llm_config, dict
+    ):
+        scope._request_model = agent_instance.llm_config.get("model", "unknown")
+    else:
+        scope._request_model = "unknown"
+    scope._response_model = scope._request_model
+
+    # Set agent-specific attributes
+    span.set_attribute(SemanticConvention.GEN_AI_AGENT_MESSAGE_TYPE, "send")
+    span.set_attribute(SemanticConvention.GEN_AI_AGENT_RECIPIENT, recipient_name)
+
+    # Content capture for sent message
+    if capture_message_content:
+        span.set_attribute(SemanticConvention.GEN_AI_CONTENT_COMPLETION, str(message))
+
+    common_agent_logic(
+        scope,
+        pricing_info,
+        environment,
+        application_name,
+        metrics,
+        capture_message_content,
+        disable_metrics,
+        version,
+        SemanticConvention.GEN_AI_OPERATION_TYPE_CHAT,
+    )
+
+
+def process_groupchat_operation(
+    group_name,
+    participants,
+    messages,
+    sender,
+    max_turns,
+    request_model,
+    pricing_info,
+    server_port,
+    server_address,
+    environment,
+    application_name,
+    metrics,
+    start_time,
+    span,
+    capture_message_content=False,
+    disable_metrics=False,
+    version="1.0.0",
+    **kwargs,
+):
+    """
+    Process GroupChat operation and generate Telemetry
+    """
+
+    # Create scope object
+    scope = type("GenericScope", (), {})()
+
+    scope._start_time = start_time
+    scope._end_time = time.time()
+    scope._span = span
+    scope._group_name = group_name
+    scope._participants = participants
+    scope._server_address, scope._server_port = server_address, server_port
+    scope._sender_name = getattr(sender, "name", "Unknown") if sender else "Unknown"
+
+    # Add required model attributes for common_agent_logic
+    scope._request_model = request_model
+    scope._response_model = request_model
+
+    # Set agent name for groupchat
+    scope._agent_name = group_name
+
+    # Set GroupChat-specific attributes
+    span.set_attribute(
+        SemanticConvention.GEN_AI_GROUPCHAT_PARTICIPANTS, ",".join(participants)
+    )
+    span.set_attribute(
+        SemanticConvention.GEN_AI_WORKFLOW_AGENT_COUNT, len(participants)
+    )
+    span.set_attribute(SemanticConvention.GEN_AI_WORKFLOW_EXECUTION_TYPE, "groupchat")
+
+    if max_turns:
+        span.set_attribute(SemanticConvention.GEN_AI_GROUPCHAT_TURN_COUNT, max_turns)
+
+    # Content capture for GroupChat
+    if capture_message_content and messages:
+        span.set_attribute(SemanticConvention.GEN_AI_CONTENT_PROMPT, str(messages))
+
+    # Use framework operation type for GroupChat
+    common_agent_logic(
+        scope,
+        pricing_info,
+        environment,
+        application_name,
+        metrics,
+        capture_message_content,
+        disable_metrics,
+        version,
+        SemanticConvention.GEN_AI_OPERATION_TYPE_FRAMEWORK,
+    )
+
+
+def process_speaker_selection(
+    last_speaker,
+    selected_speaker,
+    selector,
+    agents,
+    request_model,
+    pricing_info,
+    server_port,
+    server_address,
+    environment,
+    application_name,
+    metrics,
+    start_time,
+    span,
+    capture_message_content=False,
+    disable_metrics=False,
+    version="1.0.0",
+    **kwargs,
+):
+    """
+    Process speaker selection and generate Telemetry
+    """
+
+    # Create scope object
+    scope = type("GenericScope", (), {})()
+
+    scope._start_time = start_time
+    scope._end_time = time.time()
+    scope._span = span
+    scope._last_speaker = last_speaker
+    scope._selected_speaker = selected_speaker
+    scope._server_address, scope._server_port = server_address, server_port
+
+    # Add required model attributes for common_agent_logic
+    scope._request_model = request_model
+    scope._response_model = request_model
+
+    # Set agent name for speaker selection
+    scope._agent_name = "speaker_selection"
+
+    # Set speaker selection attributes
+    span.set_attribute(
+        SemanticConvention.GEN_AI_GROUPCHAT_SPEAKER_SELECTION, selected_speaker
+    )
+    span.set_attribute(SemanticConvention.GEN_AI_AGENT_SENDER, last_speaker)
+
+    if selector:
+        span.set_attribute(SemanticConvention.GEN_AI_AGENT_ROLE, "selector")
+
+    # Set agent count
+    if agents:
+        span.set_attribute(SemanticConvention.GEN_AI_WORKFLOW_AGENT_COUNT, len(agents))
+
+    # Use agent operation type for speaker selection
+    common_agent_logic(
+        scope,
+        pricing_info,
+        environment,
+        application_name,
+        metrics,
+        capture_message_content,
+        disable_metrics,
+        version,
+        SemanticConvention.GEN_AI_OPERATION_TYPE_AGENT,
+    )

openlit/semcov/__init__.py

@@ -498,3 +498,13 @@ class SemanticConvention:
     # Standard Task Attributes (framework-agnostic)
     GEN_AI_TASK_DESCRIPTION = "gen_ai.task.description"
     GEN_AI_TASK_EXPECTED_OUTPUT = "gen_ai.task.expected_output"
+
+    GEN_AI_GROUPCHAT_PARTICIPANTS = "gen_ai.groupchat.participants"
+    GEN_AI_GROUPCHAT_SPEAKER_SELECTION = "gen_ai.groupchat.speaker_selection"
+    GEN_AI_GROUPCHAT_MESSAGE_COUNT = "gen_ai.groupchat.message_count"
+    GEN_AI_GROUPCHAT_TURN_COUNT = "gen_ai.groupchat.turn_count"
+
+    GEN_AI_AGENT_RECIPIENT = "gen_ai.agent.recipient"
+    GEN_AI_AGENT_SENDER = "gen_ai.agent.sender"
+    GEN_AI_AGENT_MESSAGE_TYPE = "gen_ai.agent.message_type"
+    GEN_AI_AGENT_REPLY_MODE = "gen_ai.agent.reply_mode"

{openlit-1.34.31.dist-info → openlit-1.34.32.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: openlit
-Version: 1.34.31
+Version: 1.34.32
 Summary: OpenTelemetry-native Auto instrumentation library for monitoring LLM Applications and GPUs, facilitating the integration of observability into your GenAI-driven projects
 License: Apache-2.0
 Keywords: OpenTelemetry,otel,otlp,llm,tracing,openai,anthropic,claude,cohere,llm monitoring,observability,monitoring,gpt,Generative AI,chatGPT,gpu

{openlit-1.34.31.dist-info → openlit-1.34.32.dist-info}/RECORD

@@ -1,5 +1,5 @@
 openlit/__helpers.py,sha256=KovxaOSxiDeVW63IDZ1CtryKE65ij2kjuYTSvCLCypI,20988
-openlit/__init__.py,sha256=eZTnbu_X9tQeln3vsmHv99QSsciLB7jrOZCSljSJvDg,18538
+openlit/__init__.py,sha256=0opC1ShHgX7U27JHZC5XEbo2JPwO2jFlUPyAjSLb4Vo,18672
 openlit/_instrumentors.py,sha256=jqGvYkvMc5WdZdhhHn6BjzEJsZnCxkHkgMpQ7mbRDXE,5746
 openlit/evals/__init__.py,sha256=nJe99nuLo1b5rf7pt9U9BCdSDedzbVi2Fj96cgl7msM,380
 openlit/evals/all.py,sha256=X1RXC6A-TVlKvUX-4ZSafiu1B4C_40zzXGqvcb0a0kU,7310
@@ -13,10 +13,10 @@ openlit/guard/prompt_injection.py,sha256=MlX4nWqBRfRUCkoIsfv_OMazJnFyKDZJYzMTM1R
 openlit/guard/restrict_topic.py,sha256=ElQ-vJVeGyS2qqsIsWM72nmtRmodVO1BJimXv3vwfU0,6756
 openlit/guard/sensitive_topic.py,sha256=aaaeEAj6_tcV9wMmGpsFPypDHTbbXrRof7NtWGazuSs,5676
 openlit/guard/utils.py,sha256=K7A5LIDHM3ctsLYfo-jHuoByMuCPKK1llJDjZBpM8L4,8108
-openlit/instrumentation/ag2/__init__.py,sha256=KoSuHsSOnoVY7n3tmC5IB1JQyxaEZf605V2qfVXTm1s,2132
-openlit/instrumentation/ag2/ag2.py,sha256=LEhuPSoTtNnqkOwjzjv2Y57CDjgAhShYBQ_Wc9F1ukw,4077
-openlit/instrumentation/ag2/async_ag2.py,sha256=GP4fmapsR6pWpld7KyY9jOvj9OfE_J4sXFyhG5uZDN4,4153
-openlit/instrumentation/ag2/utils.py,sha256=IqOd6Y_EwhARla0N-4vLiM2oY4UVTIU4LoYgC0ycnHI,7803
+openlit/instrumentation/ag2/__init__.py,sha256=SxxQ3CIpCiVA1G2_nRvd5F7gqCnN5ButIKzrbSUFA30,4617
+openlit/instrumentation/ag2/ag2.py,sha256=ehRRyVpz7IbMTO1jQqP6rEf5YHLSDbo0lq0r_BhvL0w,17816
+openlit/instrumentation/ag2/async_ag2.py,sha256=FXnnt38LyNQjv0ha-QbgXSOAjZYN0UKuiYVe8B2Mjfo,18064
+openlit/instrumentation/ag2/utils.py,sha256=qmbgt0g3_YwcL9pjTrtSc-G3KOobyUMAr30SQ0o2Kd0,17086
 openlit/instrumentation/ai21/__init__.py,sha256=RyJdKcjcfEkRjECi02M7lApGmEu-OhbIfZwP8XI4oT8,3008
 openlit/instrumentation/ai21/ai21.py,sha256=prcC0bV6hlET72qQ00uswTGixKVfu-b8B9_EpLhpk_g,6685
 openlit/instrumentation/ai21/async_ai21.py,sha256=nr_Gaw6BLqredXzq_kcQpxDntYTCYFEnIXouBA3bZGo,6811
@@ -159,8 +159,8 @@ openlit/instrumentation/vllm/vllm.py,sha256=zdzKUkQYmpFlOQ8rObzRiVZEyHOaJFxUagwC
 openlit/otel/events.py,sha256=iOyKIYHA-QYq5bnHVTV_JKeGC5Tsi1icc5nOa3Km_fA,3825
 openlit/otel/metrics.py,sha256=ipH2NB65yOG7yGB32vqmMQ5HjWSeKAk3q4hzGRXRBOs,7238
 openlit/otel/tracing.py,sha256=6KZc-Yubq-S7wEPshUiMNFkw8XN5WvrqIovWaq3gsKw,3301
-openlit/semcov/__init__.py,sha256=2KET4cIKtoYfeSAS0O8waYA5Of-hwYX7HvJyxU-1lt0,22518
-openlit-1.34.31.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-openlit-1.34.31.dist-info/METADATA,sha256=W2HpGFrGhQjZtGyzVP4eGzpCBMYXtbAB8ZNIgpn6B-g,23552
-openlit-1.34.31.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-openlit-1.34.31.dist-info/RECORD,,
+openlit/semcov/__init__.py,sha256=d3XrCplVR-Bry4sx8AEjkvSvJIPtRds5StZhcCRr02s,23018
+openlit-1.34.32.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+openlit-1.34.32.dist-info/METADATA,sha256=dsRHoblJ0gDndZbgGKm6lBmU0BOpLJt6X-edRYWQzE4,23552
+openlit-1.34.32.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+openlit-1.34.32.dist-info/RECORD,,