45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
|
import os
|
|||
|
|
import asyncio
|
|||
|
|
|
|||
|
|
|
|||
|
|
from agentscope.agent import ReActAgent
|
|||
|
|
from agentscope.formatter import OllamaChatFormatter
|
|||
|
|
from agentscope.memory import InMemoryMemory
|
|||
|
|
from agentscope.model import OpenAIChatModel
|
|||
|
|
from agentscope.tool import ToolResponse, Toolkit, execute_python_code
|
|||
|
|
from agentscope.message import ImageBlock, Msg, TextBlock, URLSource
|
|||
|
|
|
|||
|
|
LLAMA_CPP_BASE_URL = os.getenv("LLAMA_CPP_BASE_URL", "http://yyplab.site:8033/v1")
|
|||
|
|
LLAMA_CPP_MODEL_NAME = os.getenv("LLAMA_CPP_MODEL_NAME", "Qwen3.5-35B-A3B-UD-Q8_K_XL.gguf")
|
|||
|
|
LLAMA_CPP_API_KEY = os.getenv("LLAMA_CPP_API_KEY", "EMPTY")
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def creating_react_agent() -> None:
|
|||
|
|
"""创建一个 ReAct 智能体并运行一个简单任务。"""
|
|||
|
|
# 准备工具
|
|||
|
|
toolkit = Toolkit()
|
|||
|
|
toolkit.register_tool_function(execute_python_code)
|
|||
|
|
|
|||
|
|
yyp_bot = ReActAgent(
|
|||
|
|
name="yyp_bot",
|
|||
|
|
sys_prompt="你是一个名为 yyp_bot 的助手",
|
|||
|
|
model=OpenAIChatModel(
|
|||
|
|
model_name=LLAMA_CPP_MODEL_NAME,
|
|||
|
|
api_key=LLAMA_CPP_API_KEY,
|
|||
|
|
client_kwargs={"base_url": LLAMA_CPP_BASE_URL}
|
|||
|
|
),
|
|||
|
|
memory=InMemoryMemory(),
|
|||
|
|
formatter=OllamaChatFormatter(),
|
|||
|
|
toolkit=toolkit,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
msg = Msg(
|
|||
|
|
name="user",
|
|||
|
|
content="你好！yyp_bot，用 Python 运行 Hello World。",
|
|||
|
|
role="user",
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
await yyp_bot(msg)
|
|||
|
|
|
|||
|
|
|
|||
|
|
asyncio.run(creating_react_agent())
|