NgK(jddlZddlmZddlmZmZmZmZddlZdZ dZ dde e dZ GddZ dS) N)SimpleNamespace)AnyDictOptionalSequencezD SELECT DISTINCT ?elem WHERE { ?elem a owl:DatatypeProperty . } zB SELECT DISTINCT ?elem WHERE { ?elem a owl:ObjectProperty . } )classesrelsdtpropsopropsc4eZdZdZ ddeded ed ed ed eed eedededdfdZ e defdZ e de eeffdZ de eeffdZdede eeffdZde eefddfdZdedeefdZddZdS)NeptuneRdfGraphaNeptune wrapper for RDF graph operations. Args: host: endpoint for the database instance port: port number for the database instance, default is 8182 use_iam_auth: boolean indicating IAM auth is enabled in Neptune cluster use_https: whether to use secure connection, default is True client: optional boto3 Neptune client credentials_profile_name: optional AWS profile name region_name: optional AWS region, e.g., us-west-2 service: optional service name, default is neptunedata sign: optional, whether to sign the request payload, default is True Example: .. code-block:: python graph = NeptuneRdfGraph( host=', port= ) schema = graph.get_schema() OR graph = NeptuneRdfGraph( host=', port= ) schema_elem = graph.get_schema_elements() #... change schema_elements ... graph.load_schema(schema_elem) *Security note*: Make sure that the database connection uses credentials that are narrowly-scoped to only include necessary permissions. Failure to do so may result in data corruption or loss, since the calling code may attempt commands that would result in deletion, mutation of data if appropriately prompted or reading sensitive data if such data is present in the database. The best way to guard against such negative outcomes is to (as appropriate) limit the permissions granted to the credentials used with this tool. See https://python.langchain.com/docs/security for more information. TFN neptunedatahostport use_https use_iam_authclientcredentials_profile_name region_nameservicesignreturnc ||_||_d|d|d|_ |||_nddl} || ||_n| |_i} |r|| d<|rdnd} | d |d|| d <| r|jj|fi| |_n2dd lm} dd l m }|jj|fi| d || i|_nb#t$rtdt$r=}t|jdkrtd|td|d}~wwxYwd|_i|_|dS)Nzhttps://:z/sparqlr) profile_namerhttpshttpz:// endpoint_url)UNSIGNED)Configconfig)signature_versionzRCould not import boto3 python package. Please install it with `pip install boto3`.UnknownServiceErrorzgNeptuneGraph requires a boto3 version 1.28.38 or greater.Please install it with `pip install -U boto3`.zCould not load credentials to authenticate with AWS client. Please check that credentials in the specified profile name are valid.)rrquery_endpointrboto3Sessionsessionbotocorer botocore.configr! ImportError Exceptiontype__name__ ValueErrorschemaschema_elements_refresh_schema)selfrrrrrrrrrr' client_paramsprotocolr r!es h/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/graphs/neptune_rdf_graph.py__init__zNeptuneRdfGraph.__init__Ks)&======0 !$  +7#(==>V=#W#WDLL$)==??DL " ?3>M-0&/;77V3;0M0M0M0Mt0M0M n- "5$,"5g"O"O"O"ODKK111111666666"5$,"5##'## &vAAA###DK    >    Aww#888!E !.   /1 sB.C "D,/8D''D,c|jS)z; Returns the schema of the graph database. )r1r4s r8 get_schemazNeptuneRdfGraph.get_schemas {c|jS)N)r2r;s r8get_schema_elementsz#NeptuneRdfGraph.get_schema_elementss ##r=c8|jdS)z\ Obtain Neptune statistical summary of classes and predicates in the graph. detailed)mode)rget_rdf_graph_summaryr;s r8 get_summaryzNeptuneRdfGraph.get_summarys{00j0AAAr=querycd|i}|}d}|jr|j}|}|j}|j}d}|j} d} t||| |j} ddl m } | d|j || } dd l m }|| ||j| d | jd <| j}ni}d |d <t!jd|j || }t%j|j}|S) z$ Run Neptune query. rENz neptune-db) access_key secret_keytokenregionr) AWSRequestPOST)methodurldataparams) SigV4Authz!application/x-www-form-urlencodedz Content-Type)rMrNheadersrO)rr)get_credentialsget_frozen_credentialsrGrHrIrrbotocore.awsrequestrKr& botocore.authrQadd_authrRrequestsrequestjsonloadstext)r4rE request_datarO request_hdr credentialsrGrHr session_tokenrPcredsrKrYrQqueryres json_resps r8rEzNeptuneRdfGraph.querys\ '    N,6688K%<<>>K$/J$/J"G'-MF#%%#' E 7 6 6 6 6 6 j4#6T&G 0 / / / / / IeWd&6 7 7 @ @ I I I.QGON +!/KKK*MK '#t2Kd   Jx}-- r=r2c i}tD]U}g}||D]0}|d}|d}d|d|d}||1d|||<Vd|dd |d d |d d |d|_dS)z Generates and sets schema from schema_elements. Helpful in cases where introspected schema needs pruning. urilocal ()z, zIn the following, each IRI is followed by the local name and optionally its description in parentheses. The graph supports the following node types: rz1 The graph supports the following relationships: r z9 The graph supports the following OWL object properties: r z7 The graph supports the following OWL data properties: r N) ELEM_TYPESappendjoinr1) r4r2elem_strelemres_listelem_recrerfres_strs r8 load_schemazNeptuneRdfGraph.load_schemas  1 1DH+D1 ) )uo ).c..e...((((!YYx00HTNN $ " $ $  $ $ " $ $! $ $ r=iric"d|vr(|d}|dd|dgSd|vrM|d}d|dt|dz d|dgStd|d)z1 Split IRI into prefix and local #r/zUnexpected IRI 'z ', contains neither '#' nor '/'.)splitrklenr0)r4rrtokenss r8_get_local_namezNeptuneRdfGraph._get_local_names #::YYs^^FayOOOVBZ0 0 CZZYYs^^FxxqVQ 788;;;VBZH HUUUUVV Vr=c"i|jd<|}g}|dddD]d}|}||}||dd}|||d|jdvrd|jd|d<e||jd<g}|ddd D]i}|D]d}|}||}||dd}|||d|jdvrd|jd|d<ej||jd <tD]} t| } | s|| } g}| d d D]p}|d d}||}||dd}|||d|jdvrd|jd|d<q||j| <||jdS)z5 Query Neptune to introspect schema. distinct_prefixespayload graphSummaryrrw)rerfry predicatesr resultsbindingsrmvalueN)r2rDr{rjrigetrErq) r4summaryreslistcrerz elem_recordrprmqitemss r8r3zNeptuneRdfGraph._refresh_schemasu5701""$$#N3I> K KAC))#..F"%q ::K NN; ' ' 'ay 45H IIIGJ$%89&)D*1Y'#N3LA O OA O O--c22&)F1I>> {+++!9D$89L$MMMKND()<=fQiH  O(/V$ 1 1Dt$$A JJqMMEG9%j1 O Oi(--c22&)F1I>> {+++!9D$89L$MMMKND()<=fQiH)0D  & & -.....r=)rTFNNNrT)rN)r/ __module__ __qualname____doc__strintboolrrr9propertyr<rr?rDrErqrr{r3r=r8r r s))\"26%)$EEEE E  E  E#+3-Ec]EEE EEEENCX $T#s(^$$$X$BT#s(^BBBB ++ c3h++++Z 4S> d    : W3 W8C= W W W W././././././r=r ) rZtypesrtypingrrrrrX DTPROP_QUERY OPROP_QUERYrir rr=r8rs !!!!!!000000000000     I/I/I/I/I/I/I/I/I/I/r=