import asyncio
from autogen_agentchat.agents import AssistantAgent
from autogen_ext.models.openai import OpenAIChatCompletionClient

async def main() -> None:
    # Create a model client
    model_client = OpenAIChatCompletionClient(model="gpt-4o")
    
    # Create an assistant agent
    agent = AssistantAgent(
        name="assistant",
        model_client=model_client
    )
    
    # Run the agent with a task
    result = await agent.run(task="Say 'Hello World!'")
    print(result)
    
    # Clean up
    await model_client.close()

# Run the async function
asyncio.run(main())

import asyncio
from autogen_agentchat.agents import AssistantAgent
from autogen_agentchat.ui import Console
from autogen_ext.models.openai import OpenAIChatCompletionClient

# Define a tool as a simple Python function
async def get_weather(city: str) -> str:
    """Get the current weather for a given city.
    
    Args:
        city: The name of the city to get weather for
        
    Returns:
        A string describing the weather
    """
    # In a real application, you'd call a weather API here
    return f"The weather in {city} is 73 degrees and sunny."

async def main() -> None:
    # Create a model client
    model_client = OpenAIChatCompletionClient(
        model="gpt-4o",
        # api_key="sk-..."  # Optional if OPENAI_API_KEY is set
    )
    
    # Create an agent with the weather tool
    agent = AssistantAgent(
        name="weather_agent",
        model_client=model_client,
        tools=[get_weather],  # Pass the function directly
        system_message="You are a helpful weather assistant.",
        reflect_on_tool_use=True,  # Agent reflects on tool results
    )
    
    # Run with streaming output to the console
    await Console(
        agent.run_stream(task="What's the weather in New York?")
    )
    
    # Clean up
    await model_client.close()

asyncio.run(main())