test_personalized_planner.py

import asyncio
import os
from dotenv import load_dotenv
from tools.personalized_planner import create_personalized_plan

load_dotenv()

TEST_USER_NAME = "Tracy"
TEST_NEW_DESTINATION = "Kyoto"
TEST_TRIP_DURATION_DAYS = 5
TEST_USER_REQUEST = "我会在下个月八月搭乘芬兰航空去京都，想要一个文化沉浸和美食体验的行程，最好能有一些放松的时刻。"

async def run_test():
    print("🚀 Running test for personalized travel plan...\n")
    result = await create_personalized_plan(
        user_name=TEST_USER_NAME,
        new_destination=TEST_NEW_DESTINATION,
        trip_duration_days=TEST_TRIP_DURATION_DAYS,
        user_request=TEST_USER_REQUEST
    )
    print("\n📒 Generated Plan:\n")
    print(result)

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