aigenrec/luminabackend
0
1�
2 ]R3iA � �f � d dl mZ d dlmZ d dlmZ d dlmZ d dlZd dl Z G d� d� Z3 e4� Z y)� )�TogetherEmbeddings)�settings)�List)�loggerNc �J � e Zd Zd� Zdee deee fd�Zdedee fd�Zy)�EmbeddingServicec � � t j t j d< t t j5 t j �� | _ y )N�TOGETHER_API_KEY)�model�together_api_key)r r6 �os�environr �EMBEDDING_MODEL�7embeddings)�selfs �/D:\lumina\backend\services\embedding_service.py�__init__zEmbeddingService.__init__ s5 � �)1�)B�)B��8�9�%�&�,��*�*� %� 6� 6�10��� �texts�returnc � �R � �K � g }t j � }d}t dt |� |� D ]9 }||||z �|j d�� fd�� � d{ ��� }|j |� �; |S 7 �# t $ r'}t j dt |� � �� � d}~ww xY w�w)z&Generate embeddings for multiple texts� r Nc �: �� �j j � � S )N)r �embed_documents)�batchr s ��r �<lambda>z6EmbeddingService.generate_embeddings.<locals>.<lambda> s �� �D�O�O� ;� ;�E� Br z Error generating embeddings: )11�asyncio�get_running_loop�range�len�run_in_executor�extend� Exceptionr �error�str) r r r �loop�12batch_size�i�batch_embeddings�er s ` @r �generate_embeddingsz$EmbeddingService.generate_embeddings s� �� �� � ��J��+�+�-�D��J��1�c�%�j�*�5�� �a��J��/��)-�)=�)=��B�*� $� � �!�!�"2�3�
6� � � $�� � � �L�L�8��Q���A� B� �� �s; �B'�AA4 �A2�A4 �1B'�2A4 �4 B$�="B � B$�$B'�textc � � K � | j j |� S # t $ r'}t j dt |� � �� � d}~ww xY w�w)z$Generate embedding for a single textz Error generating embedding: N)r � embed_queryr# r r$ r% )r r, r* s r �generate_embeddingz#EmbeddingService.generate_embedding# sJ � �� � ��?�?�.�.�t�4� 4��� � �L�L�7��A��x�@� A� �� �s$ �A� �A� A�"A13�14A�AN) �__name__�15__module__� __qualname__r r r% �floatr+ r/ � r r r r s> � �16��t�C�y� �T�$�u�+�=N� �&�S� �T�%�[� r r ) �langchain_togetherr �config.settingsr �typingr � utils.loggerr r
r r �embedding_servicer4 r r �<module>r: s- �� 1� $� � � � �!� !�F %�&� r