import asyncio
from tantra.agent import Agent
from tantra.tools import Tool
from tantra.session import Session

# Define a simple tool
@Tool
async def set_reminder(task: str, time: str):
    """Set a reminder for a task at a specific time"""
    print(f"Reminder set: {task} at {time}")
    return "Reminder set successfully!"

# Create and run agent
async def main():
    # Create a session with tools
    session = Session(tools=[set_reminder])
    
    # Create an agent with the session
    agent = Agent(session)
    
    # Start the conversation
    await agent.on_event({"type": "SESSION_START"})
    
    # Main conversation loop
    try:
        while True:
            user_input = input("\nYou: ").strip()
            if not user_input:
                break
                
            # Process user input
            await agent.think(user_input)
    except KeyboardInterrupt:
        pass
    
    # End the session gracefully
    await agent.on_event({"type": "SESSION_END"})

# Run the agent
if __name__ == "__main__":
    asyncio.run(main())

# User: "set a reminder"
# Agent: "What task would you like to be reminded about?"
# User: "team meeting"
# Agent: "What time should I set the reminder for?"
# User: "tomorrow at 2pm"
# Agent: "I'll set a reminder for your team meeting tomorrow at 2pm"

# Agent remembers facts and previous interactions
await agent.memory.add_fact("user_timezone", "EST")
await agent.memory.add_fact("meeting_preference", "mornings")

# Facts influence the agent's decisions
# User: "set a meeting"
# Agent: "Since you prefer mornings, would you like me to schedule it for tomorrow morning?"

# A basic AI assistant that can:
# - Set reminders
# - Search the web
# - Check weather
# - Search vector databases

LOG_LEVEL=DEBUG poetry run python examples/assistant_agent.py

# Shows how to create custom tools and use them with the agent
# Includes examples of:
# - Parameter validation
# - Async tool execution
# - Error handling

poetry run python examples/custom_tool_agent.py