38 lines
1015 B
Python
38 lines
1015 B
Python
|
|
import asyncio
|
|
import sys
|
|
import os
|
|
|
|
# Add project path
|
|
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
|
|
|
from module_admin.service.search_service import SearchService
|
|
|
|
async def test_intent_router():
|
|
print("=" * 60)
|
|
print("Intent Router Test")
|
|
print("=" * 60)
|
|
|
|
# Test Case 1: Internal Question
|
|
q1 = "康达什么时候成立的？"
|
|
print(f"\nQuery 1: {q1}")
|
|
intent1 = await SearchService.classify_intent(q1)
|
|
print(f"Result: {intent1} (Expected: RAG)")
|
|
|
|
# Test Case 2: External Question
|
|
q2 = "今天上海的天气怎么样？"
|
|
print(f"\nQuery 2: {q2}")
|
|
intent2 = await SearchService.classify_intent(q2)
|
|
print(f"Result: {intent2} (Expected: SEARCH)")
|
|
|
|
# Test Case 3: Fuzzy Internal
|
|
q3 = "公司有什么新产品？"
|
|
print(f"\nQuery 3: {q3}")
|
|
intent3 = await SearchService.classify_intent(q3)
|
|
print(f"Result: {intent3} (Expected: RAG)")
|
|
|
|
print("-" * 60)
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(test_intent_router())
|