afulara/PythonicRAG-FastAPI-React
0
1�
2 ʱ�g� � � � d dl mZ d dlmZmZ d dlZd dlmZ d dlZd dlZ G d� d� Z e3dk( rX e � Z e ej e j d� � � e ej e j dd g� � � yy)4� )� load_dotenv)� AsyncOpenAI�OpenAIN)�Listc � � e Zd Zd defd�Zdee deee fd�Zdedee fd�Zdee deee fd�Z dedee fd �Z5y6) �EmbeddingModel�embeddings_model_namec �� � t � t j d� | _ t � | _ t
� | _ | j � t d� �| j t _7 || _ y )N�OPENAI_API_KEYzUOPENAI_API_KEY environment variable is not set. Please set it to your OpenAI API key.) r �os�getenv�openai_api_keyr � async_clientr �client�8ValueError�openai�api_keyr )�selfr s ��/Users/aqsa/AI Projects/AIE6 Class Projects/AIE6-DeployPythonicRAG-Fastapi-react-app/backend/aimakerspace/openai_utils/embedding.py�__init__zEmbeddingModel.__init__9 s_ � ��
� �i�i�(8� 9��� '�M��� �h�� � � � � &� �g��
� �,�,���%:��"� � list_of_text�returnc � �� K � | j j j || j �� � d { ��� }|j D �cg c] }|j10 �� c}S 7 �(c c}w �w�N��input�model�r �11embeddings�creater �data� embedding�r r �embedding_responser s r �async_get_embeddingsz#EmbeddingModel.async_get_embeddings sd � �� �#'�#4�#4�#?�#?�#F�#F� �d�&@�&@� $G� $12� 13�� 8J�7N�7N�O��14�$�$�O�O� 15�� P�s! �5A'�A �A'�16A"� A'�"A'�textc � � K � | j j j || j �� � d { ��� }|j d j17 S 7 � �w�Nr r r �r r' r# s r �async_get_embeddingz"EmbeddingModel.async_get_embedding sS � �� � �+�+�6�6�=�=��d� 8� 8� >� 18� 19� � �~�~�a� �*�*�*� 20�s �5A�A� Ac � � | j j j || j �� }|j D �cg c] }|j21 �� c}S c c}w r �r r r! r r" r# r$ s r �get_embeddingsz EmbeddingModel.get_embeddings% sP � � !�[�[� 3� 3� :� :� �d�&@�&@� ;� 22�� 8J�7N�7N�O��23�$�$�O�O��Os � Ac � � | j j j || j �� }|j d j24 S r) r- r* s r �
get_embeddingz EmbeddingModel.get_embedding, sC � ��K�K�*�*�1�1��d� 8� 8� 2� 25� � �~�~�a� �*�*�*r N)ztext-embedding-3-small) �__name__�26__module__� __qualname__�strr r �floatr& r+ r. r0 � r r r r s� � � ;�c� ;�P�t�C�y� P�T�$�u�+�EV� P�+�c� +�d�5�k� +�P�4��9� P��d�5�k�9J� P�+�#� +�$�u�+� +r r �__main__z
Hello, world!zGoodbye, world!)�dotenvr r r r �typingr r �asyncior r1 �embedding_model�print�runr+ r& r6 r r �<module>r> s� �� � &�
� � � �(+� (+�V �z��$�&�O� �+�'�+�+�o�9�9�/�J�27K�L� ��� � � � 0� 0�/�CT�1U� V� 28�� r