_jZ UddlZddlZddlmZmZmZmZddlmZeee e e fge fZ dddddd d Zee e fed <d d dZee e fed<GddZGddeZGddeZdS)N)AnyCallableDictUnion)Metadatac`ttjdt|S)Nz^(l2|cosine|ip)$)boolrematchstrps e/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/chromadb/segment/impl/vector/hnsw_params.pyr s!D*=s1vv!F!FGGc,t|tSN isinstanceintr s rrr sjC&8&8rc,t|tSrrr s rrr s 1c 2 2rc,t|tSrrr s rrrs 1c**rc,t|tSrrr s rrrs*Q"4"4rc:t|ttfSr)rrfloatr s rrrsJq3,$?$?r) hnsw:spacehnsw:construction_efhnsw:search_efhnsw:Mhnsw:num_threadshnsw:resize_factorparam_validatorsc8t|to|dkSNrr s rrrsAs!3!3!=Arc8t|to|dkSr$rr s rrrsZ3%7%7%AAEr)hnsw:batch_sizehnsw:sync_thresholdpersistent_param_validatorsceZdZededeeeffdZedeeefdeee fddfdZ dS)Paramsmetadatareturncri}|D]\}}|dr|||< |S)Nzhnsw:)items startswith)r,segment_metadataparamvalues r_selectzParams._selectsM$NN,, 0 0LE5(( 0*/ 'r validatorsNc|D]A\}}||vrtd||||std|d|BdS)zValidates the metadatazUnknown HNSW parameter: z"Invalid value for HNSW parameter: z = N)r/ ValueError)r,r5r2r3s r _validatezParams._validate#s%NN,, Y YLE5J&& !CE!C!CDDD$:e$U++ Y !We!W!WPU!W!WXXX Y Y Yr) __name__ __module__ __qualname__ staticmethodrrr rr4 Validatorr8rrr+r+s ( tCH~   \ YDcNYS)^8LYQUYYY\YYYrr+cxeZdZUeed<eed<eed<eed<eed<eed<defdZe ded efd Z d S) HnswParamsspaceconstruction_ef search_efM num_threads resize_factorr,c|pi}t|dd|_t|dd|_t|dd|_t|dd|_t|dtj|_ t|d d |_ dS) Nrl2rdrrr r!g333333?) r getrArrBrCrDmultiprocessing cpu_countrErrF)selfr,s r__init__zHnswParams.__init__6s>rlD99:: "8<<0F#L#LMMX\\*:C@@AAX\\(B//00 LL+_-F-H-H I I  #8<<0Dc#J#JKKrr-czt|}t|t|S)z1Validate and return only the relevant hnsw params)r@r4r8r")r,r1s rextractzHnswParams.extractAs6&--h77-/?@@@rN) r9r:r;r __annotations__rrrrOr<rQr>rrr@r@.s JJJNNN FFF L L L L L ( x   \   rr@cZeZdZUeed<eed<deffd ZededefdZxZ S)PersistentHnswParams batch_sizesync_thresholdr,ct|t|dd|_t|dd|_dS)Nr'rIr(i)superrOrrKrUrV)rNr, __class__s rrOzPersistentHnswParams.__init__Ms^ """hll+rrr_s ------------###### eCeO,-t3 4 HG8822**44?? **$sI~&>=AA55T#y.1 YYYYYYYY(        6     :     r