知你客服
This commit is contained in:
186
backend/scripts/generate_android_log_agent.py
Normal file
186
backend/scripts/generate_android_log_agent.py
Normal file
@@ -0,0 +1,186 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
生成Android日志获取Agent
|
||||
"""
|
||||
import sys
|
||||
import os
|
||||
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||||
|
||||
from app.core.database import SessionLocal
|
||||
from app.models.agent import Agent
|
||||
from app.models.workflow import Workflow
|
||||
import uuid
|
||||
from datetime import datetime
|
||||
|
||||
def generate_android_log_agent():
|
||||
"""生成Android日志获取Agent"""
|
||||
db = SessionLocal()
|
||||
|
||||
try:
|
||||
# 检查是否已存在
|
||||
existing = db.query(Agent).filter(Agent.name == "Android日志获取助手").first()
|
||||
if existing:
|
||||
print(f"Agent 'Android日志获取助手' 已存在，ID: {existing.id}")
|
||||
return existing.id
|
||||
|
||||
# 创建工作流
|
||||
workflow_id = str(uuid.uuid4())
|
||||
nodes = [
|
||||
{
|
||||
"id": "start",
|
||||
"type": "start",
|
||||
"position": {"x": 100, "y": 200},
|
||||
"data": {"label": "开始"}
|
||||
},
|
||||
{
|
||||
"id": "intent-recognize",
|
||||
"type": "llm",
|
||||
"position": {"x": 400, "y": 200},
|
||||
"data": {
|
||||
"label": "意图识别",
|
||||
"provider": "deepseek",
|
||||
"model": "deepseek-chat",
|
||||
"temperature": 0.3,
|
||||
"max_tokens": 200,
|
||||
"prompt": """分析用户请求，提取以下信息：
|
||||
1. 命令类型：logcat（获取日志）、devices（列出设备）、shell（执行shell命令）
|
||||
2. 日志标签过滤（如果有）
|
||||
3. 日志级别过滤（V/D/I/W/E/F/S，如果有）
|
||||
4. 最大行数（默认100）
|
||||
|
||||
用户请求：{{input.query}}
|
||||
|
||||
请以JSON格式返回：
|
||||
{
|
||||
"command": "logcat|devices|shell",
|
||||
"filter_tag": "标签或shell命令（可选）",
|
||||
"level": "V|D|I|W|E|F|S（可选）",
|
||||
"max_lines": 100
|
||||
}"""
|
||||
}
|
||||
},
|
||||
{
|
||||
"id": "json-parse",
|
||||
"type": "json",
|
||||
"position": {"x": 700, "y": 200},
|
||||
"data": {
|
||||
"label": "解析参数",
|
||||
"operation": "parse",
|
||||
"json_path": "$"
|
||||
}
|
||||
},
|
||||
{
|
||||
"id": "llm-with-tools",
|
||||
"type": "llm",
|
||||
"position": {"x": 1000, "y": 200},
|
||||
"data": {
|
||||
"label": "执行ADB命令",
|
||||
"provider": "deepseek",
|
||||
"model": "deepseek-chat",
|
||||
"temperature": 0.7,
|
||||
"max_tokens": 2000,
|
||||
"enable_tools": True,
|
||||
"selected_tools": ["adb_log"],
|
||||
"prompt": """根据解析的参数，使用adb_log工具获取Android设备日志。
|
||||
|
||||
参数：
|
||||
- command: {{json-parse.output.command}}
|
||||
- filter_tag: {{json-parse.output.filter_tag}}
|
||||
- level: {{json-parse.output.level}}
|
||||
- max_lines: {{json-parse.output.max_lines}}
|
||||
|
||||
请调用adb_log工具获取日志，然后分析日志内容，提取关键信息并总结。"""
|
||||
}
|
||||
},
|
||||
{
|
||||
"id": "end",
|
||||
"type": "end",
|
||||
"position": {"x": 1300, "y": 200},
|
||||
"data": {"label": "结束"}
|
||||
}
|
||||
]
|
||||
|
||||
edges = [
|
||||
{
|
||||
"id": "e1",
|
||||
"source": "start",
|
||||
"target": "intent-recognize",
|
||||
"sourceHandle": "right",
|
||||
"targetHandle": "left"
|
||||
},
|
||||
{
|
||||
"id": "e2",
|
||||
"source": "intent-recognize",
|
||||
"target": "json-parse",
|
||||
"sourceHandle": "right",
|
||||
"targetHandle": "left"
|
||||
},
|
||||
{
|
||||
"id": "e3",
|
||||
"source": "json-parse",
|
||||
"target": "llm-with-tools",
|
||||
"sourceHandle": "right",
|
||||
"targetHandle": "left"
|
||||
},
|
||||
{
|
||||
"id": "e4",
|
||||
"source": "llm-with-tools",
|
||||
"target": "end",
|
||||
"sourceHandle": "right",
|
||||
"targetHandle": "left"
|
||||
}
|
||||
]
|
||||
|
||||
workflow = Workflow(
|
||||
id=workflow_id,
|
||||
name="Android日志获取工作流",
|
||||
description="通过ADB命令获取Android设备日志的工作流",
|
||||
nodes=nodes,
|
||||
edges=edges,
|
||||
created_at=datetime.now(),
|
||||
updated_at=datetime.now()
|
||||
)
|
||||
|
||||
db.add(workflow)
|
||||
db.commit()
|
||||
db.refresh(workflow)
|
||||
|
||||
# 创建Agent
|
||||
agent_id = str(uuid.uuid4())
|
||||
agent = Agent(
|
||||
id=agent_id,
|
||||
name="Android日志获取助手",
|
||||
description="通过ADB命令获取和分析Android设备日志的智能助手。支持获取logcat日志、列出设备、执行shell命令等功能。",
|
||||
workflow_config={
|
||||
"workflow_id": workflow_id,
|
||||
"nodes": nodes,
|
||||
"edges": edges
|
||||
},
|
||||
created_at=datetime.now(),
|
||||
updated_at=datetime.now()
|
||||
)
|
||||
|
||||
db.add(agent)
|
||||
db.commit()
|
||||
db.refresh(agent)
|
||||
|
||||
print(f"✅ Android日志获取Agent创建成功！")
|
||||
print(f" Agent ID: {agent_id}")
|
||||
print(f" Agent名称: {agent.name}")
|
||||
print(f" 工作流ID: {workflow_id}")
|
||||
print(f"\n使用示例：")
|
||||
print(f" python3 test_workflow_tool.py -a \"Android日志获取助手\" -i '{{\"query\": \"获取最近的错误日志\"}}'")
|
||||
|
||||
return agent_id
|
||||
|
||||
except Exception as e:
|
||||
db.rollback()
|
||||
print(f"❌ 创建Agent失败: {e}")
|
||||
import traceback
|
||||
traceback.print_exc()
|
||||
return None
|
||||
finally:
|
||||
db.close()
|
||||
|
||||
if __name__ == "__main__":
|
||||
generate_android_log_agent()
|
||||
Reference in New Issue
Block a user