U h@sDddlZddlmZmZmZmZddlmZmZGdddeZ dS)N)AnyCallableDictOptional) BaseModelroot_validatorc@seZdZUdZdZeeeefed<dZ ee ed<eed<e ed<e ege fed<Gdd d Z ed d eed d dZeedddZeeee fdddZdS)GraphQLAPIWrapperzWrapper around GraphQL API. To use, you should have the ``gql`` python package installed. This wrapper will use the GraphQL API to conduct queries. Ncustom_headersfetch_schema_from_transportgraphql_endpoint gql_client gql_functionc@seZdZdZdS)zGraphQLAPIWrapper.ConfigZforbidN)__name__ __module__ __qualname__extrarrI/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/utilities/graphql.pyConfigsrT)pre)valuesreturnc Csz ddlm}m}ddlm}Wn0tk rP}ztd|W5d}~XYnX|d}||d|d}|d d }|||d } | |d <||d <|S)z;Validate that the python package exists in the environment.r)Clientgql)RequestsHTTPTransportz_Could not import gql python package. Try installing it with `pip install gql`. Received error: Nr r )urlheadersr T) transportr r r )rrZgql.transport.requestsr ImportErrorget) clsrrrrerrr clientrrrvalidate_environments(  z&GraphQLAPIWrapper.validate_environment)queryrcCs||}tj|ddS)z(Run a GraphQL query and get the results.)indent)_execute_queryjsondumps)selfr$resultrrrrun/s zGraphQLAPIWrapper.runcCs||}|j|}|S)z/Execute a GraphQL query and return the results.)r r execute)r*r$Z document_noder+rrrr'4s  z GraphQLAPIWrapper._execute_query)rrr__doc__r rrstr__annotations__r boolrrrrr#r,r'rrrrrs r) r(typingrrrrZlangchain_core.pydantic_v1rrrrrrrs