xieqiaoyi/agent-travel/query-info

import asyncio
from agentscope.message import Msg
from agentscope.model import OpenAIChatModel
from config_agentscope import init_agentscope
from config import LLM_CONFIG
from agents.information_query_agent import InformationQueryAgent
import json
async def query_info(user_query: str):
    init_agentscope()
    model = OpenAIChatModel(
        model_name=LLM_CONFIG["model_name"],
        api_key=LLM_CONFIG["api_key"],
        client_kwargs={"base_url": LLM_CONFIG["base_url"], "timeout": 60},
        temperature=LLM_CONFIG.get("temperature", 0.7),
        max_tokens=LLM_CONFIG.get("max_tokens", 2000),
    )
    agent = InformationQueryAgent(name="InformationQueryAgent", model=model)
    user_msg = Msg(name="user", content=user_query, role="user")
    result = await agent.reply(user_msg)
    return json.loads(result.content) if isinstance(result.content, str) else result.content
# 使用
data = asyncio.run(query_info("北京明天天气怎么样？"))
# data: {"query_type": "天气查询"|"网络搜索", "query_success": bool, "results": {"summary": "...", "sources": [...]}}