messages=[TextMessage(source='user', models_usage=None, metadata={}, content='What is the capital of France?', type='TextMessage'), TextMessage(source='assistant', models_usage=RequestUsage(prompt_tokens=13, completion_tokens=8), metadata={}, content='The capital of France is Paris.\n', type='TextMessage')] stop_reason=None

from autogen_agentchat.agents import AssistantAgent
from autogen_core.models import ChatCompletionClient, UserMessage
from autogen_agentchat.messages import TextMessage
# Print loaded versions
import importlib.metadata

def print_versions():
    packages = ["autogen-agentchat", "autogen-core"]
    for pkg in packages:
        try:
            version = importlib.metadata.version(pkg)
            print(f"{pkg} version: {version}")
        except importlib.metadata.PackageNotFoundError:
            print(f"{pkg} is not installed or version not found")

print_versions()

model_config = {
    'provider': 'autogen_ext.models.openai.AzureOpenAIChatCompletionClient', 
    'config': 
        {
            'model': 'gpt-4o-mini', 
            'azure_endpoint': os.getenv('AZURE_ENDPOINT_URL'),
            'api_version': '2025-02-01-preview', 
            'max_tokens': 500,
            'temperature': 0.1
        }
}
model_client = ChatCompletionClient.load_component(model_config)

async def main():
    agent = AssistantAgent(
            name="assistant",
            model_client=model_client,
            model_client_stream=True,
            system_message = "You are a helpful assistant",
            reflect_on_tool_use=True
        )
    request = TextMessage(content="What is the capital of France?", source="user")
    stream = agent.run_stream(task=request)
    async for message in stream:
        if isinstance(message, TextMessage):
            print(message.model_dump())
            
await main()

model_client = OpenAIChatCompletionClient(
    ...,
    stream_options={"include_usage": True},
)