servicetoolkit.py

from agentscope.agents.react_agent import ReActAgent

import agentscope
from agentscope.message import Msg
import os
import gradio as gr

from agentscope.service import(
ServiceToolkit,
ServiceResponse,
ServiceExecStatus,
)

# openai_api_key = os.getenv('OPENAI_API_KEY')


# 此Agent的模型配置，按需修改
'''
OPENAI_CFG_DICT ={
"config_name":"openai_cfg",# 此配置的名称，必须保证唯一
"model_type":"openai",# 模型类型
"model_name":"gpt-3.5-turbo",# 模型名称
"api_key": openai_api_key,# OpenAI API key. 如果没有设置，将使用环境变量中的 OPENAI_API_KEY
}
'''

QIANWEN_CFG_DICT = {
        "model_type": "dashscope_chat",
        "config_name": "qwen",
        "model_name": "qwen-72b-chat",
        "api_key": os.environ.get("DASHSCOPE_API_KEY"),
        "generate_args": {
            "temperature": 0.5
        }
    }

def add_name(a: str, b: str)->str:
    """
	计算任何的名字或字符串相加
	

    Args:
        a (str): 参数1
        b (str): 参数2

    Returns:
        str: 结果
    
	"""
    output = a + b + 'hhh'
    status =ServiceExecStatus.SUCCESS
    return ServiceResponse(status, output)


class ToolDemo:
	def __init__(self):
	# Prepare the tools for the agent
	        service_toolkit =ServiceToolkit()
	        service_toolkit.add(add_name)

	        agentscope.init(model_configs=[QIANWEN_CFG_DICT])

	        self.agent =ReActAgent(
	            name="assistant",
	            model_config_name="qwen",
	            verbose=True,
	            service_toolkit=service_toolkit,
	            max_iters=1,)

	def invoke(self, query):
		msg =Msg("user", query, role="user")
		return self.agent(msg)

if __name__ =='__main__':
    tool_demo = ToolDemo()
    response = tool_demo.invoke("woai加xiaoyezi等于几")
    print(response)