from autogen_core.tools import BaseTool, FunctionTool
from autogen_agentchat.agents import AssistantAgent
from autogen_agentchat.messages import TextMessage
from typing import List
from autogen_ext.models.openai import OpenAIChatCompletionClient
from autogen_core import CancellationToken
import asyncio


async def get_current_time() -> str:
    return "The current time is 12:00 PM."

async def get_stock_price(stock: str ="MFST") -> str:
    return "The stock price is 500 dollors."


async def main():
    tools: List[BaseTool] = [
        FunctionTool(
            get_current_time,
            name="get_current_time",
            description="Get current time in Paris.",
        ),
        FunctionTool(
            get_stock_price,
            name="get_stock_price",
            description="Fetch the stock price for a given company."
        ),
    ]
    model_client = OpenAIChatCompletionClient(
            model="gemini-2.5-flash",
            api_key="",
    )

    response = await AssistantAgent(
            name="Agent",
            model_client=model_client,
            model_client_stream=True,
            reflect_on_tool_use=True,
            tools=tools,
            max_tool_iterations=10,
            system_message= "You are an assistant with some tools that can be used to answer some questions",
        ).on_messages([TextMessage(content="What is current time of Paris now? And what is stock price of MSFT", source="user")], CancellationToken())

    print(response)



if __name__ == "__main__":
    asyncio.run(main())