Zi%ddlmZmZddlmZmZmZmZmZm Z ddl m Z m Z m Z mZmZmZmZmZmZmZeGddZeGddZeGdd ZGd d Zd S) ) dataclassfield)ListDictAnyUnionSetOptional) KNNFilterGroupByLimit ProjectionScanRankSelectWhereKeyceZdZUeed<dS) CountPlanscanN)__name__ __module__ __qualname__r__annotations___/root/projects/butler/venv/lib64/python3.11/site-packages/chromadb/execution/expression/plan.pyrrs JJJJJrrceZdZUeed<eeZeed<eeZ eed<ee Z e ed<dS)GetPlanrdefault_factoryfilterlimit projectionN) rrrrrrr r#rr$rr%rrrr r sr JJJU6222FF2225///E5///"U:>>>J >>>>>rr cheZdZUeed<eed<eeZeed<ee Z e ed<dS)KNNPlanrknnr!r#r%N) rrrrrr rr r#rr%rrrr'r's^ JJJ HHHU6222FF222"U:>>>J >>>>>rr'cBeZdZdZ 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 dfdZdS)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$selectc|d|_npt|tr||_nSt|trtj||_n$t dt |j|d|_npt|tr||_nSt|trtj||_n$t dt |j|t|_ npt|tr||_ nSt|trtj||_ n$t dt |j|t|_ nt|tr||_ nt|trtj|dd|_ nSt|trtj||_ n$t dt |j|t|_dSt|tr ||_dSt|trtj||_dSt|t"t$fr*tjdt#|i|_dSt 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 =DKK u % % DKK t $ $ /%00DKKZDKKDXZZ  <DJJ d # # DJJ d # # --DJJW$t**BUWW   $YYDNN ' * * %DNN $ ' ' $.x88DNNbhI`bb  =''DKK u % % DKK s # # /EQ*G*GHHDKK t $ $ /%00DKK_eI]__  >!88DLLL  ' ' !DLLL  % % !+F33DLLL s , , !+VT&\\,BCCDLLLhQUV\Q]Q]Qfhh rreturnc|j|jnd|j|jnd|j|j|jdS)z5Return a JSON-serializable dictionary representation.N)r#r,r-r$r.)r2to_dictr8r9r:r<)r?s rrCzSearch.to_dicts04{/Fdk))+++D,0J,BDJ&&(((..00[((**l**,,    rcttjtjtjtjh}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|SY!WXXX +^+     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.selectsHT+++ +^+     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.wheres2^+<     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.ranks2+^+<     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_bys2"++<     rrr0ctt||}t|j|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$rF)rr*r2r8r9r<)r?r$r0 new_limits rr$z Search.limitsDu555 +^<     r)NNNNN)rAr*)r)rrr__doc__r rrrstrrrr rr;rrr r@rCrLrr.r+r,r-r$rrrr*r*'s>>9=6:=A=AOS XXeT#s(^345XuT4S>123X5$sCx.!89: X eT#s(^S89: X vtCH~tCy#c(JKL XXXXt 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*rrrrYs((((((((8888888888888888                           ??????? ? ??????? ?r r r r r r r r r r r