Python: Improved telemetry setup (#421)

* test with stack and simplified names * quick demo of agent decorator * moved builder to protocol to enhance functionality * undid chatclientAgent -> agent rename * one more * reverted AIAgent rename * final reverts * fixed foundry import * revert changes * streamlined otel and fcc decorators * cleanup of telemetry * further refinement * lots of updates * fixed typing * fix for mypy * added input and output atttributes * fix import * initial work on baking in otel * major update to telemetry * final fixes after rename * fix * fix test * updated tests * fix for tests * fixes for tests * updated based on comments * removed agent decorator * fix for Python: ServiceResponseException when using multiple tools Fixes #649 * addressed comments * fix tests * fix tests * fix tools tests * fix for conversation_id in assistants client * fix responses test * fix tests and mypy * updated test * foundry fix --------- Co-authored-by: Chris <66376200+crickman@users.noreply.github.com>
2026-06-16 21:04:09 +08:00 · 2025-09-10 16:52:42 +02:00
parent 6aa746d891
commit 82ca4065cb
45 changed files with 3246 additions and 2984 deletions
@@ -0,0 +1,59 @@
+# Copyright (c) Microsoft. All rights reserved.
+# type: ignore
+import asyncio
+from random import randint
+from typing import Annotated
+
+from agent_framework import ChatAgent
+from agent_framework.openai import OpenAIChatClient
+from agent_framework.telemetry import setup_telemetry
+from opentelemetry import trace
+from opentelemetry.trace import SpanKind
+from pydantic import Field
+
+"""
+This sample shows you can can setup telemetry with a agent.
+The agent invoke is a additional Semantic Convention that now
+will wrap the calls made by the underlying chat client and tools.
+"""
+
+
+async def get_weather(
+    location: Annotated[str, Field(description="The location to get the weather for.")],
+) -> str:
+    """Get the weather for a given location."""
+    await asyncio.sleep(randint(0, 10) / 10.0)  # Simulate a network call
+    conditions = ["sunny", "cloudy", "rainy", "stormy"]
+    return f"The weather in {location} is {conditions[randint(0, 3)]} with a high of {randint(10, 30)}°C."
+
+
+async def main():
+    # Set up the telemetry
+    setup_telemetry()
+
+    questions = ["What's the weather in Amsterdam?", "and in Paris, and which is better?", "Why is the sky blue?"]
+
+    tracer = trace.get_tracer("agent_framework")
+    with tracer.start_as_current_span("Scenario: Agent Chat", kind=SpanKind.CLIENT):
+        print("Running scenario: Agent Chat")
+        print("Welcome to the chat, type 'exit' to quit.")
+        agent = ChatAgent(
+            chat_client=OpenAIChatClient(),
+            tools=get_weather,
+            name="WeatherAgent",
+            instructions="You are a weather assistant.",
+        )
+        thread = agent.get_new_thread()
+        for question in questions:
+            print(f"User: {question}")
+            print(f"{agent.display_name}: ", end="")
+            async for update in agent.run_stream(
+                question,
+                thread=thread,
+            ):
+                if update.text:
+                    print(update.text, end="")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())