NgddlZddlmZddlmZmZmZejeZ edZ dZ GddZ dS)N)Template)AnyDictOptionalz MATCH ()-[e:`$edge_type`]->() WITH e limit 1 MATCH (m)-[:`$edge_type`]->(n) WHERE id(m) == src(e) AND id(n) == dst(e) RETURN "(:" + tags(m)[0] + ")-[:$edge_type]->(:" + tags(n)[0] + ")" AS rels ceZdZdZ ddeded ed ed ed ed dfdZd efdZddZ e d efdZ ddede e ded efdZddZddeded eeeffdZdS) NebulaGraphaNebulaGraph wrapper for graph operations. NebulaGraph inherits methods from Neo4jGraph to bring ease to the user space. *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. rootnebula 127.0.0.1%spaceusernamepasswordaddressportsession_pool_sizereturnNcZ ddl}ddl}n#t$rtdwxYw||_||_||_||_||_||_| |_ d|_ | dS#t$r} td| d} ~ wwxYw)z*Create a new NebulaGraph wrapper instance.rNz^Please install NebulaGraph Python client and pandas first: `pip install nebula3-python pandas`z!Could not refresh schema. Error: )nebula3pandas ImportErrorrrrrrr_get_session_pool session_poolschemarefresh_schema Exception ValueError) selfrrrrrrrres c/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/graphs/nebula_graph.py__init__zNebulaGraph.__init__$s  NNN MMMM   6   !      !2 2244  F    ! ! ! ! ! F F FDDDEE E Fs  %3B B*B%%B*ct|j|j|j|j|jgs Jdddlm}ddlm }m }ddl m }|}|j |_ ||j|j|j|j|jfg}n#|$rtdwxYw ||n7#|$rtdt"$r}td|d}~wwxYw|S) NzXPlease provide all of the following parameters: username, password, address, port, spacer)SessionPoolConfig)AuthFailedExceptionInValidHostname) SessionPoolz^Could not connect to NebulaGraph database. Please ensure that the address and port are correctzcCould not connect to NebulaGraph database. Please ensure that the username and password are correctz(Error initializing session pool. Error: )allrrrrrnebula3.Configr&nebula3.Exceptionr'r(nebula3.gclient.net.SessionPoolr)rmax_sizer init RuntimeError)r!r&r'r(r)configrr"s r#rzNebulaGraph._get_session_poolFs ]DM4<DJ O     7    544444JJJJJJJJ??????""$$0 &;   , *+ LL    F    M   f % % % %"   K  M M MKKKLL L Ms$#*BB#'B==C1C,,C1c |jdS#t$r(}td|Yd}~dSd}~wwxYw)Nz%Could not close session pool. Error: )rcloserloggerwarning)r!r"s r#__del__zNebulaGraph.__del__nss H   # # % % % % % H H H NNF1FF G G G G G G G G G Hs AA  Ac|jS)z.Returns the schema of the NebulaGraph database)r)r!s r# get_schemazNebulaGraph.get_schemats {rqueryparamsretryc Bddlm}m}ddlm}|pi} |j||}|s3t d| d|d|S#|$r:t d|j td|j t$rn}|tkrL|dz }t d |d td |d |||||cYd }~Std|d }~w||f$rl|tkr]|dz }t d|d td||_||||cYSYd SwxYw)zQuery NebulaGraph database.r)IOErrorExceptionNoValidSessionException)TTransportExceptionz-Error executing query to NebulaGraph. Error: z Query: z zhNo valid session found in session pool. Please consider increasing the session pool size. Current size: z0Error executing query to NebulaGraph. Retrying (/z )... query: z Error: Nz-Connection issue with NebulaGraph. Retrying (z)... to recreate session pool)r,r>r?%nebula3.fbthrift.transport.TTransportr@rexecute_parameter is_succeededr4r5 error_msgrr r0 RETRY_TIMESexecuter) r!r:r;r<r>r?r@resultr"s r#rHzNebulaGraph.executeysOOOOOOOOMMMMMM2, :&88GGF&&(( )$..00))#))) M&    NN:!%!7::    :!%!7::   V V V{"" "!&"")4""#"" "" ||E6599999999 !TQR!T!TUUU#%56 : : :{"" V!&VV)4VVV%)$:$:$<$<!||E6599999#"" :s-A#A88AF?AD(FD((A0FFcnggg}}}|ddD]}|}|gd}|d|d}|d|d} }t|D]O} |d|| | | fP|||d dD]f} | } | gd } |d | d}|d|d} }t|D]O} | d|| | | fP|| |t | d }t|dkr-||dhd|d|d|d|_ dS)z? Refreshes the NebulaGraph schema information. z SHOW TAGSName)tag propertieszDESCRIBE TAG ``FieldTyperMz SHOW EDGES)edgerMzDESCRIBE EDGE `) edge_typerelsrzNode properties: z Edge properties: z Relationships:  N) rH column_valuescastrangerow_sizeappend rel_query substitutelenr)r! tags_schemaedge_types_schema relationshipsrLtag_name tag_schemarpropstypesirRedge_type_name edge_schemas r#rzNebulaGraph.refresh_schemas9;B & << ,,::6BB + +CxxzzH!)<>mF## 1vvzz$$QqTYY[[111 0  0 0 1 0 0+ 0 0 0 r9c|||}|}i}t|D].}||}||}d|D||</|S)N)r<c6g|]}|S)rV).0xs r# z%NebulaGraph.query..s 66616688666r9)rHkeysrWcol_sizerU) r!r:r<rIcolumnsdcol_numcol_namecol_lists r#r:zNebulaGraph.queryse511++--V__..// 7 7Gw'H++H55H66X666AhKKr9)r r r r r)rN)Nr)r)__name__ __module__ __qualname____doc__strintr$rrr6propertyr8rdictrHrrr:rjr9r#r r sz& "!# F F F F F  F  F F  F F F FD&3&&&&PHHHH CX2:2:S2:(4.2:2:TW2:2:2:2:h! ! ! ! F3s4S>r9r ) loggingstringrtypingrrr getLoggerrur4rZrGr rjr9r#rs&&&&&&&&&&  8 $ $ H    EEEEEEEEEEr9