From 89f5b31923f139c8f2fdf9ffac78cb82f304bd17 Mon Sep 17 00:00:00 2001 From: Kye Date: Fri, 6 Oct 2023 01:18:58 -0400 Subject: [PATCH] clean up Former-commit-id: 00dbee3cc647fe62ae94000d7fd96fecff38f51b --- swarms/tools/autogpt.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/swarms/tools/autogpt.py b/swarms/tools/autogpt.py index 735b4e0b..f0dfaf72 100644 --- a/swarms/tools/autogpt.py +++ b/swarms/tools/autogpt.py @@ -8,6 +8,7 @@ from typing import Optional import pandas as pd from langchain.agents import tool from langchain.agents.agent_toolkits.pandas.base import create_pandas_dataframe_agent +from langchain.chains.qa_with_sources.loading import load_qa_with_sources_chain from langchain.docstore.document import Document ROOT_DIR = "./data/" @@ -15,11 +16,14 @@ ROOT_DIR = "./data/" from langchain.chains.qa_with_sources.loading import BaseCombineDocumentsChain from langchain.text_splitter import RecursiveCharacterTextSplitter from langchain.tools import BaseTool +from langchain.tools.file_management.read import ReadFileTool +from langchain.tools.file_management.write import WriteFileTool from pydantic import Field from swarms.utils.logger import logger + @contextmanager def pushd(new_dir): """Context manager for changing the current working directory."""