lQj%ddlmZmZddlmZmZmZmZmZm Z ddl m Z m Z m Z mZmZmZmZmZmZmZeGddZeGddZeGdd ZGd d Zy ) ) dataclassfield)ListDictAnyUnionSetOptional) KNNFilterGroupByLimit ProjectionScanRankSelectWhereKeyceZdZUeed<y) CountPlanscanN)__name__ __module__ __qualname__r__annotations__g/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/chromadb/execution/expression/plan.pyrrs JrrcleZdZUeed<eeZeed<eeZ eed<ee Z e ed<y)GetPlanrdefault_factoryfilterlimit projectionN) rrrrrrr r#rr$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'ceZdZdZ ddeeeeee ffdeee eee ffdeee eee ffdeee eee fe fdeeeeee feeeeff dZd eee ffd Zdd Zd eeefd dfd Zdeeeeee ffd dfdZdeee eee ffd dfdZdeee eee ffd dfdZdde de d dfdZy)Searcha\Payload for hybrid search operations. Can be constructed by directly providing the parameters, or by using the builder pattern. Examples: 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, select=["#document", "#score"], ) Builder pattern: (Search() .where(Key("status") == "active") .rank(Knn(query=[0.1, 0.2])) .limit(10) .select(Key.DOCUMENT)) Nwhererankgroup_byr$selectcR|d|_ndt|tr||_nLt|trtj||_n!t dt |j|d|_ndt|tr||_nLt|trtj||_n!t dt |j|t|_ 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|t"t$fr&tjdt#|i|_yt d t |j) aWInitialize a Search payload. 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 group_by: GroupBy configuration for grouping and aggregating results (defaults to None) Dict will be converted using GroupBy.from_dict() 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"]) Nz1where must be a Where object, dict, or None, got z/rank must be a Rank object, dict, or None, got z6group_by must be a GroupBy 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 _group_byr_limitintr_selectlistset)selfr+r,r-r$r.s r__init__zSearch.__init__Es4 =DK u %DK t $//%0DKCDKDXDXCYZ  <DJ d #DJ d #-DJA$t*BUBUAVW   $YDN ' *%DN $ '$..x8DNHhI`I`Hab  ='DK u %DK s #//EQ*GHDK t $//%0DKHeI]I]H^_  >!8DL  '!DL  %!++F3DL s ,!++VT&\,BCDLPQUV\Q]QfQfPgh rreturnc:|j|jjnd|j|jjnd|jj|jj|j jdS)z5Return a JSON-serializable dictionary representation.N)r#r,r-r$r.)r2to_dictr8r9r:r<)r?s rrCzSearch.to_dictst04{{/Fdkk))+D,0JJ,BDJJ&&(..0[[((*ll**,   rcttjtjtjtj h}t |j|j|j|j|S)zBSelect all predefined keys (document, embedding, metadata, score).r1r+r,r-r$r.) rrDOCUMENT EMBEDDINGMETADATASCOREr*r2r8r9r:)r? new_selects r select_allzSearch.select_allsQ#,, s||SYY!WX ++^^++   rr1ctt|}t|j|j|j |j |S)zSelect specific keys to return. Args: *keys: Key objects or string key names. Returns: Search: A new Search with updated selection. rErF)rr>r*r2r8r9r:)r?r1rKs rr.z Search.selects>T+ ++^^++   rcrt||j|j|j|jS)aQSet the where clause for filtering. Args: where: Where expression, dict, or None. Example: search.where((Key("status") == "active") & (Key("score") > 0.5)) search.where({"status": "active"}) search.where({"$and": [{"status": "active"}, {"score": {"$gt": 0.5}}]}) rF)r*r8r9r:r<)r?r+s rr+z Search.wheres0^^++<<   r rank_exprcrt|j||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}]}) rF)r*r2r9r:r<)r?rOs rr,z Search.ranks0++^^++<<   rcrt|j|j||j|jS)a#Set the group_by configuration for grouping and aggregating results Args: group_by: A GroupBy object, dict, or None for grouping Dicts will be converted using GroupBy.from_dict() Example: search.group_by(GroupBy( keys=[Key("category")], aggregate=MinK(keys=[Key.SCORE], k=3) )) search.group_by({ "keys": ["category"], "aggregate": {"$min_k": {"keys": ["#score"], "k": 3}} }) rF)r*r2r8r:r<)r?r-s rr-zSearch.group_bys0"++++<<   rr0ct||}t|j|j|j||j S)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$rF)rr*r2r8r9r<)r?r$r0 new_limits rr$z Search.limits<u5 ++^^<<   r)NNNNN)rAr*)r)rrr__doc__r rrrstrrrr rr;rrr r@rCrLrr.r+r,r-r$rrrr*r*'s>9=6:=A=AOS XeT#s(^345XuT4S>123X5$sCx.!89: X eT#s(^S89: X vtCH~tCy#c(JKL Xt c3h    E#s(O  $ 8E%c3h*?$@A h & huT4S>-A'BC  * %c3h0G*H!I h 2 3  H rr*N) dataclassesrrtypingrrrrr r &chromadb.execution.expression.operatorr r r rrrrrrrrr r'r*rrrrYst(88      ?? ? ?? ?r r r