mBNj$fddlmZmZmZmZddlmZddlmZddl m Z ddl m Z ddl mZdZd Zd Zd Zd Zd ZdZdZdZGdde Zy))AnyDictListOptional)SchemaVersionMismatchException) ExecutionPlan)Graph) GraphSchema) QueryResultz DB.INDEXESzDB.CONSTRAINTSz GRAPH.COPYz GRAPH.QUERYz GRAPH.DELETEz GRAPH.EXPLAINz GRAPH.SLOWLOGz GRAPH.PROFILEzGRAPH.RO_QUERYc *eZdZdZdeffd Z d@dedeeeefdee de d e f d Z dAdedeeeefdee d e fd Z dAdedeeeefdee d e fd Zd efdZdBdZdZdZdCded efdZdCded efdZ dDdede deedeeed e f dZdedededed e f dZdeded e fdZdeded e fdZdeded e fd Zdeded e fd!Zdeded e fd"Zdeded e fd#Zd e fd$Zdd%dededed&ed e f d'Z ded e fd(Z!ded e fd)Z"d*d+d,ded-e d.ed e fd/Z#d0ed e fd1Z$d0ed e fd2Z%d*d+d,d0ed-e d.ed e fd3Z&d4ededefd5Z'defd6Z(d0efd7Z)defd8Z*d0efd9Z+d4ededefd:Z,defd;Z-d0efd<Z.defd=Z/d0efd>Z0d eeeeffd?Z1xZ2S)E AsyncGraphz/ Graph, collection of nodes and edges. namecFt|||t||_y)z} Create a new graph. Args: client: The client object. name (str): Graph ID N)super__init__r schema)selfclientr __class__s g/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/falkordb/asyncio/graph.pyrzAsyncGraph.__init__s  &!$' Nqparamstimeout read_onlyreturncK|}|j||z}|rtnt}||j|dg}t |t r|j d|gn | td |j|d{}t|} | j|d{| S7*7#t$r,} |jj| j| d} ~ wwxYww)a Executes a query asynchronously against the graph. See: https://docs.falkordb.com/commands/graph.query.html Args: q (str): The query. params (dict): Query parameters. timeout (int): Maximum query runtime in milliseconds. read_only (bool): Whether the query is read-only. Returns: QueryResult: query result set. z --compactrNz+Timeout argument must be a positive integer)_build_params_header RO_QUERY_CMD QUERY_CMDr isinstanceintextend Exceptionexecute_commandr parserrrefreshversion) rrrrrquerycmdcommandresponse query_resultes r_queryzAsyncGraph._query*s.))&1E9 (lY!499e[A gs # NNIw/ 0  IJ J 1T117;;H&t,L$$X. . . < .-  KK   *G  sNA&C")B*;B&<#B*B( B*%C"&B*(B** C3'CCC"cHK|j|||dd{S7w)aa Executes a query asynchronously against the graph. See: https://docs.falkordb.com/commands/graph.query.html Args: q (str): The query. params (dict): Query parameters. timeout (int): Maximum query runtime in milliseconds. Returns: QueryResult: query result set. FrrrNr0rrrrs rr*zAsyncGraph.query^s%([[67e[TTTT " "cHK|j|||dd{S7w)a_ Executes a read-only query against the graph. See: https://docs.falkordb.com/commands/graph.ro_query.html Args: q (str): The query. params (dict): Query parameters. timeout (int): Maximum query runtime in milliseconds. Returns: QueryResult: query result set. Tr2Nr3r4s rro_queryzAsyncGraph.ro_queryts%([[67d[SSSSr5clonecK|jt|j|d{t|j|S7w)z Creates a copy of graph Args: clone (str): Name of cloned graph Returns: AsyncGraph: the cloned graph N)r&COPY_CMD_namerr)rr8s rcopyzAsyncGraph.copys;""8TZZ???$++u-- @s%AAAcK|jj|jt|jd{S7w)z Deletes the graph. See: https://docs.falkordb.com/commands/graph.delete.html Returns: None N)rclearr& DELETE_CMDr;rs rdeletezAsyncGraph.deletes5 ))*djjAAAAs>AAAc^K|jt|jd{S7w)a Get a list containing up to 10 of the slowest queries issued against the graph. Each item in the list has the following structure: 1. a unix timestamp at which the log entry was processed 2. the issued command 3. the issued query 4. the amount of time needed for its execution, in milliseconds. See: https://docs.falkordb.com/commands/graph.slowlog.html Returns: List: List of slow log entries. Nr& SLOWLOG_CMDr;r@s rslowlogzAsyncGraph.slowlogs%$))+tzzBBBBs $-+-cbK|jt|jdd{y7w)z Reset the slowlog. See: https://docs.falkordb.com/commands/graph.slowlog.html Returns: None RESETNrCr@s r slowlog_resetzAsyncGraph.slowlog_resets$""; GDDDs %/-/r*cK|j||z}|jt|j|d{}t |S7w)a Execute a query and produce an execution plan augmented with metrics for each operation's execution. Return an execution plan, with details on results produced by and time spent in each operation. See: https://docs.falkordb.com/commands/graph.profile.html Args: query (str): The query to profile. params (dict): Query parameters. Returns: ExecutionPlan: The profile information. N)rr& PROFILE_CMDr;rrr*rplans rprofilezAsyncGraph.profilesI ))&1E9))+tzz5IIT""J9A A A cK|j||z}|jt|j|d{}t |S7w)a{ Get the execution plan for a given query. GRAPH.EXPLAIN returns an ExecutionPlan object. See: https://docs.falkordb.com/commands/graph.explain.html Args: query (str): The query for which to get the execution plan. params (dict): Query parameters. Returns: ExecutionPlan: The execution plan. N)rr& EXPLAIN_CMDr;rrKs rexplainzAsyncGraph.explainsI))&1E9))+tzz5IIT""JrN procedureargsemitc>K|xsg}d}t|dkDr'i}t|D]\}}d|}|||<d|z||<d|ddj|d} |%t|dkDr| d dj|z } |j| || d{S7w) a@ Call a procedure. Args: procedure (str): The procedure to call. read_only (bool): Whether the procedure is read-only. args: Procedure arguments. emit: Procedure yield. Returns: QueryResult: The result of the procedure call. Nrparam$zCALL (,)zYIELD )rr)len enumeratejoinr0) rrRrrSrTriarg param_namers rcall_procedurezAsyncGraph.call_procedures,zr t9q=F$D/3$QC[ %(z" *Q* I;a/q 1  D A 6#((4.)* *A[[6Y[GGGGsBBBBidx_type entity_typelabel attributecK|dk(rd|d}n|dk(rd|d}n td|dk(r d |d |d}n)|d k(r d |d |d}n|d k(r d|d |d}n td|j|d{S7w)aDrop a graph index. Args: idx_type (str): The type of index ("RANGE", "FULLTEXT", "VECTOR"). entity_type (str): The type of entity ("NODE" or "EDGE"). label (str): The label of the node or edge. attribute (str): The attribute to drop the index on. Returns: Any: The result of the index dropping query. NODE(e:rZEDGE()-[e:]->()Invalid entity typeRANGEzDROP INDEX FOR z ON (e.VECTORzDROP VECTOR INDEX FOR FULLTEXTzDROP FULLTEXT INDEX FOR zInvalid index typeN) ValueErrorr*)rrbrcrdrepatternrs r _drop_indexzAsyncGraph._drop_indexs& & E7!nG F "ugU+G23 3 w !'')A>A  !(  1EA  #*7)79+QGA12 2ZZ]"""sA0A92A73A9cFK|jdd||d{S7w)aVDrop a range index for a node. See: https://docs.falkordb.com/commands/graph.query.html#deleting-an-index-for-a-node-label Args: label (str): The label of the node. attribute (str): The attribute to drop the index on. Returns: Any: The result of the index dropping query. rmrgNrrrrdres rdrop_node_range_indexz AsyncGraph.drop_node_range_index?%%%gvuiHHHH !!cFK|jdd||d{S7w)aZDrop a full-text index for a node. See: https://docs.falkordb.com/commands/graph.query.html#deleting-an-index-for-a-node-label Args: label (str): The label of the node. attribute (str): The attribute to drop the index on. Returns: Any: The result of the index dropping query. rorgNrtrus rdrop_node_fulltext_indexz#AsyncGraph.drop_node_fulltext_indexL%%%j&%KKKKrxcFK|jdd||d{S7w)aWDrop a vector index for a node. See: https://docs.falkordb.com/commands/graph.query.html#deleting-an-index-for-a-node-label Args: label (str): The label of the node. attribute (str): The attribute to drop the index on. Returns: Any: The result of the index dropping query. rnrgNrtrus rdrop_node_vector_indexz!AsyncGraph.drop_node_vector_indexY%%%hyIIIIrxcFK|jdd||d{S7w)a^Drop a range index for an edge. See: https://docs.falkordb.com/commands/graph.query.html#deleting-an-index-for-a-relationship-type Args: label (str): The label of the edge. attribute (str): The attribute to drop the index on. Returns: Any: The result of the index dropping query. rmriNrtrus rdrop_edge_range_indexz AsyncGraph.drop_edge_range_indexfrwrxcFK|jdd||d{S7w)abDrop a full-text index for an edge. See: https://docs.falkordb.com/commands/graph.query.html#deleting-an-index-for-a-relationship-type Args: label (str): The label of the edge. attribute (str): The attribute to drop the index on. Returns: Any: The result of the index dropping query. roriNrtrus rdrop_edge_fulltext_indexz#AsyncGraph.drop_edge_fulltext_indexsr{rxcFK|jdd||d{S7w)a_Drop a vector index for an edge. See: https://docs.falkordb.com/commands/graph.query.html#deleting-an-index-for-a-relationship-type Args: label (str): The label of the edge. attribute (str): The attribute to drop the index on. Returns: Any: The result of the index dropping query. rnriNrtrus rdrop_edge_vector_indexz!AsyncGraph.drop_edge_vector_indexr~rxcHK|jtd{S7w)zRetrieve a list of graph indices. See: https://docs.falkordb.com/commands/graph.query.html#procedures Returns: list: List of graph indices. N)ra GRAPH_INDEXESr@s r list_indiceszAsyncGraph.list_indicess((7777r5)options propertiescK|dk(rd|d}n|dk(rd|d}n td|dk(rd }d |d |d }|d jtdj|z }|dz }|`d}|j D];\} } t | t r|| dz| zdzz }%|| dzt | zd zz }=|dddz}|d|z }|j|d{S7w)aCreate a typed index for nodes or edges. Args: idx_type (str): The type of index ("RANGE", "FULLTEXT", "VECTOR"). entity_type (str): The type of entity ("NODE" or "EDGE"). label (str): The label of the node or edge. properties: Variable number of property names to be indexed. options (dict, optional): Additional options for the index. Returns: Any: The result of the index creation query. rgrhrZrirjrkrlrmzCREATE z INDEX FOR z ON (rYze.{0}N{z:'z',:}z OPTIONS )rpr]mapformatitemsr"strr*) rrbrcrdrrrqr options_mapkeyvalues r_create_typed_indexzAsyncGraph._create_typed_indexs( & E7!nG F "ugU+G23 3 w HhZ{7)5 9 SXXc'..*5 66 S  K%mmo UeS)3:#5#<<>AA A> A A A  AcK |j|g|d{|jdd|g|d{S7!#t$rY*wxYw7w)a? Create edge unique constraint See: https://docs.falkordb.com/commands/graph.constraint-create.html The constraint is created asynchronously, use list constraints to pull on constraint creation status Note: unique constraints require the existence of a range index over the constraint properties, this function will create any missing range indices Args: relation (str): Edge relationship-type to apply constraint to properties: Variable number of property names to constrain Nr RELATIONSHIP)rr%rrrrs rcreate_edge_unique_constraintz(AsyncGraph.create_edge_unique_constraintasn$ .$..xE*E E E-T,, nh 1;   F    rcHK|jdd|g|d{S7w)a Create node mandatory constraint See: https://docs.falkordb.com/commands/graph.constraint-create.html The constraint is created asynchronously, use list constraints to pull on constraint creation status Args: label (str): Node label to apply constraint to properties: Variable number of property names to constrain MANDATORYrgNrrs r create_node_mandatory_constraintz+AsyncGraph.create_node_mandatory_constraint|s*-T,,[&%U*UUUUr5cHK|jdd|g|d{S7w)a Create edge mandatory constraint See: https://docs.falkordb.com/commands/graph.constraint-create.html The constraint is created asynchronously, use list constraints to pull on constraint creation status Args: relation (str): Edge relationship-type to apply constraint to properties: Variable number of property names to constrain rrNrrs r create_edge_mandatory_constraintz+AsyncGraph.create_edge_mandatory_constraints6-T,,  4>    r5c xK|jdd|j|||dt|g|d{S7w)a) Drops a constraint Args: constraint_type (str): Type of constraint to drop entity_type (str): Type of entity to drop constraint from label (str): entity's label / relationship-type properties: entity's properties to remove constraint from rDROPrNrrs r_drop_constraintzAsyncGraph._drop_constraintsQ*T))   II      O      rcHK|jdd|g|d{S7w)aL Drop node unique constraint See: https://docs.falkordb.com/commands/graph.constraint-create.html Note: the constraint supporting range index is not removed Args: label (str): Node label to remove the constraint from properties: properties to remove constraint from rrgNrrs rdrop_node_unique_constraintz&AsyncGraph.drop_node_unique_constraints*+T**8VUPZPPPPr5cHK|jdd|g|d{S7w)aX Drop edge unique constraint See: https://docs.falkordb.com/commands/graph.constraint-create.html Note: the constraint supporting range index is not removed Args: label (str): Edge relationship-type to remove the constraint from properties: properties to remove constraint from rrNrrs rdrop_edge_unique_constraintz&AsyncGraph.drop_edge_unique_constraints6+T** nh 1;    r5cHK|jdd|g|d{S7w)a  Drop node mandatory constraint See: https://docs.falkordb.com/commands/graph.constraint-create.html Args: label (str): Node label to remove the constraint from properties: properties to remove constraint from rrgNrrs rdrop_node_mandatory_constraintz)AsyncGraph.drop_node_mandatory_constraints*+T**;S SSSSr5cHK|jdd|g|d{S7w)a Drop edge mandatory constraint See: https://docs.falkordb.com/commands/graph.constraint-create.html Args: label (str): Edge relationship-type to remove the constraint from properties: properties to remove constraint from rrNrrs rdrop_edge_mandatory_constraintz)AsyncGraph.drop_edge_mandatory_constraints6+T**  4>    r5c K|jtd{j}g}|D](}|j|d|d|d|d|dd*|S7?w)z Lists graph's constraints See: https://docs.falkordb.com/commands/graph.constraint-create.html#listing-constraints Returns: [Dict[str, object]]: list of constraints Nrr )typerdr entitytypestatus)raGRAPH_LIST_CONSTRAINTS result_setappend)rresult constraintsrows rlist_constraintszAsyncGraph.list_constraintssu++,BCCOO C   F V"%a&"%a&!!f  DsAAAA)NNF)NN)rN)N)TNN)3__name__ __module__ __qualname____doc__rrrrobjectr#boolr r0r*r7r<rArErHrrMrQrrarrrvrzr}rrrrrrrrrrrrrrrrrrrrrr __classcell__)rs@rrrs4 (S ( /3!% 2 2c6k*+2# 2  2  2n/3!% U Uc6k*+U# U  U2/3!% T Tc6k*+T# T  T, . . BC( E#3# #(#3# #0#$( (H(H(Htn (H tCy! (H  (HX$#$#$# $#  $#  $#L I I I I LC LC LK L J# J# J+ J I I I I LC LC LK L J# J# J+ J8K8 -#-#-# -#  -# -#^ 3   c ; $#.   !   2 c ;  (#.   !   2 " 14 =@ ,SS4 C 6 VC V s  " 14 =@ 2 Qs Q # T# T  S  T#v+->(?rrN)typingrrrrfalkordb.exceptionsrfalkordb.execution_planrfalkordb.graphr graph_schemar r.r rrr:r!r?rPrDrJr rrrrsZ,,>1 %% )        k k r