100 lines
3.7 KiB
Python
100 lines
3.7 KiB
Python
import asyncio
|
|
from openai.types.responses import ResponseTextDeltaEvent
|
|
from agents.run import Runner
|
|
from m12_agents_sdk_swarm.agent import triage_agent,amap_server
|
|
from m12_agents_sdk_swarm.tools import context_variables
|
|
|
|
# 不启用Chatgpt官方提供的Tracing(无需配置api_key)
|
|
from agents.tracing import set_tracing_disabled
|
|
set_tracing_disabled(True)
|
|
|
|
# 启用Chatgpt官方提供的Tracing(需要配置api_key)
|
|
# from config import CHATGPT_API_KEY
|
|
# from agents.tracing import set_tracing_export_api_key
|
|
# set_tracing_export_api_key(CHATGPT_API_KEY)
|
|
|
|
|
|
async def main():
|
|
print('✈️ 客服系统启动...\n')
|
|
messages = [] # 对话历史
|
|
cur_agent = triage_agent # 当前Agent
|
|
|
|
try:
|
|
await amap_server[0].connect()
|
|
print("✅ MCP Server (amap) connected.")
|
|
except Exception as e:
|
|
print(f"⚠️ Failed to connect MCP Server: {e}")
|
|
|
|
while True:
|
|
# 获取用户输入
|
|
user_input = input("\nUser: ")
|
|
if user_input == "quit":
|
|
break
|
|
messages.append({"role": "user", "content": user_input})
|
|
|
|
# 启动流式响应
|
|
result = Runner.run_streamed(cur_agent, input=messages, context=context_variables)
|
|
|
|
# 状态变量
|
|
current_agent_name = None # 当前Agent名称
|
|
is_printing = False # 是否正在打印文本
|
|
|
|
# 处理事件流
|
|
async for event in result.stream_events():
|
|
|
|
# 事件1: 文本流（逐Token输出）
|
|
if event.type == "raw_response_event":
|
|
if isinstance(event.data, ResponseTextDeltaEvent):
|
|
# 第一次打印时显示Agent标签
|
|
if not is_printing:
|
|
agent_label = f"[{current_agent_name}]" if current_agent_name else ""
|
|
print(f"🤖 {agent_label} ", end="", flush=True)
|
|
is_printing = True
|
|
# 逐字输出
|
|
print(event.data.delta, end="", flush=True)
|
|
|
|
# 事件2: Agent切换
|
|
elif event.type == "agent_updated_stream_event":
|
|
new_agent = event.new_agent.name
|
|
if current_agent_name is None:
|
|
# 第一次设置Agent名称
|
|
current_agent_name = new_agent
|
|
else:
|
|
# Agent发生切换
|
|
if is_printing:
|
|
print() # 先换行
|
|
is_printing = False
|
|
print(f"🔀 [系统]: {current_agent_name} → {new_agent}")
|
|
current_agent_name = new_agent
|
|
|
|
# 事件3: 工具调用
|
|
elif event.type == "run_item_stream_event":
|
|
# 工具调用开始
|
|
if event.name == "tool_called":
|
|
if is_printing:
|
|
print() # 先换行
|
|
is_printing = False
|
|
tool_name = event.item.raw_item.name
|
|
tool_args = event.item.raw_item.arguments # 获取工具参数 调试需要时可加入
|
|
# 区分转接工具和业务工具
|
|
if tool_name.startswith("transfer_"):
|
|
print(f"📞 [转接]: {tool_name}")
|
|
else:
|
|
print(f"🔧 [工具]: {tool_name}")
|
|
|
|
# 工具输出结果
|
|
elif event.name == "tool_output":
|
|
print(f"✅ [结果]: {event.item.output}")
|
|
is_printing = False
|
|
|
|
# 结束本轮，确保换行
|
|
if is_printing:
|
|
print()
|
|
|
|
# 更新状态
|
|
messages = result.to_input_list() # 获取完整对话历史
|
|
cur_agent = result.last_agent # 获取最后激活的Agent
|
|
|
|
|
|
if __name__ == '__main__':
|
|
asyncio.run(main()) |