Calendar Assistant

from datetime import datetime, timedelta
import os
from agno.agent import Agent
from agno.models.nebius import Nebius
from agno.tools.calcom import CalComTools
from dotenv import load_dotenv

load_dotenv()

INSTRUCTIONS = f"""You're a scheduling assistant. Today is {datetime.now()}.
You can help users by:
    - Finding available time slots using get_available_slots(start_date, end_date)
    - Creating new bookings using create_booking(start_time, name, email)
    - Managing existing bookings using get_upcoming_bookings(email)
    - Rescheduling bookings using reschedule_booking(booking_uid, new_start_time, reason)
    - Cancelling bookings using cancel_booking(booking_uid, reason)

IMPORTANT STEPS for booking:
1. First check available slots using get_available_slots
2. Then create booking using create_booking with the exact start_time, name, and email
3. Finally verify the booking was created using get_upcoming_bookings

Remember:
- Dates should be in YYYY-MM-DD format
- Times should be in YYYY-MM-DDTHH:MM:SS+TZ format
- Always confirm details before making changes
"""

# Create the CalCom tools instance
calcom_tools = CalComTools(
    user_timezone="Asia/Kolkata",
    api_key=os.environ['CALCOM_API_KEY'],
    event_type_id=os.environ.get('CALCOM_EVENT_TYPE_ID')
)

# Create the agent
agent = Agent(
    name="Calendar Assistant",
    instructions=[INSTRUCTIONS],
    model=Nebius(
        id="Qwen/Qwen3-30B-A3B",
        api_key=os.getenv("NEBIUS_API_KEY")
    ),
    tools=[calcom_tools],
    show_tool_calls=True,
    tool_choice="auto",
    markdown=True,
)

def book_example_call():
    # Get today's date and tomorrow's date
    today = datetime.now().strftime("%Y-%m-%d")
    tomorrow = (datetime.now() + timedelta(days=1)).strftime("%Y-%m-%d")
    
    # First, check available slots
    print("Checking available slots...")
    agent.print_response(f"""
    Please check available slots between {today} and {tomorrow}
    """)
    
    # Then book a specific slot
    print("\nAttempting to book a call...")
    agent.print_response("""
    Please book a call with these details:
    - Start Time: 2025-03-22T21:30:00+05:30
    - Name: John Doe
    - Email: [email protected]
    
    After booking, please verify the booking exists.
    """)

if __name__ == "__main__":
    book_example_call()

agent.print_response("""
Please check available slots between 2024-03-20 and 2024-03-21
""")

agent.print_response("""
Please book a call with these details:
- Start Time: 2024-03-22T21:30:00+05:30
- Name: John Doe
- Email: [email protected]
""")

calcom_tools = CalComTools(
    user_timezone="America/New_York",  # Change timezone here
    api_key=os.environ['CALCOM_API_KEY'],
    event_type_id=os.environ.get('CALCOM_EVENT_TYPE_ID')
)