nicer

Signed-off-by: Adrian Cole <[email protected]>
open-telemetry · Nov 15, 2024 · f0e9100 · f0e9100
1 parent 42a8390
commit f0e9100
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/instrumentation-genai/opentelemetry-instrumentation-openai-v2/example/main.py b/instrumentation-genai/opentelemetry-instrumentation-openai-v2/example/main.py
@@ -11,15 +11,14 @@
 
 def main():
     client = OpenAI()
-    messages = [
-        {
-            "role": "user",
-            "content": "Write a short poem on OpenTelemetry.",
-        },
-    ]
-    model = os.getenv("CHAT_MODEL", "gpt-4o-mini")
     chat_completion = client.chat.completions.create(
-        model=model, messages=messages
+        model=os.getenv("CHAT_MODEL", "gpt-4o-mini"),
+        messages=[
+            {
+                "role": "user",
+                "content": "Write a short poem on OpenTelemetry.",
+            },
+        ],
     )
     print(chat_completion.choices[0].message.content)