J-jw bUddlZddlZddlmZmZmZej eZda ejdze d<dejfdZ efde dejfdZde d ee d eeed ee d eedeeeff d Zde deededeefdZefde ddfdZefde ddfdZde defdZy)N)CHROMA_PERSIST_DIRDEFAULT_COLLECTION MIN_SCORE_clientreturnct:tjddtjt tatS)NT)parentsexist_ok)path)rrmkdirchromadbPersistentClientstr-/var/www/vps2.regionflexible.com/rag/store.py _get_clientr s2  =++5G1HI Nrnamec<tj|ddiS)Nz hnsw:spacecosine)rmetadata)rget_or_create_collectionrs rget_collectionrs' = 1 1 ) 2 rcollection_nameids embeddings documents metadatasc t|}|j|}t|d t fd|D}t ||z }|j ||||||fS)N)rrc3,K|] }|vsd yw)Nr).0i existing_sets r zupsert..%s8a!<"78s )rrrr)rgetsetsumlenupsert) rrrrrcollexistingaddedskippedr%s @rr+r+st / *DxxCx Hx'L 838 8E#hGKK   '>rquery_embeddingtop_kc t|}|j}|dk(rgSt||}|j|g|gd}g}t |dd|dd|ddD]V\}} } t dd| z } | t kr |j|| | jd d | jd d| d X|jd d|S)Nr)rr distances)query_embeddings n_resultsincluderrr3gg?source chunk_index)textscorer7r9rc |dS)Nr;r)hs rz#similarity_search..Ts AgJrT)keyreverse) rcountminqueryzipmaxrappendr'sort) rr0r1r,rAnresultshitsdocmetadistr;s rsimilarity_searchrN0s / *D JJLE z  E5Ajj)*7G D Q Q Q T4 Ct$ 9   ((8R0#xx q9    $ II&I5 Krc tj|tjd|dy#t$r%}tj d|d|Yd}~yd}~wwxYw)Nu Colección 'z ' eliminada.u#No se pudo eliminar la colección 'z': )rdelete_collectionloggerinfo Exceptionwarning)rexcs rrPrPXs\M ''- l4& 56 Mrgs:HH   8 $,0 " "T )0X.. 2h6I6I cT%[!Cy  Dz   38_ *%%%[% % $Z %P#5MCMM/"c"4"CDr