Ng7ddlZddlmZmZddlmZmZmZmZm Z m Z Gdde Z GddeZ Gdd e ZGd d e ZdS) N)ABCabstractmethod)AnyDictListOptionalTupleUnioncFeZdZdZdeeeffdZdefdZde fdZ dS)NeptuneQueryExceptionz"Exception for the Neptune queries. exceptionct|tr(d|vr|dnd|_d|vr|dnd|_dS||_d|_dS)Nmessageunknowndetails) isinstancedictrr)selfr s d/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/graphs/neptune_graph.py__init__zNeptuneQueryException.__init__ s` i & & %3< 3I3I9Y//yDL3< 3I3I9Y//yDLLL$DL$DLLLreturnc|jSN)rrs r get_messagez!NeptuneQueryException.get_message |rc|jSr)rrs r get_detailsz!NeptuneQueryException.get_detailsrrN) __name__ __module__ __qualname____doc__r strrrrrrrrr r st,,%%T "2%%%%SSrr c&eZdZdZedefdZeifdededefdZ ede fdZ de e ee effdZd e ede efd Zd e ed e de fd Zd e ed e eefde fdZddZdS)BaseNeptuneGraphz Abstract base class for Neptune.rc|jS)z)Return the schema of the Neptune database)schemars r get_schemazBaseNeptuneGraph.get_schemas {rqueryparamsctrNotImplementedError)rr+r,s rr+zBaseNeptuneGraph.query !###rctrr.rs r _get_summaryzBaseNeptuneGraph._get_summary$r0rcR|}|d}|d}||fS)z(b) WITH a,e,b LIMIT 3000 RETURN DISTINCT labels(a) AS from, type(e) AS edge, labels(b) AS to LIMIT 10 z(:`{a}`)-[:`{e}`]->(:`{b}`)e_labelfromredgeto)aeb)formatr+append) rr8 triple_querytriple_template triple_schemalabelqdatadtriples r _get_tripleszBaseNeptuneGraph._get_triples/s 8  - -E##E#22A::a==D - -(//ilai1T71:0$$V,,,,  - rr7typesc d}g}|D]}||}|||d}ti}|dD]Q} | dD]4\} } || |t | jf5Rd|D|d} || |S)Nz_ MATCH (a:`{n_label}`) RETURN properties(a) AS props LIMIT 100 )n_labelrH propertiesrRpropscg|] \}}||d S)propertytyper%.0kvs r z9BaseNeptuneGraph._get_node_properties..T$HHHdaAq99HHHr)rRlabelsrCr+setitemsaddrWr rD) rr7rNnode_properties_querynode_propertiesrHrIrJsprZr[nps r_get_node_propertiesz%BaseNeptuneGraph._get_node_propertiesDs!   ' 'E%,,U,;;A"$**Q--@@DBA,' 8 8gJ,,..88DAqEE1eDGG$45677778IHaHHHB  " "2 & & & &rc d}g}|D]}||}|||d}ti}|dD]Q} | dD]4\} } || |t | jf5R|d|Dd} || |S)Nzf MATCH ()-[e:`{e_label}`]->() RETURN properties(e) AS props LIMIT 100 r;rQrRrScg|] \}}||d SrUr%rXs rr\z9BaseNeptuneGraph._get_edge_properties..lr]r)rWrRr_) rr8rNedge_properties_queryedge_propertiesrHrIrJrerfrZr[eps r_get_edge_propertiesz%BaseNeptuneGraph._get_edge_properties[s!   ' 'E%,,U,;;A"$**Q--@@DBA,' 8 8gJ,,..88DAqEE1eDGG$45677778HHaHHHB  " "2 & & & &rNcddddddd}|\}}||}|||}|||}d|d |d |d |_d S) zA Refreshes the Neptune graph schema information. STRINGDOUBLEINTEGERLISTMAPBOOLEAN)r$floatintlistrboolz4 Node properties are the following: z< Relationship properties are the following: z6 The relationships are the following: z N)r9rMrhrnr))rrNr7r8rGrdrls r_refresh_schemaz BaseNeptuneGraph._refresh_schemars    "--//())(33 33HeDD33HeDD            r)rN)r r!r"r#rVr$r*rrr+rr2r rr9rMrhrrnrzr%rrr'r'sq** CX/1$$3$$d$$$^$$d$$$^$"U49d3i#78""""T#Y49*T#Yt.T#YtCH~RV.      rr'c eZdZdZ d dededeedeeddf dZifd ed ede eeffd Z de fd Z dS)NeptuneAnalyticsGraphaYNeptune Analytics wrapper for graph operations. Parameters: client: optional boto3 Neptune client credentials_profile_name: optional AWS profile name region_name: optional AWS region, e.g., us-west-2 graph_identifier: the graph identifier for a Neptune Analytics graph Example: .. code-block:: python graph = NeptuneAnalyticsGraph( graph_identifier='' ) *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. Ngraph_identifierclientcredentials_profile_name region_namercP |||_nqddl}|||}n|}||_|r|d||_n|d|_nb#t$rt dt $r=}t |jdkrt d|td |d}~wwxYw | dS#t $r$}td t|d d}~wwxYw) z6Create a new Neptune Analytics graph wrapper instance.Nr profile_namez neptune-graph)rRCould not import boto3 python package. Please install it with `pip install boto3`.UnknownServiceErrorzgNeptuneGraph requires a boto3 version 1.34.40 or greater.Please install it with `pip install -U boto3`.Could not load credentials to authenticate with AWS client. Please check that credentials in the specified profile name are valid.)Could not get schema for Neptune databaserdetail) r~boto3Sessionr} ImportError ExceptionrWr ValueErrorrzr r$)rr}r~rrrsessionrAs rrzNeptuneAnalyticsGraph.__init__s% !$  +7#mm9QmRRGG$mmooG(8%B")..'[#1##DKK#*.."A"ADK   >    Aww#888!E !.     " " " " "   'J!!ff  s0A;A>>"C 8CC!C77 D%D  D%r+r,c6 |j|j||d}tj|dddS#t$r$}tdt|dd}~wwxYw) Query Neptune database. OPEN_CYPHER)graphIdentifier queryString parameterslanguagepayloadzUTF-8results,An error occurred while executing the query.rrN) r~ execute_queryr}jsonloadsreaddecoderr r$)rr+r,resprAs rr+zNeptuneAnalyticsGraph.querys ;,, $ 5!!& -D :d9o2244;;GDDEEiP P   'M"1vv  sA'A** B4BBc. |j|jd}n1#t$r$}t dt |dd}~wwxYw |d}|S#t$r*t d|jdwxYw)Ndetailed)rmodez/Summary API error occurred on Neptune Analyticsr graphSummary,Summary API did not return a valid response.)r~get_graph_summaryr}rr r$contentrrresponserAr6s rr2z"NeptuneAnalyticsGraph._get_summarys {44 $ 5J5HH   ' Q"1vv   ~.GN   'M'/6688  s !$ AA  AA 4B)NNN) r r!r"r#r$rrrrrr+r2r%rrr|r|s<26%) 8888#+3- 8 c] 8  8888t023d38n$drr|ceZdZdZ ddedededed eed eed ed dfd Z ifdede d e eeffdZ d e fdZ dS) NeptuneGrapha Neptune wrapper for graph operations. Parameters: host: endpoint for the database instance port: port number for the database instance, default is 8182 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 sign: optional, whether to sign the request payload, default is True Example: .. code-block:: python graph = NeptuneGraph( host='', port=8182 ) *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. TNhostport use_httpsr~rrsignrc |||_nddl}|||} n|} i} |r|| d<|rdnd} | d|d|| d <|r| jdi| |_n-dd lm} dd lm} | j di| d | | i|_nb#t$rtdt$r=}t|j dkrtd|td|d}~wwxYw | dS#t$r$}tdt|dd}~wwxYw)z,Create a new Neptune graph wrapper instance.Nrrrhttpshttpz://: endpoint_url neptunedata)UNSIGNED)Configconfig)signature_versionrrzgNeptuneGraph requires a boto3 version 1.28.38 or greater.Please install it with `pip install -U boto3`.rrr)r)r~rrbotocorerbotocore.configrrrrWr rrzr r$)rrrrr~rrrrr client_paramsprotocolrrrAs rrzNeptuneGraph.__init__0s0 !$  +7#mm9QmRRGG$mmooG " ?3>M-0&/;77V3;0M0M0M0Mt0M0M n- "0'."P"P-"P"PDKK111111666666"0'.%##'## &vAAA###DK    >    Aww#888!E !.     " " " " "   'J!!ff  s0BB"C;>8C66C;?D ED>>Er+r,c |j|dS#t$r$}tdt |dd}~wwxYw)r)openCypherQueryrrrN)r~execute_open_cypher_queryrr r$)rr+r,rAs rr+zNeptuneGraph.queryxss ;888OO    'M"1vv  s # AA  Ac* |j}n1#t$r$}tdt |dd}~wwxYw |dd}|S#t$r*td|jdwxYw)Nz`Summary API is not available for this instance of Neptune,ensure the engine version is >=1.2.1.0rrrr)r~get_propertygraph_summaryrr r$rrrs rr2zNeptuneGraph._get_summarys {<<>>HH   'A #1vv    y).9GN   'M'/6688  s  A AA A4B)rTNNNT)r r!r"r#r$rwryrrrrrr+r2r%rrrrsD26%)FFFF F  F #+3- Fc]FF FFFFP02  3  d38n    drr)rabcrrtypingrrrrr r rr r'r|rr%rrrs& ########::::::::::::::::I$s s s s s ss s s l,DNNNNN#NNNNNr