3i$SSKJrJr SSKJrJrJrJrJrJ r SSK J r J r J r JrJrJrJrJrJrJr \"SS55r\"SS55r\"SS 55r"S S 5rg ) ) dataclassfield)ListDictAnyUnionSetOptional) KNNFilterLimit ProjectionScanRankSelectValWhereKeyc \rSrSr%\\S'Srg) CountPlanscanN)__name__ __module__ __qualname____firstlineno__r__annotations____static_attributes__r\/home/james-whalen/.local/lib/python3.13/site-packages/chromadb/execution/expression/plan.pyrrs Jr rcb\rSrSr%\\S'\"\S9r\\S'\"\ S9r \ \S'\"\ S9r \ \S'Sr g) GetPlanrdefault_factoryfilterlimit projectionrN)rrrrrrrr r'r r(rr)rrr r!r#r#s4 J62FF2/E5/":>J >r r#cV\rSrSr%\\S'\\S'\"\S9r \\S'\"\ S9r \ \S'Sr g) KNNPlanrknnr%r'r)rN) rrrrrrr rr r'rr)rrr r!r+r+s* J H62FF2":>J >r r+ch\rSrSrSrSS\\\\\ \ 44S\\\ \\ \ 44S\\\ \\ \ 4\ 4S\\\\\ \ 4\\ \\ 44SjjrS \\ \ 44S jrSS jrS \\\ 4S S4S jrS\\\\\ \ 44S S4SjrS\\\ \\ \ 44S S4SjrSS\ S\ S S4SjjrSrg)Search'aPayload 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(selectcUcSUlOn[U[5(aXlOR[U[5(a[R"U5UlO![ S[ U5R35eUcSUlOn[U[5(aX lOR[U[5(a[R"U5UlO![ S[ U5R35eUc[5Ul O[U[5(aX0l O[U[5(a[R"USS.5Ul OR[U[5(a[R"U5Ul O![ S[ U5R35eUc[5Ulg[U[5(aX@lg[U[5(a[R"U5Ulg[U[[ 45(a'[R"S[U505Ulg[ S[ U5R35e) aVInitialize 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)selfr1r2r(r3s r!__init__Search.__init__Ws. =DK u % %K t $ $//%0DKCDKDXDXCYZ  <DJ d # #J d # #-DJA$t*BUBUAVW  ='DK u % %K s # #//EQ*GHDK t $ $//%0DKHeI]I]H^_  >!8DL  ' '!L  % %!++F3DL s , ,!++VT&\,BCDLPQUV\Q]QfQfPgh r returnc URbURR5OSURbURR5OSURR5URR5S.$)z9Convert the Search to a dictionary for JSON serializationN)r'r2r(r3)r8to_dictr>r?rA)rDs r!rISearch.to_dictsd04{{/Fdkk))+D,0JJ,BDJJ&&([[((*ll**,   r c[[R[R[R[R 1S9n[ URURURUS9$)zASelect all predefined keys (document, embedding, metadata, score)r7r1r2r(r3) rrDOCUMENT EMBEDDINGMETADATASCOREr/r8r>r?)rD new_selects r! select_allSearch.select_allsG#,, s||SYY!WX ++DJJdkk*  r r7cz[[U5S9n[URURUR US9$)zSelect specific keys Args: *keys: Variable number of Key objects or string key names Returns: New Search object with updated select configuration rLrM)rrCr/r8r>r?)rDr7rRs r!r3 Search.selects4T+ ++DJJdkk*  r c*UcSnOe[U[5(aUnOM[U[5(a[R"U5nO![ S[ U5R 35e[X RURURS9$)abSet 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}}]}) Nr5rM) r9rr:r;r<r=rr/r>r?rA)rDr1converted_wheres r!r1 Search.wheres ="O u % %#O t $ $#ooe4OCDKDXDXCYZ ! $++dll  r rank_exprc*UcSnOe[U[5(aUnOM[U[5(a[R"U5nO![ S[ U5R 35e[URX RURS9$)aSet 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 rM) r9rr:r;r<r=rr/r8r?rA)rDrZconverted_ranks r!r2 Search.ranks  !N  4 ( (&N  4 ( (!^^I6NFtIG_G_F`a ++N++dll  r r6cf[X!S9n[URURX0RS9$)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) )r6r(rM)r r/r8r>rA)rDr(r6 new_limits r!r( Search.limits.5 ++DJJi   r )r?r>rAr8)NNNN)rGr/)r)rrrr__doc__r rrrstrrrr r@rrr rErIrSrr3r1r2r(rrr r!r/r/'sX-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   r r/N) dataclassesrrtypingrrrrr r &chromadb.execution.expression.operatorr r r rrrrrrrrr#r+r/rr r!rfst(88      ?? ? ?? ?` ` r