diff --git a/libs/langchain/langchain/agents/load_tools.py b/libs/langchain/langchain/agents/load_tools.py index ab3a34cfee..24adaf867d 100644 --- a/libs/langchain/langchain/agents/load_tools.py +++ b/libs/langchain/langchain/agents/load_tools.py @@ -353,9 +353,7 @@ def _get_scenexplain(**kwargs: Any) -> BaseTool: def _get_graphql_tool(**kwargs: Any) -> BaseTool: - graphql_endpoint = kwargs["graphql_endpoint"] - wrapper = GraphQLAPIWrapper(graphql_endpoint=graphql_endpoint) - return BaseGraphQLTool(graphql_wrapper=wrapper) + return BaseGraphQLTool(graphql_wrapper=GraphQLAPIWrapper(**kwargs)) def _get_openweathermap(**kwargs: Any) -> BaseTool: @@ -455,7 +453,7 @@ _EXTRA_OPTIONAL_TOOLS: Dict[str, Tuple[Callable[[KwArg(Any)], BaseTool], List[st ), "stackexchange": (_get_stackexchange, []), "sceneXplain": (_get_scenexplain, []), - "graphql": (_get_graphql_tool, ["graphql_endpoint"]), + "graphql": (_get_graphql_tool, ["graphql_endpoint", "custom_headers"]), "openweathermap-api": (_get_openweathermap, ["openweathermap_api_key"]), "dataforseo-api-search": ( _get_dataforseo_api_search,