a bg@sXddlZddlmZddlmZddlmZddlmZddl m Z GdddeZ dS) N)Optional)CallbackManagerForToolRun)BaseTool) ConfigDict)GraphQLAPIWrapperc@sTeZdZUdZeed<dZeed<dZeed<e ddZ 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 }} }} descriptionT)Zarbitrary_types_allowedN) tool_input run_managerreturncCs|j|}tj|ddS)N)indent)rrunjsondumps)selfr r resultrt/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/tools/graphql/tool.py_runs zBaseGraphQLTool._run)N)__name__ __module__ __qualname____doc__r__annotations__r strr rZ model_configrrrrrrrr s   r) rtypingrZlangchain_core.callbacksrZlangchain_core.toolsrZpydanticrZ%langchain_community.utilities.graphqlrrrrrrs