a bg@sHddlZddlmZmZmZmZddlmZmZm Z GdddeZ dS)N)AnyCallableDictOptional) BaseModel ConfigDictmodel_validatorc@seZdZUdZdZeeeefed<dZ ee ed<eed<dZ e ed<e ege fed<edd Zed d eee d 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_functionZforbid)extrabefore)mode)valuesreturnc Csz ddlm}m}ddlm}Wn2tyR}ztd|WYd}~n d}~00|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 clientrs/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/utilities/graphql.pyvalidate_environments*  z&GraphQLAPIWrapper.validate_environment)queryrcCs||}tj|ddS)z(Run a GraphQL query and get the results.)indent)_execute_queryjsondumps)selfr"resultrrr run1s zGraphQLAPIWrapper.runcCs||}|j|}|S)z/Execute a GraphQL query and return the results.)rr execute)r(r"Z document_noder)rrr r%6s  z GraphQLAPIWrapper._execute_query)__name__ __module__ __qualname____doc__r rrstr__annotations__r boolr rrrZ model_configr classmethodr!r*r%rrrr r s  r ) r&typingrrrrZpydanticrrrr rrrr s