from openai import OpenAI
from openai.agents import Agent, Runner

client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))

agent = Agent(
    name="TaskAssistant",
    instructions="""You are a helpful task management assistant.
Use the provided tools to help users manage their todo list.
Always confirm actions with friendly responses.
If a user's request is ambiguous, ask for clarification.""",
    model="gpt-4o",
    tools=[]  # MCP tools registered here
)

tools = [
    {
        "type": "function",
        "function": {
            "name": "add_task",
            "description": "Create a new task",
            "parameters": {
                "type": "object",
                "properties": {
                    "user_id": {"type": "string"},
                    "title": {"type": "string"},
                    "description": {"type": "string"}
                },
                "required": ["user_id", "title"]
            }
        }
    }
]
agent.tools = tools

async def run_agent(messages: list[dict]) -> tuple[str, list[str]]:
    """Run agent with conversation history."""
    runner = Runner(agent=agent)
    result = await runner.run(messages=messages)
    
    # Extract response
    response = result.messages[-1].content[0].text.value
    
    # Extract tool calls
    tool_calls = [
        call.function.name 
        for msg in result.messages 
        if hasattr(msg, 'tool_calls') 
        for call in msg.tool_calls
    ]
    
    return response, tool_calls

messages = [
    {"role": "user", "content": "Add a task"},
    {"role": "assistant", "content": "I've added the task"},
    {"role": "user", "content": "Show my tasks"}
]