51 lines
1.0 KiB
Python
51 lines
1.0 KiB
Python
from langchain.agents import initialize_agent, AgentType
|
||
from langchain_openai import ChatOpenAI
|
||
from langchain_core.tools import tool
|
||
|
||
from conf import settings
|
||
|
||
|
||
# todo: 第一步：定义工具函数
|
||
@tool
|
||
def add(a: int, b: int) -> int:
|
||
"""
|
||
将数字a与数字b相加
|
||
Args:
|
||
a: 第一个数字
|
||
b: 第二个数字
|
||
"""
|
||
return a + b
|
||
|
||
@tool
|
||
def multiply(a: int, b: int) -> int:
|
||
"""
|
||
将数字a与数字b相乘
|
||
Args:
|
||
a: 第一个数字
|
||
b: 第二个数字
|
||
"""
|
||
return a * b
|
||
|
||
# 加载工具
|
||
tools = [add, multiply]
|
||
|
||
# todo: 第二步：初始化模型
|
||
llm = ChatOpenAI(
|
||
base_url=settings.base_url,
|
||
api_key=settings.api_key,
|
||
model=settings.model_name,
|
||
temperature=0.1
|
||
)
|
||
|
||
# todo: 第三步：创建Agent
|
||
agent = initialize_agent(
|
||
tools,
|
||
llm,
|
||
AgentType.STRUCTURED_CHAT_ZERO_SHOT_REACT_DESCRIPTION,
|
||
verbose=True
|
||
)
|
||
|
||
# todo: 第四步：调用Agent
|
||
query = "2+1等于多少？"
|
||
result = agent.invoke(query)
|
||
print(f'result: {result["output"]}') |