dhskSSKJr SSKrSSKrSSKJr SSKJrJrJ r J r J r J r J r Jr SSKJr SSKJr \(aSSKr"SS\5r"S S \5rg) ) annotationsN)md5) TYPE_CHECKINGAnyDictList NamedTuplePatternTupleUnion) GraphDocument) GraphStorec6\rSrSrSrSSjrS SjrS SjrSrg) AGEQueryExceptionzException for the AGE queries.c[U[5(a%SU;aUSOSUlSU;aUSOSUlgXlSUlg)Nmessageunknowndetails) isinstancedictrr)self exceptions \/var/www/html/shao/venv/lib/python3.13/site-packages/langchain_community/graphs/age_graph.py__init__AGEQueryException.__init__sE i & &3< 3I9Y/yDL3< 3I9Y/yDL$L$DLcUR$N)rrs r get_messageAGEQueryException.get_message ||rcUR$r)rr s r get_detailsAGEQueryException.get_detailsr#r)rrN)rzUnion[str, Dict]returnNoner'str)r'r) __name__ __module__ __qualname____firstlineno____doc__rr!r%__static_attributes__rrrrs(%rrc\rSrSr%SrSSSSSSS .r\R"S 5rS \ S 'S!S"S jjr S#Sjr S$Sjr S%Sjr S&Sjr\S'Sj5rS(SjrS)SjrS*Sjr\S+Sj5r\S,Sj5r\S-Sj5r\S.Sj5r\S/Sj5r04S0Sjjr\S1S2Sjj5r\S3Sj5rS4S5SjjrS rg)6AGEGraph!a Apache AGE wrapper for graph operations. Args: graph_name (str): the name of the graph to connect to or create conf (Dict[str, Any]): the pgsql connection config passed directly to psycopg2.connect create (bool): if True and graph doesn't exist, attempt to create it *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. STRINGDOUBLEINTEGERLISTMAPBOOLEAN)r*floatintlistrboolz [^0-9a-zA-Z]+r label_regexcXlSSKnUR"S 0UD6UlUR 5nSR U5nURU5 UR5nUcU(a>SR U5nURU5 URR5 O[SR U55eURU5 UR5nURUlUR5 SSS5 g![a [S5ef=f!URan [S[U 5S.5eSn A ff=f!,(df  g=f) zCreate a new AGEGraph instance.rNzXCould not import psycopg2 python package. Please install it with `pip install psycopg2`.z9SELECT graphid FROM ag_catalog.ag_graph WHERE name = '{}'zS SELECT ag_catalog.create_graph('{}'); zCould not create the graphrdetailzFGraph "{}" does not exist in the database and "create" is set to Falser1) graph_namepsycopg2 ImportErrorconnect connection _get_cursorformatexecutefetchonecommitErrorrr* Exceptiongraphidrefresh_schema) rrCconfcreaterDcursgraph_id_querydatacreate_statementes rrAGEGraph.__init__Ds_ %  #**2T2    4PVV  LL (==?D|(z*%  %56..0$= &,  ^,}} <
>/+G*-a&+ s<DAE+D,-AED),E<EEE E)cSSKnURRURR S9nUR S5 UR S5 U$![an[S5UeSnAff=f)z4 get cursor, load age extension and set search path rNIUnable to import psycopg2, please install with `pip install -U psycopg2`.)cursor_factoryz LOAD 'age';z.SET search_path = ag_catalog, "$user", public;)psycopg2.extrasrErGcursorextrasNamedTupleCursorrJ)rrDrWr]s rrHAGEGraph._get_cursorsu   " ''x7W7W'X()KL  -  sA A2! A--A2cURS5nU(aUSSO/nURS5nU(aUSSO/nXB4$)z Get all labels of a graph (for both edges and vertices) by querying the graph metadata table directly Returns Tuple[List[str]]: 2 lists, the first containing vertex labels and the second containing edge labels z;MATCH ()-[e]-() RETURN collect(distinct label(e)) as labelsrlabelsz5MATCH (n) RETURN collect(distinct label(n)) as labels)query)re_labels_recordse_labelsn_labels_recordsn_labelss r _get_labelsAGEGraph._get_labelss_ :: M 5E#A&x0":: G 5E#A&x0"!!rc SSKnSn/nUR5nUHnURURUS9nUR U5 UR 5n U Hxn UR[R"U R5S[R"U R5[R"U R5SS.5 Mz M SSS5 U$![an[S5UeSnAff=f!URan[S[U5S.5eSnAff=f!,(df  U$=f) aO Get a set of distinct relationship types (as a list of dicts) in the graph to be used as context by an llm. Args: e_labels (List[str]): a list of edge labels to filter for Returns: List[Dict[str, str]]: relationships as a list of dicts in the format "{'start':, 'type':, 'end':}" rNrZa( SELECT * FROM ag_catalog.cypher('{graph_name}', $$ MATCH (a)-[e:`{e_label}`]->(b) WITH a,e,b LIMIT 3000 RETURN DISTINCT labels(a) AS from, type(e) AS edge, labels(b) AS to LIMIT 10 $$) AS (f agtype, edge agtype, t agtype); rCe_label)starttypeendzError fetching triplesrA)rDrErHrIrCrJfetchallappendjsonloadsfedgetrMrr*) rrerDrW triple_query triple_schemarSlabelqrUds r _get_triplesAGEGraph._get_tripless7        4! ''4??E'RLLO==?D!&,,)-ACC);(, 166(:'+zz!##q'9" " 0S -  B ~~+'?&)!f! 0sFC( D5BDD5( D2 C>>DD2D--D22D55 EcFURU5nURU5$)a= Get a set of distinct relationship types (as a list of strings) in the graph to be used as context by an llm. Args: e_labels (List[str]): a list of edge labels to filter for Returns: List[str]: relationships as a list of strings in the format "(:``)-[:``]->(:``)" )r|_format_triples)rretripless r_get_triples_strAGEGraph._get_triples_strs%##H-##G,,rcTSnUVs/sHo!R"S0UD6PM nnU$s snf)aw Convert a list of relationships from dictionaries to formatted strings to be better readable by an llm Args: triples (List[Dict[str,str]]): a list relationships in the form {'start':, 'type':, 'end':} Returns: List[str]: a list of relationships in the form "(:``)-[:``]->(:``)" z$(:`{start}`)-[:`{type}`]->(:`{end}`)r1)rI)rtriple_templatetriplerxs rrAGEGraph._format_tripless6AHOPf//9&9 PQs%c SSKnSn/nUR5nUHnURURUS9nUR U5 UR5n [05n U Hln [R"U R5R5H7upU RXR [#U 5R$45 M9 Mn U V V s/sH upXS.PM sn n US .nUR'U5 M SSS5 U$![an[S5UeSnAff=f!UR an[S[U5S.5eSnAff=fs sn n f!,(df  U$=f) a Fetch a list of available node properties by node label to be used as context for an llm Args: n_labels (List[str]): a list of node labels to filter for Returns: List[Dict[str, Any]]: a list of node labels and their corresponding properties in the form "{ 'labels': , 'properties': [ { 'property': , 'type': },... ] }" rNrZz SELECT * FROM ag_catalog.cypher('{graph_name}', $$ MATCH (a:`{n_label}`) RETURN properties(a) AS props LIMIT 100 $$) AS (props agtype); )rCn_labelzError fetching node propertiesrApropertyrn) propertiesrbrDrErHrIrCrJrMrr*rpsetrrrspropsitemsaddtypesrnr+rq)rrgrDrWnode_properties_querynode_propertiesrSryrzrUsr{kvnps r_get_node_propertiesAGEGraph._get_node_propertiessz*  !     4!)00#1LLO}}GA!% 177 3 9 9 ;q**T!W-=-=">?@!<KL"L!$!#=!"L# &&r*9" >_ -  0 ~~+'G&)!f$#M5 >RD E&D1 BE&E -E& D. D))D.1EEE E&& E5c SSKnSn/nUR5nUHnURURUS9nUR U5 UR5n [05n U Hln [R"U R5R5H7upU RXR [#U 5R$45 M9 Mn U V V s/sH upXS.PM sn n US .nUR'U5 M SSS5 U$![an[S5UeSnAff=f!UR an[S[U5S.5eSnAff=fs sn n f!,(df  U$=f) a Fetch a list of available edge properties by edge label to be used as context for an llm Args: e_labels (List[str]): a list of edge labels to filter for Returns: List[Dict[str, Any]]: a list of edge labels and their corresponding properties in the form "{ 'labels': , 'properties': [ { 'property': , 'type': },... ] }" rNrZz SELECT * FROM ag_catalog.cypher('{graph_name}', $$ MATCH ()-[e:`{e_label}`]->() RETURN properties(e) AS props LIMIT 100 $$) AS (props agtype); rkzError fetching edge propertiesrAr)rrnr)rrerDrWedge_properties_queryedge_propertiesrSryrzrUrr{rrrs r_get_edge_propertiesAGEGraph._get_edge_propertiesPsz,  !     4!)00#1LLO}}GA!% 177 3 9 9 ;q**T!W-=-=">?@!<KL"L!$!#=!"L! &&r*9" >[ -  , ~~+'G&)!f$#M5 >rcRUR5upURU5nURU5nURU5nSUSUSUR U5S3UlUVs0sH ofSUS_M snUVs0sH ofSUS_M snU0S.Ulg s snfs snf) zf Refresh the graph schema information by updating the available labels, relationships, and properties z4 Node properties are the following: z< Relationship properties are the following: z6 The relationships are the following: z rbrrn) node_props rel_props relationshipsmetadataN)rhr|rrrschemastructured_schema)rrgrerxrrels rrPAGEGraph.refresh_schemas"--/))(3 33H=33H=    m ,-.   ETTObh<L)99OTAPQ2V*b&66Q* " TQs *BB$cUR$)zReturns the schema of the Graph)rr s r get_schemaAGEGraph.get_schemas{{rcUR$)z*Returns the structured schema of the Graph)rr s rget_structured_schemaAGEGraph.get_structured_schemas%%%rcUR5nSU;a"URS5SR5$UR5(dUS;aSU3$URSS5RSS5$) ac Convert a cypher return field to a pgsql select field If possible keep the cypher column name, but create a generic name if necessary Args: field (str): a return field from a cypher query to be formatted for pgsql idx (int): the position of the field in the return statement Returns: str: the field to be used in the pgsql select statement z as )truefalsenullcolumn_(_))stripsplit isnumericreplace)fieldidxs r _get_col_nameAGEGraph._get_col_namest  U?;;v&r*002 2 __  %+D"DSE? "==c*223; ;rcUR5(d [S5eSn[R"SU[RS9n/nUGHWnUR5R 5S;aM([R "SU[R5nU(dMWXVR5SnUR5RS5S RS 5S RS 5S RS 5S RS5nUV s/sH)oR5(dMU R5PM+ n n SU ;a [S5e[U 5H2up[RX5n X;dM!URU 5 M4 GMZ U(dSnOSRSU55nURUUUSS9$s sn f)a Convert a Cyper query to an Apache Age compatible Sql Query. Handles combined queries with UNION/EXCEPT operators Args: query (str) : A valid cypher query, can include UNION/EXCEPT operators graph_name (str) : The name of the graph to query Returns : str : An equivalent pgSql query wrapped with ag_catalog.cypher Raises: ValueError : If query is empty, contain RETURN *, or has invalid field names zEmpty query providedzlSELECT {projection} FROM ag_catalog.cypher('{graph_name}', $$ {query} $$) AS ({fields});z\b(UNION\b|\bEXCEPT)\b)flags)UNIONEXCEPTz\breturn\b(?![^"]*")Ndistinctrzorder byrskiplimit,*z6Apache Age does not support RETURN * in Cypher queriesza agtype, c3(# UH oS3v M g7f)z agtypeNr1).0rs r 'AGEGraph._wrap_query..s"M*WG#4*s)rCrcfields projection)r ValueErrorrer IGNORECASEuppersearchrolower enumerater3rrqjoinrI)rcrCtemplateparts all_fieldspart return_match return_clauserrt clean_filedsrr field_name fields_strs r _wrap_queryAGEGraph._wrap_querys"{{}}34 4 2EO Dzz|!!#'::99%> ((*'#D#K#KE#R"%a& A% sRBD+B<D8DD B9( B44B9<D AD  DD D'NcV/nUR5H2up4SUS[R"U53nURU5 M4 UbHSU;aBUR[ U[ 5(aS[R"U53OSU35 SSR U5-S-$)a_ Convert a dictionary of properties to a string representation that can be used in a cypher query insert/merge statement. Args: properties (Dict[str,str]): a dictionary containing node/edge properties id (Union[str, None]): the id of the node or None if none exists Returns: str: the properties dictionary as a properly formatted string `z`: rzid: {r})rrrdumpsrqrr*r)rrrrrprops r_format_propertiesAGEGraph._format_propertiess$$&DAqcTZZ]O,D LL ' >d*4 LL+5b#+>+>$tzz"~&'d2$K TYYu%%++rcN[R"[RSU5$)z remove any disallowed characters from a label and replace with '_' Args: label (str): the original label Returns: str: the sanitized version of the label r)rsubr3r?)rys rclean_graph_labelsAGEGraph.clean_graph_labelssvvh**C77rc U(dSOSnSnUGHnU(a}URRRS5(dS[URRR S55R 5URRS'URHnURURS'U(aXURURURUR5URURR5S9nOQUR[RUR5URUR5URS9nURU5 M UR GHbnURRURRS'UR"RUR"RS'[RURR5URURR5[RUR"R5URUR"R5[RUR5R%5URUR5S.n UR"S 0U D6nURU5 GMe GM g ) a insert a list of graph documents into the graph Args: graph_documents (List[GraphDocument]): the list of documents to be inserted include_source (bool): if True add nodes for the sources with MENTIONS edges to the entities they mention Returns: None z_ MERGE (n:`{label}` {{`id`: "{id}"}}) SET n = {properties} z MERGE (n:`{label}` {properties}) MERGE (d:Document {d_properties}) MERGE (d)-[:MENTIONS]->(n) z MERGE (from:`{f_label}` {f_properties}) MERGE (to:`{t_label}` {t_properties}) MERGE (from)-[:`{r_label}` {r_properties}]->(to) rzutf-8)ryr d_properties)ryrr)f_label f_propertiest_label t_propertiesr_label r_propertiesNr1)sourcerrr page_contentencode hexdigestnodesrrrIrnrr3rrcrtargetr) rgraph_documentsinclude_sourcenode_insert_queryedge_insert_querydocnodercruinputss radd_graph_documentsAGEGraph.add_graph_documentss=("      #Czz**..t4403 //66w?1ikJJ''-  (,%!-44"ii#'#:#:4??#K%)%<%r'r()r'z psycopg2.extras.NamedTupleCursor)r'zTuple[List[str], List[str]])re List[str]r'List[Dict[str, str]])rerr'r)rrr'r)rgrr'List[Dict[str, Any]])rerr'r)r'r(r))r'r)rr*rr<r'r*)rcr*rCr*r'r*)rr r'r)rcr*rrr'rr)rrrzUnion[str, None]r'r*)ryr*r'r*)F)rzList[GraphDocument]rr>r'r()r+r,r-r.r/rrcompiler?__annotations__rrHrhr|r staticmethodrrrrPrrrrrrrcrrrr0r1rrr3r3!s.  E::o6K6EI<"<"%3<"=A<" <"|"",9v-"$FPEN >&&<<0J J X44l02,\;?,",(8, ,,2 8 8LQK"2K"DHK" K"K"rr3) __future__rrrrhashlibrtypingrrrrr r r r )langchain_community.graphs.graph_documentr &langchain_community.graphs.graph_storerr\rDrNrr3r1rrr&s@" TTTC= $\ "z\ "r