From c4da8d0813f74b5e1ca25e5f79d8b7588d35d89e Mon Sep 17 00:00:00 2001 From: DrKroll <71385874+DrKroll@users.noreply.github.com> Date: Sat, 30 Mar 2024 01:46:24 +0100 Subject: [PATCH] langchain[patch]: load ReadFileTool (#14301) --------- Co-authored-by: Dr. Simon Kroll Co-authored-by: Bagatur <22008038+baskaryan@users.noreply.github.com> Co-authored-by: Eugene Yurtsev Co-authored-by: Bagatur --- libs/langchain/langchain/agents/load_tools.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/libs/langchain/langchain/agents/load_tools.py b/libs/langchain/langchain/agents/load_tools.py index f87d1b8c25e..48712aae69d 100644 --- a/libs/langchain/langchain/agents/load_tools.py +++ b/libs/langchain/langchain/agents/load_tools.py @@ -18,6 +18,7 @@ import warnings from typing import Any, Dict, List, Optional, Callable, Tuple from mypy_extensions import Arg, KwArg +from langchain_community.tools.file_management import ReadFileTool from langchain_core.tools import Tool from langchain_core.language_models import BaseLanguageModel from langchain_core.callbacks import BaseCallbackManager @@ -413,6 +414,10 @@ def _get_google_cloud_texttospeech(**kwargs: Any) -> BaseTool: return GoogleCloudTextToSpeechTool(**kwargs) +def _get_file_management_tool(**kwargs: Any) -> BaseTool: + return ReadFileTool(**kwargs) + + def _get_reddit_search(**kwargs: Any) -> BaseTool: return RedditSearchRun(api_wrapper=RedditSearchAPIWrapper(**kwargs)) @@ -501,6 +506,7 @@ _EXTRA_OPTIONAL_TOOLS: Dict[str, Tuple[Callable[[KwArg(Any)], BaseTool], List[st ), "eleven_labs_text2speech": (_get_eleven_labs_text2speech, ["eleven_api_key"]), "google_cloud_texttospeech": (_get_google_cloud_texttospeech, []), + "read_file": (_get_file_management_tool, []), "reddit_search": ( _get_reddit_search, ["reddit_client_id", "reddit_client_secret", "reddit_user_agent"],