ugZ UddlZddlZddlmZmZmZmZddlmZeee e e fge fZ dddddd d Zee e fed <d d dZee e fed<GddZGddeZGddeZy)N)AnyCallableDictUnion)MetadatacRttjdt|S)Nz^(l2|cosine|ip)$)boolrematchstrps ]/var/www/openai/venv/lib/python3.12/site-packages/chromadb/segment/impl/vector/hnsw_params.pyr sD*=s1v!FGc"t|tSN isinstanceintr s rrr s jC&8rc"t|tSrrr s rrr s 1c 2rc"t|tSrrr s rrrs 1c*rc"t|tSrrr s rrrs *Q"4rc.t|ttfSr)rrfloatr s rrrsJq3,$?r) hnsw:spacehnsw:construction_efhnsw:search_efhnsw:Mhnsw:num_threadshnsw:resize_factorparam_validatorsc0t|txr|dkDSNrr s rrrsAs!3!=A!=rc0t|txr|dkDSr$rr s rrrsZ3%7%AAE%Ar)hnsw:batch_sizehnsw:sync_thresholdpersistent_param_validatorscbeZdZededeeeffdZedeeefdeee fddfdZ y)Paramsmetadatareturnchi}|jD]\}}|jds|||<|S)Nzhnsw:)items startswith)r,segment_metadataparamvalues r_selectzParams._selects>$NN,LE5(*/ '- r validatorsNc|jD]3\}}||vrtd||||r$td|d|y)zValidates the metadatazUnknown HNSW parameter: z"Invalid value for HNSW parameter: z = N)r/ ValueError)r,r5r2r3s r _validatezParams._validate#s]%NN,LE5J& #;E7!CDD$:e$U+ #EeWCPUw!WXX -r) __name__ __module__ __qualname__ staticmethodrrr rr4 Validatorr8rrr+r+sj ( tCH~  YDcNYS)^8LYQUYYrr+cpeZdZUeed<eed<eed<eed<eed<eed<defdZe ded efd Z y ) HnswParamsspaceconstruction_ef search_efM num_threads resize_factorr,c|xsi}t|jdd|_t|jdd|_t|jdd|_t|jdd|_t|jdtj|_ t|jd d |_ y) Nrl2rdrrr r!g333333?) r getrArrBrCrDmultiprocessing cpu_countrErrF)selfr,s r__init__zHnswParams.__init__6s>rlD9: "8<<0F#LMX\\*:C@AX\\(B/0 LL+_-F-F-H I #8<<0Dc#JKrr-cdtj|}tj|t|S)z1Validate and return only the relevant hnsw params)r@r4r8r")r,r1s rextractzHnswParams.extractAs-&--h7-/?@rN) r9r:r;r __annotations__rrrrOr<rQr>rrr@r@.sS JN F L L ( x  rr@cTeZdZUeed<eed<deffd ZededefdZxZ S)PersistentHnswParams batch_sizesync_thresholdr,ct||t|jdd|_t|jdd|_y)Nr'rIr(i)superrOrrKrUrV)rNr, __class__s rrOzPersistentHnswParams.__init__MsA "hll+M,L0KL/77A&&'7Hr) r9r:r;rrRrrOr<rQ __classcell__)rYs@rrTrTIs>OMM  ( x  rrT)rLr typingrrrrchromadb.typesrr rrr r=r"rRr)r+r@rTr>rrr_s --# eCeO,-t3 4 H82*4? *$sI~&>A5T#y.1 YY(  6 : r