first commit
This commit is contained in:
闫旭隆
194
tests/debug_research_v2.py
Normal file
194
tests/debug_research_v2.py
Normal file
@ -0,0 +1,194 @@
|
||||
"""
|
||||
调试研究流程 V2 - 检查虚拟文件系统
|
||||
|
||||
使用方法：
|
||||
export PYTHONIOENCODING=utf-8 && python tests/debug_research_v2.py
|
||||
"""
|
||||
|
||||
import sys
|
||||
import os
|
||||
import json
|
||||
from datetime import datetime
|
||||
|
||||
# 添加项目根目录到Python路径
|
||||
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||||
|
||||
from src.agents.coordinator import create_research_coordinator
|
||||
from src.config import Config
|
||||
|
||||
|
||||
def debug_research_with_files(question: str, depth: str = "quick"):
|
||||
"""
|
||||
调试研究流程，重点检查虚拟文件系统
|
||||
|
||||
Args:
|
||||
question: 研究问题
|
||||
depth: 深度模式
|
||||
"""
|
||||
print("\n" + "🔬 " * 40)
|
||||
print("智能深度研究系统 - 调试模式 V2")
|
||||
print("🔬 " * 40)
|
||||
|
||||
print(f"\n研究问题: {question}")
|
||||
print(f"深度模式: {depth}")
|
||||
print(f"开始时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
|
||||
|
||||
# 创建Agent
|
||||
print("\n" + "="*80)
|
||||
print("创建ResearchCoordinator Agent")
|
||||
print("="*80)
|
||||
|
||||
try:
|
||||
agent = create_research_coordinator(
|
||||
question=question,
|
||||
depth=depth,
|
||||
format="technical",
|
||||
min_tier=3
|
||||
)
|
||||
print("✅ Agent创建成功")
|
||||
except Exception as e:
|
||||
print(f"❌ Agent创建失败: {e}")
|
||||
import traceback
|
||||
traceback.print_exc()
|
||||
return
|
||||
|
||||
# 执行研究
|
||||
print("\n" + "="*80)
|
||||
print("执行研究流程")
|
||||
print("="*80)
|
||||
|
||||
try:
|
||||
start_time = datetime.now()
|
||||
|
||||
result = agent.invoke({
|
||||
"messages": [
|
||||
{
|
||||
"role": "user",
|
||||
"content": f"请开始研究这个问题：{question}"
|
||||
}
|
||||
]
|
||||
})
|
||||
|
||||
end_time = datetime.now()
|
||||
duration = (end_time - start_time).total_seconds()
|
||||
|
||||
print(f"\n✅ 执行完成！耗时: {duration:.2f}秒")
|
||||
|
||||
# 分析结果
|
||||
print("\n" + "="*80)
|
||||
print("结果分析")
|
||||
print("="*80)
|
||||
|
||||
print(f"\n结果类型: {type(result)}")
|
||||
print(f"结果键: {list(result.keys())}")
|
||||
|
||||
# 检查消息
|
||||
if 'messages' in result:
|
||||
messages = result['messages']
|
||||
print(f"\n📨 消息数量: {len(messages)}")
|
||||
|
||||
print("\n所有消息内容:")
|
||||
for i, msg in enumerate(messages, 1):
|
||||
print(f"\n{'='*60}")
|
||||
print(f"消息 #{i}")
|
||||
print('='*60)
|
||||
|
||||
# 检查消息类型
|
||||
msg_type = type(msg).__name__
|
||||
print(f"类型: {msg_type}")
|
||||
|
||||
# 提取内容
|
||||
if hasattr(msg, 'content'):
|
||||
content = msg.content
|
||||
print(f"内容长度: {len(content)} 字符")
|
||||
|
||||
# 显示内容
|
||||
if len(content) > 500:
|
||||
print(f"\n内容预览:\n{content[:500]}...")
|
||||
else:
|
||||
print(f"\n完整内容:\n{content}")
|
||||
|
||||
# 检查其他属性
|
||||
if hasattr(msg, 'additional_kwargs'):
|
||||
kwargs = msg.additional_kwargs
|
||||
if kwargs:
|
||||
print(f"\n额外参数: {kwargs}")
|
||||
|
||||
if hasattr(msg, 'tool_calls'):
|
||||
tool_calls = msg.tool_calls
|
||||
if tool_calls:
|
||||
print(f"\n工具调用: {tool_calls}")
|
||||
|
||||
# 检查文件系统
|
||||
if 'files' in result:
|
||||
files = result['files']
|
||||
print("\n" + "="*80)
|
||||
print("虚拟文件系统")
|
||||
print("="*80)
|
||||
print(f"\n📁 文件数量: {len(files)}")
|
||||
|
||||
for file_path, file_info in files.items():
|
||||
print(f"\n{'='*60}")
|
||||
print(f"文件: {file_path}")
|
||||
print('='*60)
|
||||
|
||||
# 显示文件信息
|
||||
if isinstance(file_info, dict):
|
||||
for key, value in file_info.items():
|
||||
if key == 'content':
|
||||
if len(str(value)) > 300:
|
||||
print(f"{key}: {str(value)[:300]}...")
|
||||
else:
|
||||
print(f"{key}: {value}")
|
||||
else:
|
||||
print(f"{key}: {value}")
|
||||
else:
|
||||
if len(str(file_info)) > 300:
|
||||
print(f"内容: {str(file_info)[:300]}...")
|
||||
else:
|
||||
print(f"内容: {file_info}")
|
||||
|
||||
# 保存完整结果
|
||||
print("\n" + "="*80)
|
||||
print("保存调试结果")
|
||||
print("="*80)
|
||||
|
||||
output_dir = "outputs/debug"
|
||||
os.makedirs(output_dir, exist_ok=True)
|
||||
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
|
||||
|
||||
# 保存JSON结果
|
||||
output_file = os.path.join(output_dir, f"debug_v2_{timestamp}.json")
|
||||
with open(output_file, 'w', encoding='utf-8') as f:
|
||||
# 序列化结果
|
||||
serialized_result = {
|
||||
"question": question,
|
||||
"depth": depth,
|
||||
"duration_seconds": duration,
|
||||
"messages": [
|
||||
{
|
||||
"type": type(msg).__name__,
|
||||
"content": msg.content if hasattr(msg, 'content') else str(msg)
|
||||
}
|
||||
for msg in result.get('messages', [])
|
||||
],
|
||||
"files": {
|
||||
path: str(content)
|
||||
for path, content in result.get('files', {}).items()
|
||||
}
|
||||
}
|
||||
json.dump(serialized_result, f, ensure_ascii=False, indent=2)
|
||||
|
||||
print(f"✅ 调试结果已保存到: {output_file}")
|
||||
|
||||
except KeyboardInterrupt:
|
||||
print("\n\n⚠️ 用户中断执行")
|
||||
except Exception as e:
|
||||
print(f"\n\n❌ 执行失败: {e}")
|
||||
import traceback
|
||||
traceback.print_exc()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
question = "Python asyncio最佳实践"
|
||||
debug_research_with_files(question, depth="quick")
|
||||
Reference in New Issue
Block a user