You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
48 lines
1.3 KiB
48 lines
1.3 KiB
"""
|
|
测试AI智能分类功能
|
|
"""
|
|
|
|
import asyncio
|
|
import os
|
|
from dotenv import load_dotenv
|
|
from ai_agent import extract_search_keywords
|
|
|
|
# 加载环境变量
|
|
load_dotenv()
|
|
|
|
|
|
async def test_classification():
|
|
"""测试AI分类功能"""
|
|
|
|
test_cases = [
|
|
"我想做一些美食相关的短视频，主要是家常菜的制作教程",
|
|
"我想拍一些关于大学生活的有趣视频，记录校园日常",
|
|
"我想做一些关于健身的短视频，分享简单的居家锻炼方法",
|
|
"想拍一些旅游vlog，记录旅行中的美好瞬间",
|
|
"我想做游戏解说视频",
|
|
"想做一些汽车评测的内容",
|
|
"我想做短视频，但不知道拍什么"
|
|
]
|
|
|
|
print("=" * 80)
|
|
print("AI智能分类测试")
|
|
print("=" * 80)
|
|
|
|
for i, query in enumerate(test_cases, 1):
|
|
print(f"\n测试 {i}:")
|
|
print(f"用户输入: {query}")
|
|
|
|
result = extract_search_keywords(query)
|
|
|
|
if result["success"]:
|
|
print(f"✓ 分类成功")
|
|
print(f" - 主分类: {result['primary_category']}")
|
|
print(f" - 分类方法: {result['method']}")
|
|
else:
|
|
print(f"✗ 分类失败")
|
|
|
|
print("-" * 80)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(test_classification())
|
|
|