j xddlZddlmZddlmZmZmZejeZ edZ dZ GddZ y)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 ddededededededd fd Zdefd Zdd Z e defd Z ddede e dedefdZddZddededeeeffdZy ) 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. spaceusernamepasswordaddressportsession_pool_sizereturnNc2 ddl}ddl}||_||_||_||_||_||_|j|_ d|_ |jy#t$r tdwxYw#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 ImportErrorr r r rr r_get_session_pool session_poolschemarefresh_schema Exception ValueError) selfr r r r rrrres q/var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/langchain_community/graphs/nebula_graph.py__init__zNebulaGraph.__init__$s   !      !2 224  F    !# 6  $ F@DE E Fs#A"A:"A7: BBBct|j|j|j|j|j gsJdddlm}ddlm }m }ddl m }|}|j|_ ||j|j|j |j|jfg} |j!||S#|$r tdwxYw#|$r tdt"$r}td|d}~wwxYw) 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: )allr r r rr nebula3.Configr!nebula3.Exceptionr"r#nebula3.gclient.net.SessionPoolr$rmax_sizerinit RuntimeError)rr!r"r#r$configrrs rrzNebulaGraph._get_session_poolFs         7   5J?"$00 &   ,, *+ L M   f %! F  # K  MGsKL L Ms%7AC 8C CD 7DD c |jjy#t$r"}tj d|Yd}~yd}~wwxYw)Nz%Could not close session pool. Error: )rcloserloggerwarning)rrs r__del__zNebulaGraph.__del__tsC H    # # % H NNB1#F G G Hs AAAc|jS)z.Returns the schema of the NebulaGraph database)r)rs r get_schemazNebulaGraph.get_schemazs{{queryparamsretryc ddlm}m}ddlm}|xsi} |j j ||}|js*tjd|jd|d|S#|$r;tjd|jtd|jt$r^}|tkrB|dz }tjd |d td |d ||j|||cYd }~Std|d }~w||f$r[|tkrO|dz }tjd|d td|j!|_|j|||cYSYy wxYw)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'r9r:%nebula3.fbthrift.transport.TTransportr;rexecute_parameter is_succeededr/r0 error_msgrrr+ RETRY_TIMESexecuter) rr5r6r7r9r:r;resultrs rrCzNebulaGraph.executesOM2, :&&88GF&&($..012#WC) M&  NN!!%!7!7 8:  !!%!7!7 8:   V{" !!&q 6#W%S" ||E6599 #PQRPS!TUU#%56 :{" !!&q 5TV%)$:$:$<!||E6599# :s-AA..AE35AD:E3DA E32E3c`ggg}}}|jdjdD]}|j}|gd}|jd|d}|jd|jd} }t|j D]:} |dj || j| | jf<|j ||jd jdD]} | j} | gd } |jd | d}|jd|jd} }t|j D]:} | dj || j| | jf<|j | |jt j| jd }t|dkDs|j |djd|d|d|d|_ y)z? Refreshes the NebulaGraph schema information. z SHOW TAGSName)tag propertieszDESCRIBE TAG ``FieldTyperHz SHOW EDGES)edgerHzDESCRIBE EDGE `) edge_typerelsrzNode properties: z Edge properties: z Relationships:  N) rC column_valuescastrangerow_sizeappend rel_query substitutelenr)r tags_schemaedge_types_schema relationshipsrGtag_name tag_schemarpropstypesirMedge_type_name edge_schemas rrzNebulaGraph.refresh_schemas9;B & << ,::6BCxxzH!)a@AA??73Q__V5L5E1::<(L)00%(--/58==?1ST)  $ $[ 1 $$~$>mF# 1vz$$QqTYY[1J" }- 123+_B 0 r4c|j||}|j}i}t|jD]9}||}|j |}|D cgc]} | j c} ||<;|Scc} w)N)r7)rCkeysrRcol_sizerPrQ) rr5r7rDcolumnsdcol_numcol_namecol_listxs rr5zNebulaGraph.querys~e51++-V__./Gw'H++H5H-56X1668X6AhK07sA=)rootnebulaz 127.0.0.1i%)rN)Nr)r)__name__ __module__ __qualname____doc__strintrrrr1propertyr3rdictrCrrr5r4rr r s& "!# F F F F  F  F F  FD,3,\H C2:S2:(4.2:2:TW2:h! F3s4S>r4r ) loggingstringrtypingrrr getLoggerror/rUrBr rwr4rr|sE&&   8 $    KKr4