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("预订一张从北京到上海的火车票"))