mBNjeddlmZmZmZmZddlmZddlmZddl m Z ddl m Z ddl mZdZd Zd Zd Zd Zd ZdZdZdZGddZy))AnyDictListOptional)SchemaVersionMismatchException) ExecutionPlan) GraphSchema)stringify_param_value) QueryResultz DB.INDEXESzDB.CONSTRAINTSz GRAPH.COPYz GRAPH.QUERYz GRAPH.DELETEz GRAPH.EXPLAINz GRAPH.SLOWLOGz GRAPH.PROFILEzGRAPH.RO_QUERYc JeZdZdZdefdZedefdZ dBdedee ee fd ee d e de f d Z dCdedee ee fd ee de fd Z dCdedee ee fd ee de fd ZdefdZdDdZdZdZdEdedefdZdEdedefdZdeedefdZ dFded e deedeeede f dZdedededede f dZdedede fd Zdedede fd!Zdedede fd"Zdedede fd#Z dedede fd$Z!dedede fd%Z"de fd&Z#dd'dededed(ede f d)Z$dede fd*Z%dede fd+Z&d,d-d.ded/e d0ede fd1Z'd2ede fd3Z(d2ede fd4Z)d,d-d.d2ed/e d0ede fd5Z*d6ededefd7Z+defd8Z,d2efd9Z-defd:Z.d2efd;Z/d6ededefd<Z0defd=Z1d2efd>Z2defd?Z3d2efd@Z4dee ee ffdAZ5y)GGraphz/ Graph, collection of nodes and edges. namecb||_||_t||_|j|_y)z} Create a new graph. Args: client: The client object. name (str): Graph ID N)_nameclientr schemaexecute_command)selfrrs _/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/falkordb/graph.py__init__zGraph.__init__s,  !$' %55returnc|jS)zY Get the graph name. Returns: str: The graph name. )rrs rrz Graph.name+szzrNqparamstimeout read_onlyct|}|j||z}|rtnt}||j|dg}t |t r|j d|gn | td |j|}t||S#t$r,} |jj| j| d} ~ wwxYw)a Executes a query 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 Exceptionrr rrrefreshversion) rrrrrquerycmdcommandresponsees r_queryz Graph._query7s.))&1E9 (lY!499e[A gs # NNIw/ 0  IJ J +t++W5HtX. .-  KK   *G  s'B B7 'B22B7c,|j|||dS)aR Executes a query 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. Frrrr/rrrrs rr*z Graph.queryis({{1VW{NNrc,|j|||dS)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. Tr1r2r3s rro_queryzGraph.ro_querys({{1VW{MMrclonecp|jt|j|t|j|S)z Creates a copy of graph Args: clone (str): Name of cloned graph Returns: Graph: the cloned graph )rCOPY_CMDrrr)rr6s rcopyz Graph.copys+ Xtyy%8T[[%((rcv|jj|jt|jS)z Deletes the graph. See: https://docs.falkordb.com/commands/graph.delete.html Returns: None )rclearr DELETE_CMDrrs rdeletez Graph.deletes+ ##J ;;rcB|jt|jS)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. r SLOWLOG_CMDrrs rslowlogz Graph.slowlogs$##K<rr*c|j||z}|jt|j|}t |S)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. )r!r PROFILE_CMDrr rr*rplans rprofilez Graph.profiles< ))&1E9##KUCT""rc|j||z}|jt|j|}t |S)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. )r!r EXPLAIN_CMDrr rGs rexplainz Graph.explains<))&1E9##KUCT""rc<|yt|ts tdd}|jD]h\}}t|tr|j n t |}|dk(r tdd|vrtd|d|d|dt|d z }j|S) z Build parameters header. Args: params (dict): The parameters. Returns: str: The parameters header. z'params' must be a dictzCYPHER z%Cypher parameter name cannot be empty`z1Cypher parameter name cannot contain a backtick: z= (FalkorDB does not support escaped backticks in identifiers)z`= ) r$dict TypeErroritemsbytesdecodestr ValueErrorr )rr params_headerkeyvaluekey_strs rr!zGraph._build_params_headers >&$'56 6!  ,,.JC&0e&()Invalid entity typeRANGEzDROP INDEX FOR z ON (e.VECTORzDROP VECTOR INDEX FOR FULLTEXTzDROP FULLTEXT INDEX FOR zInvalid index type)rWr*)rrlrmrnropatternrs r _drop_indexzGraph._drop_indexCs & E7!nG F "ugU+G23 3 w !'')A>A  !(  1EA  #*7)79+QGA12 2zz!}rc*|jdd||S)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. rwrqr{rrnros rdrop_node_range_indexzGraph.drop_node_range_indexe BBrc*|jdd||S)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. ryrqr}r~s rdrop_node_fulltext_indexzGraph.drop_node_fulltext_indexr FE9EErc*|jdd||S)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. rxrqr}r~s rdrop_node_vector_indexzGraph.drop_node_vector_index&%CCrc*|jdd||S)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. rwrsr}r~s rdrop_edge_range_indexzGraph.drop_edge_range_indexrrc*|jdd||S)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. ryrsr}r~s rdrop_edge_fulltext_indexzGraph.drop_edge_fulltext_indexrrc*|jdd||S)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. rxrsr}r~s rdrop_edge_vector_indexzGraph.drop_edge_vector_indexrrc,|jtS)zRetrieve a list of graph indices. See: https://docs.falkordb.com/commands/graph.query.html#procedures Returns: list: List of graph indices. )rk GRAPH_INDEXESrs r list_indiceszGraph.list_indicess""=11r)options propertiesc|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|S)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. rqrrrdrsrtrurvrwrNzCREATE z INDEX FOR z ON (rcze.{0}N{z:'z',:}z OPTIONS )rWrgmapformatrSr$rVr*) rrlrmrnrrrzr options_maprYrZs r_create_typed_indexzGraph._create_typed_indexs( & E7!nG F "ugU+G23 3 w HhZ{7)5 9 SXXc'..*5 66 S  K%mmo UeS)3:#5#<  rc \|jdd|j|||dt|g|S)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 rDROPrrrs r_drop_constraintzGraph._drop_constraintsC$t##   II      O    rc,|jdd|g|S)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 rrqrrs rdrop_node_unique_constraintz!Graph.drop_node_unique_constraints %t$$XvuJzJJrc,|jdd|g|S)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 rrrrs rdrop_edge_unique_constraintz!Graph.drop_edge_unique_constraints %t$$X~xU*UUrc,|jdd|g|S)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 rrqrrs rdrop_node_mandatory_constraintz$Graph.drop_node_mandatory_constraints %t$$[&%M*MMrc,|jdd|g|S)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 rrrrs rdrop_edge_mandatory_constraintz$Graph.drop_edge_mandatory_constraints %t$$[.(XZXXrc |jtj}g}|D](}|j|d|d|d|d|dd*|S)z Lists graph's constraints See: https://docs.falkordb.com/commands/graph.constraint-create.html#listing-constraints Returns: [Dict[str, object]]: list of constraints rr)typernr entitytypestatus)rkGRAPH_LIST_CONSTRAINTS result_setappend)rresult constraintsrows rlist_constraintszGraph.list_constraintssh$$%;<GG C   F V"%a&"%a&!!f  r)NNF)NN)rN)N)TNN)6__name__ __module__ __qualname____doc__rVrpropertyrrrobjectr%boolr r/r*r5r9r=rArDr rIrLrQr!rrkr{rrrrrrrrrrrrrrrrrrrrrrrrrrrrrsk 6S 6 c  /3!% 0 0c6k*+0# 0  0  0j/3!% O Oc6k*+O# O  O2/3!% N Nc6k*+N# N  N, )# ) <=( ?#S#-#(#S#-#(8D>cF#$( (B(B(Btn (B tCy! (B  (BX  *- 69 FI   D C3 C3 C; C Fc Fc Fk F DC DC DK D C3 C3 C; C Fc Fc Fk F DC DC DK D2k2 --- -  - -^ MS M+ M P P[ P"#.     !    0 P P[ P S3 S S"#.     !    0 " 14 =@ ,M3M4XcX2 Pc P   " 14 =@ 2 K K VC V NC N Ys Y$tCK'8"9rrN)typingrrrr exceptionsrexecution_planr graph_schemar helpersr query_resultr rrr8r#r<rKr@rFr"rrrrrsX,,6)%*% )        D D r