afulara/PythonicRAG-FastAPI-React
0
1�
2 ʱ�g � �J � d dl mZmZ d dlmZ d dlZ e� G d� d� Zy)� )�OpenAI� AsyncOpenAI)� load_dotenvNc �. � e Zd Zddefd�Zddefd�Zd� Zy) �3ChatOpenAI�4model_namec �t � || _ t j d� | _ | j � t d� �y )N�OPENAI_API_KEYzOPENAI_API_KEY is not set)r �os�getenv�openai_api_key�5ValueError)�selfr s ��/Users/aqsa/AI Projects/AIE6 Class Projects/AIE6-DeployPythonicRAG-Fastapi-react-app/backend/aimakerspace/openai_utils/chatmodel.py�__init__zChatOpenAI.__init__ s8 � �$��� �i�i�(8� 9��� � � � &� � 8�9� 9� '� � text_onlyc � � t |t � s t d� �t � } |j j6 j d| j |d�|��}|r#|j d j j S |S )N�messages must be a list)�model�messagesr � ) �7isinstance�listr r �chat� completions�creater �choices�message�content)r r r �kwargs�client�responses r �runzChatOpenAI.run sx � ��(�D�)� � 6�7� 7����1�6�;�;�*�*�1�1� 8��/�/�H�9�8>�10�� ��#�#�A�&�.�.�6�6� 6��r c �F K � t |t � s t d� �t � } |j j11 j d| j |dd�|��� d { ��� }|2 3 d { ��� }|j d j j }|��/|��� �57 �;7 �26 y �w)Nr T)r r �streamr r ) r r r r r r r r r �deltar )r r r! r" r& �chunkr s r �astreamzChatOpenAI.astream s� � �� ��(�D�)� � 6�7� 7� � ��5�v�{�{�.�.�5�5� 12��/�/� ��13� � 14� 15�� "� � �%��m�m�A�&�,�,�4�4�G��"� �
�16�� ��6�s< �A B!� B� B!�&B �*B �+B �.&B!�B!� B � B!N)z gpt-4o-mini)T)�__name__�17__module__� __qualname__�strr �boolr$ r) r r r r r s � �:�3� :� �t� � r r )�openair r �dotenvr r r r r r �<module>r1 s �� &� � � �
�% � % r