afulara/PythonicRAG-FastAPI-React
0
1�
2 ʱ�g� � � � S SK Jr S SKJrJr S SKrS SKJr S SKrS SKr " S S5 r \3S:X aX \ " 5 r \ " \R " \ R S5 5 5 \ " \R " \ R SS /5 5 5 gg)4� )� load_dotenv)� AsyncOpenAI�OpenAIN)�Listc � � \ rS rSrS
S\4S jjrS\\ S\\\ 4S jrS\S\\ 4S jr S\\ S\\\ 4S jr5S\S\\ 4S6 jr S r g )�EmbeddingModel� �embeddings_model_namec �� � [ 5 [ R " S5 U l [ 5 U l [
5 U l U R c [ S5 eU R [ l7 Xl g )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_keyr9 )�selfr10 s ڃ/Users/aqsa/AI Projects/AIE6 Class Projects/AIE6-DeployPythonicRAG/fastapi-react-app/backend/aimakerspace/openai_utils/embedding.py�__init__�EmbeddingModel.__init__11 s] � ��
� �i�i�(8� 9��� '�M��� �h�� � � � � &� �g��
� �,�,���%:�"� � list_of_text�returnc � �� # � U R R R XR S9I S h v�N nUR Vs/ s H o3R12 PM sn$ N*s snf 7f�N��input�model�r �13embeddings�creater14 �data� embedding�r r �embedding_responser" s r �async_get_embeddings�#EmbeddingModel.async_get_embeddings se � � �#'�#4�#4�#?�#?�#F�#F� �&@�&@� $G� $15� 16�� 8J�7N�7N�O�7N��$�$�7N�O�O� 17�� P�s! �1A%�A �A%�A �A%� A%�textc � � # � U R R R XR S9I S h v�N nUR S R18 $ N 7f�Nr r r! �r r* r% s r �async_get_embedding�"EmbeddingModel.async_get_embedding sP � � � �+�+�6�6�=�=�� 8� 8� >� 19� 20� � �~�~�a� �*�*�*� 21�s �1A�A� Ac � � U R R R XR S9nUR Vs/ s H o3R22 PM sn$ s snf r �r r" r# r23 r$ r% r& s r �get_embeddings� EmbeddingModel.get_embeddings% sR � � !�[�[� 3� 3� :� :� �&@�&@� ;� 24�� 8J�7N�7N�O�7N��$�$�7N�O�O��Os �Ac � � U R R R XR S9nUR S R25 $ r, r1 r- s r �
get_embedding� EmbeddingModel.get_embedding, sA � ��K�K�*�*�1�1�� 8� 8� 2� 26� � �~�~�a� �*�*�*r )r r r27 r N)ztext-embedding-3-small)
�__name__�28__module__� __qualname__�__firstlineno__�strr r �floatr( r. r2 r5 �__static_attributes__� 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 r7 �embedding_model�print�runr. r( r> r r �<module>rF s} �� � &�
� � � �(+� (+�V �z��$�&�O� �'�+�+�o�9�9�/�J�29K�L� �� � � � 0� 0�/�CT�1U� V� 30�� r