a bgP*@sFddlmZddlmZmZmZmZddlmZm Z GdddZ dS))md5)AnyDictListTuple) GraphDocument Relationshipc@seZdZdZdeeeddddZeedd d Z ifee e e eefd d d Z ddddZddddZeddddZeddddZde ee eeeefeddddZdS) 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. kuzuFN)dbdatabaseallow_dangerous_requestsreturncCs^|durtdz ddl}Wnty6tdYn0||_||j|_||_|dS)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`.) ValueErrorr ImportErrorr Connectionconnr refresh_schema)selfr r r r rs/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/graphs/kuzu_graph.py__init__s   zKuzuGraph.__init__)rcCs|jS)u(Returns the schema of the Kùzu database)schemarrrr get_schema/szKuzuGraph.get_schema)queryparamsrcCsD|j||}|}g}|r@|}|tt||q|S)uQuery Kùzu database)rexecuteZget_column_nameshas_nextget_nextappenddictzip)rrrresultZ column_namesZ return_listrowrrrr4szKuzuGraph.queryc Csg}|j}|D]}g|d}|j|}|D]}||d}d}||ddkrd||vr||dD]} |d| d7}qlnt||dD] } |d 7}q||7}|d ||fq0||qg} |j} | D].} | d | d d | dd| ddqg}| D]n} | d}g|d}|jd|d}|rv|}|d}|d}|d ||fq>||qd|d|d| d|_ dS)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) rZ_get_node_table_namesZ_get_node_property_namesranger Z_get_rel_table_namesrrrr)rZnode_propertiesZnode_table_namesZ table_nameZcurrent_table_schemar%Z property_nameZ property_typeZlist_type_flagsi relationshipsZ rel_tablestableZrel_propertiesZ query_resultr$Z prop_nameZ prop_typerrrr>sX        "   zKuzuGraph.refresh_schemacCs|jddS)Nz CREATE NODE TABLE IF NOT EXISTS Chunk ( id STRING, text STRING, type STRING, PRIMARY KEY(id) ); rrrrrr_create_chunk_node_tablepsz"KuzuGraph._create_chunk_node_table) node_labelrcCs|jd|ddS)Nz- CREATE NODE TABLE IF NOT EXISTS zv ( id STRING, type STRING, PRIMARY KEY(id) ); r9)rr;rrr_create_entity_node_table|s z#KuzuGraph._create_entity_node_table)relrc Cs.|jd|jd|jjd|jjddS)Nz, CREATE REL TABLE IF NOT EXISTS z ( FROM z TO z ); )rrr'sourcetarget)rr=rrr!_create_entity_relationship_tablesz+KuzuGraph._create_entity_relationship_table)graph_documentsallowed_relationshipsinclude_sourcerc Cstdd|D}|D]}|rv||jjdsTt|jjd|jjd<|j j d|jjd|jjdd|D]}| |qz|j D]}|j j d|j d d|jid|r|d }g} |D]}| d |qtt| } |d | 7}|d 7}|r|j ||j |vr|j j d|j d|jjd|jddq|jD]\} || | jj } | jj} | jj } | jj}|j j d| d| d| j d| |ddqRqdS)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. cSsh|]}|jD] }|jqqSr)nodesr').0documentnoderrr z0KuzuGraph.add_graph_documents..idzutf-8z MERGE (c:Chunk {id: $id}) SET c.text = $text, c.type = "text_chunk" )rJtext) 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 )rJZnode_idz MATCH (e1:z4 {id: $source_id}), (e2:z4 {id: $target_id}) MERGE (e1)-[:z]->(e2) ) source_id target_idN)listr:r>metadatagetrZ page_contentencode hexdigestrrr<rDr'rJr setjoinr7r@r?)rrArBrCZ node_labelsrFr;rGZddlZ table_namesr=Z source_labelrMZ target_labelrNrrradd_graph_documentss~             zKuzuGraph.add_graph_documents)r F)F)__name__ __module__ __qualname____doc__rstrboolrpropertyrr!rrrrr:r<rr@rrrVrrrrr s( " 2   r N) hashlibrtypingrrrrZ)langchain_community.graphs.graph_documentrrr rrrrs