jFddlZddlmZddlmZmZmZmZddlm Z ddl m Z m Z m Z ddlmZej eZdZdZd Zd Zd Zd Zd ZdZdZdZdZdZdeeefdeeeffdZ deeeefdeeeefdeeeffdZ!deeefdefdZ"dedefdZ#de$e de%dee&fdZ'de$e de%dee&fd Z(Gd!d"eZ)y)#N)md5)AnyDictListOptional)get_from_dict_or_env) GraphDocumentNode Relationship) GraphStore __Entity__z SHOW SCHEMA INFO z CALL schema.node_type_properties() YIELD nodeType AS label, propertyName AS property, propertyTypes AS type WITH label AS nodeLabels, collect({key: property, types: type}) AS properties RETURN {labels: nodeLabels, properties: properties} AS output a? MATCH (n)-[e]->(m) WITH DISTINCT labels(n) AS start_node_labels, type(e) AS rel_type, labels(m) AS end_node_labels, e, keys(e) AS properties UNWIND CASE WHEN size(properties) > 0 THEN properties ELSE [null] END AS prop WITH start_node_labels, rel_type, end_node_labels, CASE WHEN prop IS NULL THEN [] ELSE [prop, valueType(e[prop])] END AS property_info RETURN start_node_labels, rel_type, end_node_labels, COLLECT(DISTINCT CASE WHEN property_info <> [] THEN property_info ELSE null END) AS properties_info zv UNWIND $data AS row CALL merge.node(row.label, row.properties, {}, {}) YIELD node RETURN distinct 'done' AS result z| UNWIND $data AS row MERGE (source {id: row.source_id}) MERGE (target {id: row.target_id}) RETURN distinct 'done' AS result z UNWIND $data AS row MATCH (source {id: row.source_id}) MATCH (target {id: row.target_id}) WITH source, target, row CALL merge.relationship(source, row.type, {}, {}, target, {}) YIELD rel RETURN distinct 'done' AS result z MERGE (d:Document {id:$document.metadata.id}) SET d.content = $document.page_content SET d += $document.metadata RETURN distinct 'done' AS result z UNWIND $data AS row MATCH (source {id: row.source_id}), (d:Document {id: $document.metadata.id}) MERGE (d)-[:MENTIONS]->(source) RETURN distinct 'done' AS result z1 Node labels and properties (name and type) are: z) Relationship labels and properties are: z7 Nodes are connected with the following relationships: datareturnc|dDcgc]O}|d|dDcgc].}|d|dDcgc]}d|djic}d0c}}|d|dd Qc}}}|d Dcgc]G}|d |dDcgc].}|d|dDcgc]}d|djic}d0c}}d Ic}}}d Scc}wcc}}wcc}}}wcc}wcc}}wcc}}}w)Nedgesend_node_labels propertieskeytypestyperrstart_node_labelsrrrrnodeslabelsrrrr)lower)redgeprop type_itemnodes s/var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/langchain_community/graphs/memgraph_graph.pyget_schema_subsetr$esa$W ! &$((9#:!%\ 2 !3 $E{.2']"-: $Yv%6%<%<%>?-:"!3 &**=%>V  &! @W  &x.!%\ 2 !3 $E{.2']"-: $Yv%6%<%<%>?-:"!3  & '# #"  0"  sQC CB>CC +C <C C(C/C >CC CC rrelsc|Dcgc]=}|d|dDcgc]}|dd|djigd c}|d|dd ?c}}|Dcgc]_}t|d ddg|d Dcgc]9}|d dd d k7r)|d |dDcgc]}d|jic}d;c}}dac}}}dScc}wcc}}wcc}wcc}}wcc}}}w)Nrproperties_inforrrrrel_typerrrrrrr)r_remove_backticks)rr%relr r"r!s r#get_reformated_schemar-sG  $''8#9!$$5 6 6!GQ /H.IJ 6&))<%=J   4 -T(^(:z - labels: z properties: )joinrNODE_PROPS_TEXTREL_PROPS_TEXTREL_TEXT) r.node_props_datarel_props_datarel_datar"r prop_typesprop_types_strr,r) start_labels end_labelss r#transform_schema_to_textr=sONHw\#((4>*B)C3GG   # ,,&D#[[6:7mDm F#mDN U }B~6FbI IO '  gv;xx$7 89 XXc"345 bd8*E*SII |  " Jxj0ABB %D#[[>B7mLm F#))+mLN tE{m2n5ERH HN & 771@Oo -b/=N^ +2#+Hx   24  )E Ms  E (E textc&|jddS)N`r*)replace)r>s r#r+r+s <<R  baseEntityLabelcg}|D]f}|jd|jiz}|rt|jtgnt|jg}||d}|j |h|S)Nid)labelr)rrEr+rBASE_ENTITY_LABELappend)rrCtransformed_nodesr"properties_dictrF node_dicts r#_transform_nodesrLsw//T477O;tyy )+< =#DII./  $?C   + rB relationshipsc\g}|D]}t|j|rtgnt|jjg|jj|rtgnt|j jg|j jd}|j ||S)N)r source_label source_id target_label target_id)r+rrGsourcerEtargetrH)rMrCtransformed_relationshipsr,rel_dicts r#_transform_relationshipsrWs!#%chh/###' 89###' 89  "((2!" %$rBceZdZdZ ddddeedeedeedeeded eed dfd Zdd Z e d efd Z e d eee ffdZ ifdeded eeee ffdZddZ ddeededed dfdZy) MemgraphGraphaMemgraph wrapper for graph operations. Parameters: url (Optional[str]): The URL of the Memgraph database server. username (Optional[str]): The username for database authentication. password (Optional[str]): The password for database authentication. database (str): The name of the database to connect to. Default is 'memgraph'. refresh_schema (bool): A flag whether to refresh schema information at initialization. Default is True. driver_config (Dict): Configuration passed to Neo4j Driver. *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. N) driver_configurlusernamepassworddatabaserefresh_schemarZrc ddl}td|idd}|dk(r|dk(rd}n"td|idd}td |id d }||f}td |id d d }|jj|fd|i|xsi|_||_d|_i|_ |j j|r |jyy#t$r tdwxYw#|jj$r td|jj$r tdwxYw#|jj$r} | d} ~ wwxYw)z-Create a new Memgraph graph wrapper instance.rNzRCould not import neo4j python package. Please install it with `pip install neo4j`.r[ MEMGRAPH_URIr*r\MEMGRAPH_USERNAMEr]MEMGRAPH_PASSWORDr^MEMGRAPH_DATABASEmemgraphauthzMCould not connect to Memgraph database. Please ensure that the url is correctz`Could not connect to Memgraph database. Please ensure that the username and password are correct)neo4j ImportErrorr GraphDatabasedriver_driver _databaser.structured_schemaverify_connectivity exceptionsServiceUnavailable ValueError AuthErrorr_ ClientError) selfr[r\r]r^r_rZrgrfes r#__init__zMemgraphGraph.__init__s  #E3<G r>h"nD+X&#H ,X&#H h'D'  "J0CZ 2u**11   ,2  " 13  LL , , .  ##% e >  L22 8 )) K  ##//  s0CC/D$CAD!$E=D??Ec|jr7tjd|jjd|_yy)NzClosing the driver connection.)rkloggerinfocloserts r#rzzMemgraphGraph.close_s2 << KK8 9 LL   DL rBc|jS)z(Returns the schema of the Graph database)r.r{s r# get_schemazMemgraphGraph.get_schemaes{{rBc|jS)z3Returns the structured schema of the Graph database)rmr{s r#get_structured_schemaz#MemgraphGraph.get_structured_schemajs%%%rBqueryparamscddlm} |jj||j|\}}}|Dcgc]}|j }}|Scc}w#|$r}|j dk(s|j dk(rd|jvsf|j dk(rd|jvsId |jvs;|j d k(rd |jvs|j d k(rd |jvsYd }~nd }~wwxYw|jj|j5} | j||}|Dcgc]}|j ncc}w}}|cd d d S#1swYy xYw)a Query the graph. Args: query (str): The Cypher query to execute. params (dict): The parameters to pass to the query. Returns: List[Dict[str, Any]]: The list of dictionaries containing the query results. r Neo4jError) database_ parameters_z+Neo.DatabaseError.Statement.ExecutionFailedz4Neo.DatabaseError.Transaction.TransactionStartFailedzin an implicit transactionz'Neo.ClientError.Statement.SemanticErrorz&in an open transaction is not possiblez+tried to execute in an explicit transaction0Memgraph.ClientError.MemgraphError.MemgraphErrorzin multicommand transactionsSchemaInfo disabledN)r^) neo4j.exceptionsrrk execute_queryrlrcodemessagesessionrun) rtrrrr_r json_datarurs r#rzMemgraphGraph.queryosX 0" 33.."4JD!Q ,004a4I0 1 "OO66QR5 AFFGG@AIIMHAIIUFFPP7199DFFPP-:5 :\\ ! !4>> ! :g;;uf-D+/04a40I0; : :sA0AAAAC9BC44C9"E#9E E##E,cddl}ddlm}|jdgk(ry |jtdj d}|.t |t|jfr|j|}n|}|Jt|}||_ t||_ y#|$r<}|jdk(r#d|jvrt j#dYd}~nd}~wwxYw|jt$Dcgc]}|d  ncc}w}}|jt&} t)|| }||_ t||_ y) zB Refreshes the Memgraph graph schema information. rNrzMATCH (n) RETURN n LIMIT 1r.rrzSchema generation with SHOW SCHEMA INFO query failed. Set --schema-info-enabled=true to use SHOW SCHEMA INFO query. Falling back to alternative queries.output)astrrr SCHEMA_QUERYget isinstancestrAST literal_evalr$rmr=r.rrrxryNODE_PROPERTIES_QUERY REL_QUERYr-) rtrrresult schema_resultrmrurrr%s r#r_zMemgraphGraph.refresh_schemas3 / ::2 3r 9  ZZ -a044X>F!j#sww&H # 0 0 8 &  , ,, 1- @ %6D "23DEDK  LL)QYY6 ; /3jj9N.OP.OUx.OPPzz)$1%>!2./@A sA?B!!C"&2CC"< D graph_documentsinclude_sourcerCc b|rK|jdtd|jdtd|jdtd|D]L}|r|jjj dsSt |jj jdj|jjd<|jtd|jji|jtd t|j|it|j|}|jt d |i|jt"d |i|s |jt$||jjd O|j'y ) a* Take GraphDocument as input as uses it to construct a graph in Memgraph. 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. - include_source (bool, optional): 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. - baseEntityLabel (bool, optional): If True, each newly created node gets a secondary __Entity__ label, which is indexed and improves import speed and performance. Defaults to False. zCREATE CONSTRAINT ON (b:z) ASSERT b.id IS UNIQUE;zCREATE INDEX ON :z(id);;rEzutf-8documentr)rrN)rrGrSmetadatarr page_contentencode hexdigestINCLUDE_DOCS_QUERY__dict__NODE_IMPORT_QUERYrLrrWrMREL_NODES_IMPORT_QUERYREL_IMPORT_QUERYINCLUDE_DOCS_SOURCE_QUERYr_)rtrrrCrr8s r#add_graph_documentsz!MemgraphGraph.add_graph_documentsso2  JJ*+<*==UV  JJ*+<*=UC D JJ*+<*=Q? @'H//33D958 44;;GD6ikOO,,T2 - HOO6J4l)B\ % % =m,== =  =rBrY)*logginghashlibrtypingrrrrlangchain_core.utilsr)langchain_community.graphs.graph_documentr r r &langchain_community.graphs.graph_storer getLoggerrrxrGrrrrrrrrr3r4r5rr$r-r=r+listrrrLrWrYrrBr#rss,,5WW=   8 $!   0  $DcN$tCH~$N  S#X  '+DcN';  #s(^ F&T#s(^&&R!C!C! DJ  $t* % %%8<% $Z%0KJKrB