o g&”i…ã@s<ddlZddlmZddlmZe e¡ZGdd„dƒZdS)éN)ÚSentenceTransformer)ÚRedisCacheRepositoryc@sbeZdZdedededefdd„Zdd„Zd ed ee fd d „Z d eed eee fdd„Z dS)ÚEmbeddingServiceÚ model_nameÚdeviceÚ cache_repoÚ cache_ttlcCs"||_||_d|_||_||_dS)N)rrÚmodelÚcacher)Úselfrrrr©r ú!app/services/embedding_service.pyÚ__init__ s  zEmbeddingService.__init__cCs"|jdurt|j|jd|_|jS)N)r)r rrr)r r r r Ú _get_models zEmbeddingService._get_modelÚtextÚreturncCsP|j |j|¡}|j |¡}|r|S| ¡j|dd ¡}|j |||j¡|S©NT)Únormalize_embeddings) r Ú embedding_keyrÚget_jsonrÚencodeÚtolistÚset_jsonr)r rÚ cache_keyÚcachedÚvecr r r Ú embed_texts zEmbeddingService.embed_textÚtextsc CsÈg}g}g}t|ƒD])\}}|j |j|¡}|j |¡}|r$| |¡q | g¡| |¡| |¡q |rb| ¡j|dd} t|| ƒD]\}} |   ¡} | ||<|j  |j |j||¡| |j ¡qD|Sr) Ú enumerater rrrÚappendrrÚziprrr) r rÚvectorsÚmissing_indicesÚ missing_textsÚiÚtxtÚkeyrÚ generatedrÚvaluesr r r Ú embed_texts!s$     $zEmbeddingService.embed_textsN) Ú__name__Ú __module__Ú __qualname__ÚstrrÚintrrÚlistÚfloatrr)r r r r r s " r) ÚloggingÚsentence_transformersrÚapp.repositories.redis_cacherÚ getLoggerr*Úloggerrr r r r Ús