
import asyncio
import os
import sys

# Add backend to sys.path
sys.path.insert(0, os.path.abspath("backend"))

from agent.lead_stage_agent.lead_search_tool import lead_search_tool
from common.sqlite_db import sqlite_db

async def test():
    # Test "áo mặc trời mưa"
    print("Testing 'áo mặc trời mưa'...")
    res = await lead_search_tool.ainvoke({
        "product_line_vn": ["Áo khoác"],
        "keywords": ["chống thấm", "trượt nước", "cản gió"],
        "reasoning": "Khách cần đồ đi mưa"
    })
    print(res)

    # Test "đồ đi làm văn phòng nữ thanh lịch"
    print("\nTesting 'đồ đi làm văn phòng nữ thanh lịch'...")
    res = await lead_search_tool.ainvoke({
        "product_line_vn": ["Áo sơ mi", "Áo Polo", "Quần Khaki", "Quần jean", "Váy liền", "Blazer"],
        "keywords": ["thanh lịch", "công sở"],
        "gender_by_product": "women",
        "reasoning": "Đồ công sở nữ"
    })
    print(res)

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