refactor: clean up whitespace and improve code consistency in OpenAI tracing examples

Author: viniciusdsmello

examples/tracing/openai/responses_api_example.py

@@ -14,23 +14,25 @@
 import openai
 from openlayer.lib import trace_openai, trace_async_openai
 
+
 def setup_environment():
     """Set up environment variables for the example."""
     # OpenAI API key
     os.environ["OPENAI_API_KEY"] = "your-openai-api-key-here"
-    
+
     # Openlayer configuration
     os.environ["OPENLAYER_API_KEY"] = "your-openlayer-api-key-here"
     os.environ["OPENLAYER_INFERENCE_PIPELINE_ID"] = "your-pipeline-id-here"
 
+
 def chat_completions_example():
     """Example using the traditional Chat Completions API with tracing."""
     print("=== Chat Completions API Example ===")
-    
+
     # Create and trace OpenAI client
     client = openai.OpenAI()
     traced_client = trace_openai(client)
-    
+
     # Use Chat Completions API normally - tracing happens automatically
     response = traced_client.chat.completions.create(
         model="gpt-4o-mini",
@@ -41,23 +43,24 @@ def chat_completions_example():
         temperature=0.7,
         max_tokens=100,
     )
-    
+
     print(f"Chat Completion Response: {response.choices[0].message.content}")
     print("✓ Chat Completions API call traced successfully")
 
+
 def responses_api_example():
     """Example using the new Responses API with tracing."""
     print("\n=== Responses API Example ===")
-    
+
     # Create and trace OpenAI client
     client = openai.OpenAI()
     traced_client = trace_openai(client)
-    
+
     # Check if Responses API is available
-    if not hasattr(traced_client, 'responses'):
+    if not hasattr(traced_client, "responses"):
         print("⚠️ Responses API not available in this OpenAI client version")
         return
-    
+
     # Use Responses API with different parameter format
     response = traced_client.responses.create(
         model="gpt-4o-mini",
@@ -66,19 +69,20 @@ def responses_api_example():
         max_output_tokens=50,
         temperature=0.5,
     )
-    
+
     # Note: The actual response structure depends on OpenAI's implementation
     print(f"Responses API Response: {response}")
     print("✓ Responses API call traced successfully")
 
+
 def streaming_chat_completions_example():
     """Example using streaming Chat Completions API with tracing."""
     print("\n=== Streaming Chat Completions Example ===")
-    
+
     # Create and trace OpenAI client
     client = openai.OpenAI()
     traced_client = trace_openai(client)
-    
+
     # Streaming chat completion
     stream = traced_client.chat.completions.create(
         model="gpt-4o-mini",
@@ -88,35 +92,36 @@ def streaming_chat_completions_example():
         stream=True,
         temperature=0.7,
     )
-    
+
     print("Streaming response: ", end="", flush=True)
     for chunk in stream:
         if chunk.choices[0].delta.content is not None:
             print(chunk.choices[0].delta.content, end="", flush=True)
     print()
     print("✓ Streaming Chat Completions call traced successfully")
 
+
 def streaming_responses_api_example():
     """Example using streaming Responses API with tracing."""
     print("\n=== Streaming Responses API Example ===")
-    
+
     # Create and trace OpenAI client
     client = openai.OpenAI()
     traced_client = trace_openai(client)
-    
+
     # Check if Responses API is available
-    if not hasattr(traced_client, 'responses'):
+    if not hasattr(traced_client, "responses"):
         print("⚠️ Responses API not available in this OpenAI client version")
         return
-    
+
     # Streaming responses
     stream = traced_client.responses.create(
         model="gpt-4o-mini",
         input="Tell me a short joke about programming.",
         stream=True,
         max_output_tokens=100,
     )
-    
+
     print("Streaming response: ", end="", flush=True)
     for event in stream:
         # Handle different types of response stream events
@@ -125,33 +130,31 @@ def streaming_responses_api_example():
     print()
     print("✓ Streaming Responses API call traced successfully")
 
+
 def function_calling_example():
     """Example using function calling with both APIs."""
     print("\n=== Function Calling Example ===")
-    
+
     # Create and trace OpenAI client
     client = openai.OpenAI()
     traced_client = trace_openai(client)
-    
+
     # Define a simple function
-    tools = [{
-        "type": "function",
-        "function": {
-            "name": "get_weather",
-            "description": "Get the current weather for a location",
-            "parameters": {
-                "type": "object",
-                "properties": {
-                    "location": {
-                        "type": "string",
-                        "description": "City name"
-                    }
+    tools = [
+        {
+            "type": "function",
+            "function": {
+                "name": "get_weather",
+                "description": "Get the current weather for a location",
+                "parameters": {
+                    "type": "object",
+                    "properties": {"location": {"type": "string", "description": "City name"}},
+                    "required": ["location"],
                 },
-                "required": ["location"]
-            }
+            },
         }
-    }]
-    
+    ]
+
     # Chat Completions with function calling
     response = traced_client.chat.completions.create(
         model="gpt-4o-mini",
@@ -161,12 +164,12 @@ def function_calling_example():
         tools=tools,
         tool_choice="auto",
     )
-    
+
     print(f"Function call response: {response.choices[0].message}")
     print("✓ Function calling with Chat Completions traced successfully")
-    
+
     # Responses API with function calling (if available)
-    if hasattr(traced_client, 'responses'):
+    if hasattr(traced_client, "responses"):
         try:
             response = traced_client.responses.create(
                 model="gpt-4o-mini",
@@ -179,14 +182,15 @@ def function_calling_example():
         except Exception as e:
             print(f"⚠️ Responses API function calling not yet supported: {e}")
 
+
 async def async_examples():
     """Examples using async clients."""
     print("\n=== Async Examples ===")
-    
+
     # Create and trace async OpenAI client
     client = openai.AsyncOpenAI()
     traced_client = trace_async_openai(client)
-    
+
     # Async chat completion
     response = await traced_client.chat.completions.create(
         model="gpt-4o-mini",
@@ -195,12 +199,12 @@ async def async_examples():
         ],
         temperature=0.1,
     )
-    
+
     print(f"Async chat response: {response.choices[0].message.content}")
     print("✓ Async Chat Completions traced successfully")
-    
+
     # Async responses (if available)
-    if hasattr(traced_client, 'responses'):
+    if hasattr(traced_client, "responses"):
         try:
             response = await traced_client.responses.create(
                 model="gpt-4o-mini",
@@ -212,26 +216,28 @@ async def async_examples():
         except Exception as e:
             print(f"⚠️ Async Responses API error: {e}")
 
+
 def main():
     """Run all examples."""
     print("OpenAI Chat Completions + Responses API Tracing Examples")
     print("=" * 60)
-    
+
     # Setup (in real usage, set these in your environment)
     setup_environment()
-    
+
     try:
         # Sync examples
         chat_completions_example()
         responses_api_example()
         streaming_chat_completions_example()
         streaming_responses_api_example()
         function_calling_example()
-        
+
         # Async examples
         import asyncio
+
         asyncio.run(async_examples())
-        
+
         print("\n🎉 All examples completed successfully!")
         print("\nKey Benefits of the New Implementation:")
         print("✓ Backward compatibility - existing Chat Completions code works unchanged")
@@ -240,10 +246,11 @@ def main():
         print("✓ Function calling - tool/function calls are properly captured in traces")
         print("✓ Enhanced metadata - Responses API provides richer traceability information")
         print("✓ Async support - both sync and async clients work seamlessly")
-        
+
     except Exception as e:
         print(f"❌ Example failed: {e}")
         print("Note: This example requires valid OpenAI API keys and Openlayer configuration")
 
+
 if __name__ == "__main__":
-    main()
+    main()