小编Mar*_*ven的帖子

Langchain 工具包可以修改吗?我们可以向 pandas_dataframe_agent 工具包添加工具吗?

我是 langchain 的新手,所以我猜这是可能的,但这表明我对 langchain 中的组件缺乏充分的了解。我已成功创建并使用 Pandas Dataframe Agent 工具包来分析小型数据框。我还成功地使用 SERPAPI 工具来利用 Google 搜索 API 并从互联网上检索答案。我的目标是将这两件事结合到一个代理中......一个代理可以在数据中存在答案时查询数据以获取答案,而在数据中不存在答案时默认使用互联网。我觉得可能有一种方法可以向现有代理 takeit 添加工具。另一种选择似乎是从头开始创建一个新代理,复制 pandas 数据框代理的功能并添加其他工具......但这超出了我目前的技能范围。

给定一个数据框 sales_df:

from langchain.agents import create_pandas_dataframe_agent
from langchain.llms import OpenAI

question = 'Which itemnumber has the most sales?'
agent = create_pandas_dataframe_agent(OpenAI(temperature=0), 
                                      sales_df,
                                      verbose=True,
                                     )
agent.run(question)
Run Code Online (Sandbox Code Playgroud)

此代码成功返回我的 pandas 数据框问题的正确答案。

鉴于上面的项目编号(出于隐私原因,xxx 已被删除):

from langchain.agents import initialize_agent, load_tools, AgentType
from langchain.llms import OpenAI

llm = OpenAI(temperature=0)
tools = load_tools(['serpapi'], serpapi_api_key=SERPAPI_API_KEY)

agent = initialize_agent(
    tools, llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True
)

agent.run("What is the description of the …
Run Code Online (Sandbox Code Playgroud)

agent pandas py-langchain

6
推荐指数
1
解决办法
1069
查看次数

标签 统计

agent ×1

pandas ×1

py-langchain ×1