q 9i$ddlmZmZddlmZmZmZmZmZm Z ddl m Z m Z m Z mZmZmZmZmZmZmZeGddZeGddZeGdd ZGd d Zy ) ) dataclassfield)ListDictAnyUnionSetOptional) KNNFilterLimit ProjectionScanRankSelectValWhereKeyceZdZUeed<y) CountPlanscanN)__name__ __module__ __qualname__r__annotations__k/var/www/html/leadgen/airagagent/rag_env/lib/python3.12/site-packages/chromadb/execution/expression/plan.pyrrs JrrcleZdZUeed<eeZeed<eeZ eed<ee Z e ed<y)GetPlanrdefault_factoryfilterlimit projectionN) rrrrrrr r#r r$rr%rrrr r s4 J62FF2/E5/":>J >rr cZeZdZUeed<eed<eeZeed<ee Z e ed<y)KNNPlanrknnr!r#r%N) rrrrrr rr r#rr%rrrr'r's* J H62FF2":>J >rr'cNeZdZdZ ddeeeeee ffdeee eee ffdeee eee fe fdeee eee feeeeffdZdeee ffd Zdd Zd eeefddfd Zdeeeeee ffddfd Zdeee eee ffddfdZdde de ddfdZy)SearchaxPayload for hybrid search operations. Can be constructed directly or using builder pattern: Direct construction with expressions: Search( where=Key("status") == "active", rank=Knn(query=[0.1, 0.2]), limit=Limit(limit=10), select=Select(keys={Key.DOCUMENT}) ) Direct construction with dicts: Search( where={"status": "active"}, rank={"$knn": {"query": [0.1, 0.2]}}, limit=10, # Creates Limit(limit=10, offset=0) select=["#document", "#score"] ) Builder pattern: (Search() .where(Key("status") == "active") .rank(Knn(query=[0.1, 0.2])) .limit(10) .select(Key.DOCUMENT)) Builder pattern with dicts: (Search() .where({"status": "active"}) .rank({"$knn": {"query": [0.1, 0.2]}}) .limit(10) .select(Key.DOCUMENT)) Filter by IDs: Search().where(Key.ID.is_in(["id1", "id2", "id3"])) Combined with metadata filtering: Search().where((Key.ID.is_in(["id1", "id2"])) & (Key("status") == "active")) Empty Search() is valid and will use defaults: - where: None (no filtering) - rank: None (no ranking - results ordered by default order) - limit: No limit - select: Empty selection Nwhererankr$selectcf|d|_ndt|tr||_nLt|trtj||_n!t dt |j|d|_ndt|tr||_nLt|trtj||_n!t dt |j|t|_ nt|tr||_ nzt|trtj|dd|_ nLt|trtj||_ n!t dt |j|t|_yt|tr||_yt|trtj||_yt|tt fr&tjdt|i|_yt dt |j) aInitialize a Search with optional parameters. Args: where: Where expression or dict for filtering results (defaults to None - no filtering) Dict will be converted using Where.from_dict() rank: Rank expression or dict for scoring (defaults to None - no ranking) Dict will be converted using Rank.from_dict() Note: Primitive numbers are not accepted - use {"$val": number} for constant ranks limit: Limit configuration for pagination (defaults to no limit) Can be a Limit object, a dict for Limit.from_dict(), or an int When passing an int, it creates Limit(limit=value, offset=0) select: Select configuration for keys (defaults to empty selection) Can be a Select object, a dict for Select.from_dict(), or a list/set of strings (e.g., ["#document", "#score"]) N1where must be a Where object, dict, or None, got z/rank must be a Rank object, dict, or None, got r)r$offsetz6limit must be a Limit object, dict, int, or None, got keysz>select must be a Select object, dict, list, set, or None, got )_where isinstancerdict from_dict TypeErrortyper_rankrr _limitintr_selectlistset)selfr+r,r$r-s r__init__zSearch.__init__Ws. =DK u %DK t $//%0DKCDKDXDXCYZ  <DJ d #DJ d #-DJA$t*BUBUAVW  ='DK u %DK s #//EQ*GHDK t $//%0DKHeI]I]H^_  >!8DL  '!DL  %!++F3DL s ,!++VT&\,BCDLPQUV\Q]QfQfPgh rreturnc|j|jjnd|j|jjnd|jj|jjdS)z9Convert the Search to a dictionary for JSON serializationN)r#r,r$r-)r2to_dictr8r9r;)r>s rrBzSearch.to_dictsd04{{/Fdkk))+D,0JJ,BDJJ&&([[((*ll**,   rcttjtjtjtj h}t |j|j|j|S)zASelect all predefined keys (document, embedding, metadata, score)r1r+r,r$r-) rrDOCUMENT EMBEDDINGMETADATASCOREr*r2r8r9)r> new_selects r select_allzSearch.select_allsG#,, s||SYY!WX ++DJJdkk*  rr1ctt|}t|j|j|j |S)zSelect specific keys Args: *keys: Variable number of Key objects or string key names Returns: New Search object with updated select configuration rDrE)rr=r*r2r8r9)r>r1rJs rr-z Search.selects4T+ ++DJJdkk*  rc|d}nZt|tr|}nGt|trtj|}n!t dt |j t||j|j|jS)aSet the where clause for filtering Args: where: A Where expression, dict, or None for filtering Dicts will be converted using Where.from_dict() Example: search.where((Key("status") == "active") & (Key("score") > 0.5)) search.where({"status": "active"}) search.where({"$and": [{"status": "active"}, {"score": {"$gt": 0.5}}]}) Nr/rE) r3rr4r5r6r7rr*r8r9r;)r>r+converted_wheres rr+z Search.wheres{ ="O u %#O t $#ooe4OCDKDXDXCYZ ! $++dll  r rank_exprc|d}nZt|tr|}nGt|trtj|}n!t dt |j t|j||j|jS)a Set the ranking expression Args: rank_expr: A Rank expression, dict, or None for scoring Dicts will be converted using Rank.from_dict() Note: Primitive numbers are not accepted - use {"$val": number} for constant ranks Example: search.rank(Knn(query=[0.1, 0.2]) * 0.8 + Val(0.5) * 0.2) search.rank({"$knn": {"query": [0.1, 0.2]}}) search.rank({"$sum": [{"$knn": {"query": [0.1, 0.2]}}, {"$val": 0.5}]}) Nz4rank_expr must be a Rank object, dict, or None, got rE) r3rr4r5r6r7rr*r2r9r;)r>rOconverted_ranks rr,z Search.ranks|  !N  4 (&N  4 (!^^I6NFtIG_G_F`a ++N$++dll  rr0cvt||}t|j|j||jS)zSet the limit and offset for pagination Args: limit: Maximum number of results to return offset: Number of results to skip (default: 0) Example: search.limit(20, offset=10) )r0r$rE)r r*r2r8r;)r>r$r0 new_limits rr$z Search.limits2u5 ++DJJi   r)NNNN)r@r*)r)rrr__doc__r rrrstrrrr r:rrr r?rBrKrr-r+r,r$rrrr*r*'sS-b9=6:=AOS IeT#s(^345IuT4S>123IeT#s(^S89: I vtCH~tCy#c(JKL IV c3h    E#s(O     8E%c3h*?$@A h 8 huT4S>-A'BC  :  3    H  rr*N) dataclassesrrtypingrrrrr r &chromadb.execution.expression.operatorr r r rrrrrrrrr r'r*rrrrYst(88      ?? ? ?? ?` ` r