File: //home/arjun/projects/calendar/calendar-planning/app/utils/__pycache__/openai_utils.cpython-312.pyc
�
�hDg
� �� � d dl mZ d dlmZ d dlmZmZ d dlZd dlm Z e � eej d �� Z G d� d e� Z G d
� d e� Z
d d ed
edefd�Zddddddd�ddd�d�dgd�d�d �gZd d �Zy) � )�OpenAI)�settings)� BaseModel�FieldN)� load_dotenv�OPENAI_API_KEY)�api_keyc �h � e Zd ZU edd�� Zeed< edd�� Zeed< edd�� Zeed< y )
� EventDetail.zName of the event�� description�namezEStarting date of the event in DD-MM-YYY format if present else use NA�datez<Location of the event city or country if present else use NA�locationN) �__name__�
__module__� __qualname__r r �str�__annotations__r r � � �I/home/arjun/projects/calendar/calendar-planning/app/utils/openai_utils.pyr r s8 � ��s�(;�<�D�3�<��s�(o�p�D�3�p��3�,j�k�H�s�kr r c �< � e Zd ZU edd�� Zeed< ee ed< y)�Event.z Event category given by the userr �category� event_detailN) r r r r r r r �listr r r r r r s � ��3�,N�O�H�s�O��{�#�#r r � user_prompt�model�returnc � � d}t j j j j |d|d�d| d�gt
d�� }|j d j }|j r |j S t d� i S ) Na. You will be provided with a Category, Year, Location, and text scraped from a website. Your task is to extract and generate unique event details that are relevant to the given Category, Year, and Location. Ensure that all event details are distinct, with no duplicate event names or repetitive entries. Each event should be explicitly tied to the inputs provided, and any event lacking sufficient information in the text should be excluded. Focus on accuracy and avoid assumptions, ensuring the generated content is clear, complete, and free from redundancy.�system)�role�content�useri @ )r �messages�response_format�
max_tokensr z No result)
�client�beta�chat� completions�parser �choices�message�parsed�print)r r �
system_prompt�
completion�results r �generate_eventsr5 s� � � H �M����!�!�-�-�3�3�� �-� 8�� � 4�
�
�� 4� �J� �
�
��
"�
*�
*�F�
�}�}� �}�}� �
�k��� r �function�
search_webz"Search the web using google search�object�stringzThe query to be searched.)�typer
zThe result of web search.)� search_query�formatr; )r: �
properties�required)r r
�
parameters)r: r6 c �� � t j j j || ||�� }|S # t $ r%}t d� t d|� �� |cY d }~S d }~ww xY w)N)r r& �tools� tool_choicez*Unable to generate ChatCompletion responsez Exception: )r) r+ r, �create� Exceptionr1 )r& rA rB r �response�es r �chat_completion_requestrG K sf � � ��;�;�*�*�1�1�� ��#� 2�
�� ��� � �
�:�;�
� �A�3� � �����s �.1 � A �A�A �A )� gpt-4o-mini)NNrH )�openair � django.confr �pydanticr r �os�dotenvr �environr) r r r �dictr5 rA rG r r r �<module>rP s� �� � � %� � � �
�
��
�
�#3�4� 5��l�)� l� $�I� $�
�� �3� �4� �< � �?� � !)�'B�%�
!)�'B� � � ,� ,�
�
�� ��0 r