Python: add agent telemetry (#283)

* add agent telemetry

* updated comments

* update AgentRunResponseUpdate

* updated create_agent var

---------

Co-authored-by: Chris <66376200+crickman@users.noreply.github.com>
This commit is contained in:
Eduard van Valkenburg
2025-07-31 12:29:38 +02:00
committed by GitHub
Unverified
parent e877ffbca1
commit caee8bfa90
6 changed files with 907 additions and 37 deletions
@@ -3,7 +3,7 @@
import sys
from collections.abc import AsyncIterable, Callable, MutableMapping, Sequence
from enum import Enum
from typing import Any, Literal, Protocol, TypeVar, runtime_checkable
from typing import Any, ClassVar, Literal, Protocol, TypeVar, runtime_checkable
from uuid import uuid4
from pydantic import BaseModel, Field
@@ -22,6 +22,7 @@ from ._types import (
ChatToolMode,
)
from .exceptions import AgentExecutionException
from .telemetry import use_agent_telemetry
if sys.version_info >= (3, 11):
from typing import Self # pragma: no cover
@@ -305,9 +306,11 @@ class ChatClientAgentThread(AgentThread):
# region ChatClientAgent
@use_agent_telemetry
class ChatClientAgent(AgentBase):
"""A Chat Client Agent."""
AGENT_SYSTEM_NAME: ClassVar[str] = "microsoft.agent_framework"
chat_client: ChatClient
instructions: str | None = None
chat_options: ChatOptions
@@ -525,7 +528,7 @@ class ChatClientAgent(AgentBase):
response_id=response.response_id,
created_at=response.created_at,
usage_details=response.usage_details,
raw_representation=response.raw_representation,
raw_representation=response,
additional_properties=response.additional_properties,
)
@@ -626,7 +629,7 @@ class ChatClientAgent(AgentBase):
message_id=update.message_id,
created_at=update.created_at,
additional_properties=update.additional_properties,
raw_representation=update.raw_representation,
raw_representation=update,
)
response = ChatResponse.from_chat_response_updates(response_updates)