o ȳg@sHddlZddlmZmZmZmZddlmZmZm Z GdddeZ dS)N)AnyCallableDictOptional) BaseModel ConfigDictmodel_validatorc@seZdZUdZdZeeeefed<dZ ee ed<eed<dZ e ed<e ege fed<edd Zed d ed ed e fddZded efddZded eee ffddZdS)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_functionforbid)extrabefore)modevaluesreturnc Cszddlm}m}ddlm}Wnty#}ztd|d}~ww|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)rrgql.transport.requestsr ImportErrorget) clsrrrrerrr clientr!a/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/langchain_community/utilities/graphql.pyvalidate_environments.  z&GraphQLAPIWrapper.validate_environmentquerycCs||}tj|ddS)z(Run a GraphQL query and get the results.)indent)_execute_queryjsondumps)selfr$resultr!r!r"run1s zGraphQLAPIWrapper.runcCs||}|j|}|S)z/Execute a GraphQL query and return the results.)rr execute)r*r$ document_noder+r!r!r"r'6s  z GraphQLAPIWrapper._execute_query)__name__ __module__ __qualname____doc__r rrstr__annotations__r boolr rrr model_configr classmethodr#r,r'r!r!r!r"r s  r ) r(typingrrrrpydanticrrrr r!r!r!r"s