U h}@sLddlZddlmZddlmZddlmZddlmZGdddeZ dS)N)Optional)CallbackManagerForToolRun)BaseTool)GraphQLAPIWrapperc@sXeZdZUdZeed<dZeed<dZeed<GdddZ d ee e ed d d Z d S)BaseGraphQLToolz%Base tool for querying a GraphQL API.graphql_wrapperZ query_graphqlnamea Input to this tool is a detailed and correct GraphQL query, output is a result from the API. If the query is not correct, an error message will be returned. If an error is returned with 'Bad request' in it, rewrite the query and try again. If an error is returned with 'Unauthorized' in it, do not try again, but tell the user to change their authentication. Example Input: query {{ allUsers {{ id, name, email }} }} descriptionc@seZdZdZdS)zBaseGraphQLTool.ConfigTN)__name__ __module__ __qualname__Zarbitrary_types_allowedr r J/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/tools/graphql/tool.pyConfigsrN) tool_input run_managerreturncCs|j|}tj|ddS)N)indent)rrunjsondumps)selfrrresultr r r_runs zBaseGraphQLTool._run)N) r r r __doc__r__annotations__rstrr rrrrr r r rr s   r) rtypingrZlangchain_core.callbacksrZlangchain_core.toolsrZ%langchain_community.utilities.graphqlrrr r r rs