22 lines
645 B
Python
22 lines
645 B
Python
from python_a2a import AIAgentRouter, AgentNetwork
|
|
from langchain_openai import ChatOpenAI
|
|
from conf import settings
|
|
|
|
# 创建网络
|
|
network = AgentNetwork(name="MyNetwork")
|
|
network.add("TicketAgent", "http://127.0.0.1:5010")
|
|
|
|
# 创建模型
|
|
llm = ChatOpenAI(
|
|
base_url=settings.base_url,
|
|
api_key=settings.api_key,
|
|
model=settings.model_name,
|
|
temperature=0.1
|
|
)
|
|
|
|
# 创建路由器
|
|
router = AIAgentRouter(llm_client=llm, agent_network=network)
|
|
agent_name, confidence = router.route_query("预订票")
|
|
print(agent_name, confidence)
|
|
client = network.get_agent(agent_name)
|
|
print(client.ask("预订一张从北京到上海的火车票")) |