Tox ruff fixes

Author: nagkumar91

instrumentation-genai/opentelemetry-instrumentation-openai-agents/examples/basic_usage.py

@@ -5,14 +5,14 @@
 
 
 from __future__ import annotations as _annotations
-from opentelemetry.instrumentation.openai_agents import OpenAIAgentsInstrumentor
 
 import asyncio
+
+#######################################
+import os
 import random
 import uuid
 
-from pydantic import BaseModel
-
 from agents import (
     Agent,
     HandoffOutputItem,
@@ -25,23 +25,24 @@
     TResponseInputItem,
     function_tool,
     handoff,
+    set_default_openai_client,
     trace,
 )
 from agents.extensions.handoff_prompt import RECOMMENDED_PROMPT_PREFIX
-
-#######################################
-
-import asyncio, os
-from openai import AsyncAzureOpenAI
-from agents import Agent, Runner, function_tool
-from agents import set_default_openai_client
-from agents import Agent, Runner, trace
-from pydantic import BaseModel
 from azure.monitor.opentelemetry import configure_azure_monitor
 from dotenv import load_dotenv
+from openai import AsyncAzureOpenAI
+from pydantic import BaseModel
+
+from opentelemetry.instrumentation.openai_agents import (
+    OpenAIAgentsInstrumentor,
+)
+
 load_dotenv()
 
-configure_azure_monitor(connection_string=os.environ["APPLICATION_INSIGHTS_CONNECTION_STRING"])
+configure_azure_monitor(
+    connection_string=os.environ["APPLICATION_INSIGHTS_CONNECTION_STRING"]
+)
 
 OpenAIAgentsInstrumentor().instrument()
 
@@ -59,6 +60,7 @@
 
 ### CONTEXT
 
+
 class AirlineAgentContext(BaseModel):
     passenger_name: str | None = None
     confirmation_number: str | None = None
@@ -70,7 +72,8 @@ class AirlineAgentContext(BaseModel):
 
 
 @function_tool(
-    name_override="faq_lookup_tool", description_override="Lookup frequently asked questions."
+    name_override="faq_lookup_tool",
+    description_override="Lookup frequently asked questions.",
 )
 async def faq_lookup_tool(question: str) -> str:
     if "bag" in question or "baggage" in question:
@@ -92,7 +95,9 @@ async def faq_lookup_tool(question: str) -> str:
 
 @function_tool
 async def update_seat(
-    context: RunContextWrapper[AirlineAgentContext], confirmation_number: str, new_seat: str
+    context: RunContextWrapper[AirlineAgentContext],
+    confirmation_number: str,
+    new_seat: str,
 ) -> str:
     """
     Update the seat for a given confirmation number.
@@ -105,14 +110,18 @@ async def update_seat(
     context.context.confirmation_number = confirmation_number
     context.context.seat_number = new_seat
     # Ensure that the flight number has been set by the incoming handoff
-    assert context.context.flight_number is not None, "Flight number is required"
+    assert (
+        context.context.flight_number is not None
+    ), "Flight number is required"
     return f"Updated seat to {new_seat} for confirmation number {confirmation_number}"
 
 
 ### HOOKS
 
 
-async def on_seat_booking_handoff(context: RunContextWrapper[AirlineAgentContext]) -> None:
+async def on_seat_booking_handoff(
+    context: RunContextWrapper[AirlineAgentContext],
+) -> None:
     flight_number = f"FLT-{random.randint(100, 999)}"
     context.context.flight_number = flight_number
 
@@ -179,12 +188,16 @@ async def main():
         user_input = input("Enter your message: ")
         with trace("Customer service", group_id=conversation_id):
             input_items.append({"content": user_input, "role": "user"})
-            result = await Runner.run(current_agent, input_items, context=context)
+            result = await Runner.run(
+                current_agent, input_items, context=context
+            )
 
             for new_item in result.new_items:
                 agent_name = new_item.agent.name
                 if isinstance(new_item, MessageOutputItem):
-                    print(f"{agent_name}: {ItemHelpers.text_message_output(new_item)}")
+                    print(
+                        f"{agent_name}: {ItemHelpers.text_message_output(new_item)}"
+                    )
                 elif isinstance(new_item, HandoffOutputItem):
                     print(
                         f"Handed off from {new_item.source_agent.name} to {new_item.target_agent.name}"
@@ -194,7 +207,9 @@ async def main():
                 elif isinstance(new_item, ToolCallOutputItem):
                     print(f"{agent_name}: Tool call output: {new_item.output}")
                 else:
-                    print(f"{agent_name}: Skipping item: {new_item.__class__.__name__}")
+                    print(
+                        f"{agent_name}: Skipping item: {new_item.__class__.__name__}"
+                    )
             input_items = result.to_input_list()
             current_agent = result.last_agent
 

instrumentation-genai/opentelemetry-instrumentation-openai-agents/examples/mcp_hotel_capture_all.py

@@ -11,16 +11,26 @@
 
 import azure.identity
 import openai
-from agents import Agent, OpenAIChatCompletionsModel, Runner, set_tracing_disabled
+from agents import (
+    Agent,
+    OpenAIChatCompletionsModel,
+    Runner,
+    set_tracing_disabled,
+)
 from agents.mcp.server import MCPServerStreamableHttp
 from agents.model_settings import ModelSettings
 from dotenv import load_dotenv
 
 from opentelemetry import trace
-from opentelemetry.instrumentation.openai_agents import OpenAIAgentsInstrumentor
+from opentelemetry.instrumentation.openai_agents import (
+    OpenAIAgentsInstrumentor,
+)
 from opentelemetry.sdk.resources import Resource
 from opentelemetry.sdk.trace import TracerProvider
-from opentelemetry.sdk.trace.export import BatchSpanProcessor, ConsoleSpanExporter
+from opentelemetry.sdk.trace.export import (
+    BatchSpanProcessor,
+    ConsoleSpanExporter,
+)
 
 load_dotenv(override=True)
 
@@ -54,7 +64,9 @@ def _set_capture_env(provider: str, base_url: str) -> None:
             "OTEL_GENAI_AGENT_DESCRIPTION",
             "Agent orchestrator that uses MCP to find hotels matching user criteria",
         ),
-        "OTEL_GENAI_AGENT_ID": os.getenv("OTEL_GENAI_AGENT_ID", "mcp-hotel-assistant"),
+        "OTEL_GENAI_AGENT_ID": os.getenv(
+            "OTEL_GENAI_AGENT_ID", "mcp-hotel-assistant"
+        ),
     }
     for env_key, value in capture_defaults.items():
         os.environ.setdefault(env_key, value)
@@ -115,7 +127,9 @@ def _build_client() -> openai.AsyncAzureOpenAI:
         )
 
     if host == "ollama":
-        base_url = os.getenv("OLLAMA_BASE_URL", "http://localhost:11434/v1").rstrip("/")
+        base_url = os.getenv(
+            "OLLAMA_BASE_URL", "http://localhost:11434/v1"
+        ).rstrip("/")
         model_name = os.getenv("OLLAMA_MODEL", "llama3.1:latest")
         api_key = os.getenv("OLLAMA_API_KEY", "none")
 
@@ -167,7 +181,9 @@ def _configure_otel() -> None:
             )
             print("[otel] Azure Monitor trace exporter configured")
     else:
-        tracer_provider.add_span_processor(BatchSpanProcessor(ConsoleSpanExporter()))
+        tracer_provider.add_span_processor(
+            BatchSpanProcessor(ConsoleSpanExporter())
+        )
         print("[otel] Console span exporter configured")
         print(
             "[otel] Set APPLICATION_INSIGHTS_CONNECTION_STRING to export to "
@@ -198,7 +214,9 @@ def _configure_otel() -> None:
     name="Assistant",
     instructions="Use the tools to achieve the task",
     mcp_servers=[mcp_server],
-    model=OpenAIChatCompletionsModel(model=api_config.model_name, openai_client=client),
+    model=OpenAIChatCompletionsModel(
+        model=api_config.model_name, openai_client=client
+    ),
     model_settings=ModelSettings(tool_choice="required"),
 )
 
@@ -211,7 +229,9 @@ async def main() -> None:
     tracer = trace.get_tracer(__name__)
     await mcp_server.connect()
     try:
-        with tracer.start_as_current_span(_root_span_name(api_config.provider)) as span:
+        with tracer.start_as_current_span(
+            _root_span_name(api_config.provider)
+        ) as span:
             message = (
                 "Find me a hotel in San Francisco for 2 nights starting from 2024-01-01. "
                 "I need a hotel with free WiFi and a pool."

instrumentation-genai/opentelemetry-instrumentation-openai-agents/examples/multi_agent_weather_capture_all.py

@@ -14,10 +14,15 @@
 from dotenv import load_dotenv
 
 from opentelemetry import trace
-from opentelemetry.instrumentation.openai_agents import OpenAIAgentsInstrumentor
+from opentelemetry.instrumentation.openai_agents import (
+    OpenAIAgentsInstrumentor,
+)
 from opentelemetry.sdk.resources import Resource
 from opentelemetry.sdk.trace import TracerProvider
-from opentelemetry.sdk.trace.export import BatchSpanProcessor, ConsoleSpanExporter
+from opentelemetry.sdk.trace.export import (
+    BatchSpanProcessor,
+    ConsoleSpanExporter,
+)
 
 load_dotenv(override=True)
 
@@ -51,7 +56,9 @@ def _set_capture_env(provider: str, base_url: str) -> None:
             "OTEL_GENAI_AGENT_DESCRIPTION",
             "Multi-agent weather service routing requests to language-specific forecasters",
         ),
-        "OTEL_GENAI_AGENT_ID": os.getenv("OTEL_GENAI_AGENT_ID", "weather-orchestrator"),
+        "OTEL_GENAI_AGENT_ID": os.getenv(
+            "OTEL_GENAI_AGENT_ID", "weather-orchestrator"
+        ),
     }
     for env_key, value in capture_defaults.items():
         os.environ.setdefault(env_key, value)
@@ -112,7 +119,9 @@ def _build_client() -> openai.AsyncAzureOpenAI:
         )
 
     if host == "ollama":
-        base_url = os.getenv("OLLAMA_BASE_URL", "http://localhost:11434/v1").rstrip("/")
+        base_url = os.getenv(
+            "OLLAMA_BASE_URL", "http://localhost:11434/v1"
+        ).rstrip("/")
         model_name = os.getenv("OLLAMA_MODEL", "llama3.1:latest")
         api_key = os.getenv("OLLAMA_API_KEY", "none")
 
@@ -164,7 +173,9 @@ def _configure_otel() -> None:
             )
             print("[otel] Azure Monitor trace exporter configured")
     else:
-        tracer_provider.add_span_processor(BatchSpanProcessor(ConsoleSpanExporter()))
+        tracer_provider.add_span_processor(
+            BatchSpanProcessor(ConsoleSpanExporter())
+        )
         print("[otel] Console span exporter configured")
         print(
             "[otel] Set APPLICATION_INSIGHTS_CONNECTION_STRING to export to "
@@ -198,21 +209,27 @@ def get_weather(city: str) -> dict[str, object]:
     name="Spanish agent",
     instructions="You only speak Spanish.",
     tools=[get_weather],
-    model=OpenAIChatCompletionsModel(model=api_config.model_name, openai_client=client),
+    model=OpenAIChatCompletionsModel(
+        model=api_config.model_name, openai_client=client
+    ),
 )
 
 english_agent = Agent(
     name="English agent",
     instructions="You only speak English.",
     tools=[get_weather],
-    model=OpenAIChatCompletionsModel(model=api_config.model_name, openai_client=client),
+    model=OpenAIChatCompletionsModel(
+        model=api_config.model_name, openai_client=client
+    ),
 )
 
 triage_agent = Agent(
     name="Triage agent",
     instructions="Handoff to the appropriate agent based on the language of the request.",
     handoffs=[spanish_agent, english_agent],
-    model=OpenAIChatCompletionsModel(model=api_config.model_name, openai_client=client),
+    model=OpenAIChatCompletionsModel(
+        model=api_config.model_name, openai_client=client
+    ),
 )
 
 
@@ -222,8 +239,12 @@ def _root_span_name(provider: str) -> str:
 
 async def main() -> None:
     tracer = trace.get_tracer(__name__)
-    with tracer.start_as_current_span(_root_span_name(api_config.provider)) as span:
-        user_request = "Hola, ¿cómo estás? ¿Puedes darme el clima para San Francisco CA?"
+    with tracer.start_as_current_span(
+        _root_span_name(api_config.provider)
+    ) as span:
+        user_request = (
+            "Hola, ¿cómo estás? ¿Puedes darme el clima para San Francisco CA?"
+        )
 
         span.set_attribute("user.request", user_request)
         span.set_attribute("api.host", os.getenv("API_HOST", "github"))