j*LddlmZddlmZmZmZmZddlmZm Z GddZ y))md5)AnyDictListTuple) GraphDocument Relationshipc eZdZdZ ddedededdfdZedefdZ ifd ed e de e eeffd Z dd Zdd ZdeddfdZdeddfdZ dde ede eeeefdeddfdZy) KuzuGraphuKùzu wrapper for graph operations. *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. dbdatabaseallow_dangerous_requestsreturnNc|dur td ddl}||_|j |j|_||_|jy#t$r tdwxYw)u0Initializes the Kùzu graph database connection.TzThe KuzuGraph class is a powerful tool that can be used to execute arbitrary queries on the database. To enable this functionality, set the `allow_dangerous_requests` parameter to `True` when constructing the KuzuGraph object.rNuSCould not import Kùzu python package.Please install Kùzu with `pip install kuzu`.) ValueErrorkuzu ImportErrorr Connectionconnr refresh_schema)selfr r rrs o/var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/langchain_community/graphs/kuzu_graph.py__init__zKuzuGraph.__init__sz $4 /5    OODGG,     @  s AA)c|jS)u(Returns the schema of the Kùzu database)schemars r get_schemazKuzuGraph.get_schema/s{{queryparamsc |jj||}|j}g}|jrE|j }|j t t|||jrE|S)uQuery Kùzu database)rexecuteget_column_nameshas_nextget_nextappenddictzip)rrr result column_names return_listrows rrzKuzuGraph.query4so""5&1..0  oo//#C   tC c$:; <oorc dg}|jj}|D]}g|d}|jj|}|D]k}||d}d}||ddkDr9d||vr||dD] } |d| dz } nt||dD]} |d z } ||z }|d j ||fm|j |g} |jj } | D]&} | j d | d d | dd| dd(g}| D]} | d}g|d}|jj d|d}|jrA|j}|d}|d}|d j ||f|jrA|j |d|d|d| d|_ y)u,Refreshes the Kùzu graph schema information) propertieslabeltype dimensionrshape[]z[]r.z(:srcz)-[:namez]->(:dst)zCALL table_info('z ') RETURN *;zNode properties: z Relationships properties: z Relationships:  N) r_get_node_table_names_get_node_property_namesranger&_get_rel_table_namesr"r$r%r)rnode_propertiesnode_table_names table_namecurrent_table_schemar. property_name property_typelist_type_flagsi relationships rel_tablestablerel_properties query_resultr, prop_name prop_types rrzKuzuGraph.refresh_schema>s299::<*J24z#J ;;JGJ!+ *= 9& A !#m,[9A=*]";;!+M!:7!CA*!Ah6N"D"'z-'@'M!NA*d2N"O/ $\299%%","  " "#7 8)+, YY335 E  U5\N$uV}oU5<.J  EvJ24z#J 99,,#J<|<L'')"++-F F $\2999i:PQ '')  ! !"6 7  01))7(89+_B 0 rc:|jjdy)Nz CREATE NODE TABLE IF NOT EXISTS Chunk ( id STRING, text STRING, type STRING, PRIMARY KEY(id) ); rr"rs r_create_chunk_node_tablez"KuzuGraph._create_chunk_node_tabless   r node_labelcB|jjd|dy)Nz- CREATE NODE TABLE IF NOT EXISTS zv ( id STRING, type STRING, PRIMARY KEY(id) ); rR)rrTs r_create_entity_node_tablez#KuzuGraph._create_entity_node_tables( --7L9   rrelc |jjd|jd|jjd|jjdy)Nz, CREATE REL TABLE IF NOT EXISTS z ( FROM z TO z ); )rr"r0sourcetarget)rrWs r!_create_entity_relationship_tablez+KuzuGraph._create_entity_relationship_tablesN ,,/HH:6jjoo&d3::??*;<   rgraph_documentsallowed_relationshipsinclude_sourcec  t|Dchc]}|jD]}|j!c}}}|D]}|r|j|jj j dsSt|jjjdj|jj d<|jjd|jj d|jjd|D]}|j||jD]}|jjd|jdd|ji|s>|jd}g} |D]}| jd |tt!| } |d j#| z }|d z }|r|jj||j|vs|jjd |jd |jj d|jd|j$D]} |j'| | jj} | jj} | j(j} | j(j}|jjd| d| d| jd| |dycc}}w)u/ Adds a list of `GraphDocument` objects that represent nodes and relationships in a graph to a Kùzu backend. Parameters: - graph_documents (List[GraphDocument]): A list of `GraphDocument` objects that contain the nodes and relationships to be added to the graph. Each `GraphDocument` should encapsulate the structure of part of the graph, including nodes, relationships, and the source document information. - allowed_relationships (List[Tuple[str, str, str]]): A list of allowed relationships that exist in the graph. Each tuple contains three elements: the source node type, the relationship type, and the target node type. Required for Kùzu, as the names of the relationship tables that need to pre-exist are derived from these tuples. - include_source (bool): If True, stores the source document and links it to nodes in the graph using the `MENTIONS` relationship. This is useful for tracing back the origin of data. Merges source documents based on the `id` property from the source document metadata if available; otherwise it calculates the MD5 hash of `page_content` for merging process. Defaults to False. idzutf-8z MERGE (c:Chunk {id: $id}) SET c.text = $text, c.type = "text_chunk" )r`text) parametersz MERGE (e:zN {id: $id}) SET e.type = "entity" z/CREATE REL TABLE GROUP IF NOT EXISTS MENTIONS (zFROM Chunk TO z, z), label STRING, triplet_source_id STRING)z] MATCH (c:Chunk {id: $id}), (e:z {id: $node_id}) MERGE (c)-[m:MENTIONS]->(e) SET m.triplet_source_id = $id )r`node_idz MATCH (e1:z4 {id: $source_id}), (e2:z4 {id: $target_id}) MERGE (e1)-[:z]->(e2) ) source_id target_idN)listnodesr0rSrYmetadatagetr page_contentencode hexdigestrr"rVr`r&setjoinrJr[rZ)rr\r]r^documentnode node_labelsrTddl table_namesrW source_labelrd target_labelres radd_graph_documentszKuzuGraph.add_graph_documentss<'6 R88>>4TYY>Y R (H--///33D958 44;;GD6ikOO,,T2 !! 'oo66t< ( < <  " * ..z:*! !!"ii[)!%dgg ""113KC"$K&1 #**^J<+HI'2"&s;'7"8K499[11CFFC ))#.yyK/ ))!&&*ii[1 '/oo&>&>t&D+/77(* 3'N --66s;"zz JJMM "zz JJMM  !!+n-!!-/""%((, &/%.  " .E( Ss$L )rF)rN)F)__name__ __module__ __qualname____doc__rstrboolrpropertyrr'rrrrrSrVr r[rrrvrrr r s QV!$IM 2C023d4S>6J3 j    C  D   \ d  % um,u $E#sC-$89u u  urr N) hashlibrtypingrrrr)langchain_community.graphs.graph_documentrr r r~rrrs))QAAr