使用 csv 文件的聊天机器人

Question

我正在尝试使用 Azure 机器人服务和 Azure 开放式人工智能创建一个聊天机器人。数据源是多个csv文件。我能够使用 langchain 色度扩展创建嵌入。但是在查询嵌入时我没有得到正确的答案。

但是如果我使用 langchain 中的 create_csv_agent ，我会得到所需的响应。我们有什么办法可以使用CSV嵌入并使用它吗？我们无法使用 csv_agent 的主要原因是，目前对于 POC，我们的源是 csv 文件，但后来源可以是 CSV、xls 或 pdf 文件格式。我们正在尝试创建一个通用的嵌入流程来处理所有可能的场景。

CSV 文件的嵌入过程或检索过程有什么特别需要做的吗？

对此的任何指示都会非常有帮助。

提前致谢。

Answer 1

LOADER_MAPPING = {
    ".csv": (CSVLoader, {}),        
    ".pdf": (PyMuPDFLoader, {}),
    ".txt": (TextLoader, {"encoding": "utf8"}),
} 
   
loader_class, loader_args = LOADER_MAPPING[ext]
loader = loader_class(file_path, **loader_args)
loader.load()