afulara/PythonicRAG-FastAPI-React
0
1�
2 ʱ�g� � �4 � d dl Z G d� d� Z G d� de� Z G d� de� Z G d� d e� Z G d3� d e� Zed k( rY ed
� Z eej dd�� � ed
� Z eej dd�� � eej � � yy)� Nc � � e Zd Zd� Zd� Zd� Zy)�4BasePromptc �F � || _ t j d� | _ y)z�5 Initializes the BasePrompt object with a prompt template.6 7 :param prompt: A string that can contain placeholders within curly braces8 z \{([^}]+)\}N)�prompt�re�compile�_pattern)�selfr s ��/Users/aqsa/AI Projects/AIE6 Class Projects/AIE6-DeployPythonicRAG-Fastapi-react-app/backend/aimakerspace/openai_utils/prompts.py�__init__zBasePrompt.__init__ s � � �� ��9�10�>�2��
� c �� � | j j | j � } | j j di |D �ci c] }||j |d� �� c}��S c c}w )z�11 Formats the prompt string using the keyword arguments provided.12 13 :param kwargs: The values to substitute into the prompt string14 :return: The formatted prompt string15 � � )r �findallr �format�get)r16 �kwargs�matches�matchs r �
format_promptzBasePrompt.format_prompt sT � � �-�-�'�'�� � �4��!�t�{�{�!�!�X�w�$W�e�U�F�J�J�u�b�,A�%A�$W�X�X��$Ws �A! c �L � | j j | j � S )z~17 Gets the list of input variable names from the prompt string.18 19 :return: List of input variable names20 )r r r )r21 s r �get_input_variablesz BasePrompt.get_input_variables s � � �}�}�$�$�T�[�[�1�1r
N)�__name__�22__module__� __qualname__r r r r r
r r r s � �3�Y�2r
r c �, � � e Zd Zdef� fd� Zdd�Z� xZS )�23RolePrompt�rolec �2 �� t �| � |� || _ y)z�24 Initializes the RolePrompt object with a prompt template and a role.25 26 :param prompt: A string that can contain placeholders within curly braces27 :param role: The role for the message ('system', 'user', or 'assistant')28 N)�superr r )r29 r r � __class__s �r r zRolePrompt.__init__" s �� � ���� ��� r
c �v � |r | j | j di |��d�S | j | j d�S )z�30 Creates a message dictionary with a role and a formatted message.31 32 :param kwargs: The values to substitute into the prompt string33 :return: Dictionary containing the role and the formatted message34 )r �contentr )r r r )r35 r r s r �create_messagezRolePrompt.create_message, s: � � � �I�I�2D�$�2D�2D�2N�v�2N�O� O� � � �d�k�k�:�:r
)T)r r r �strr r% �
__classcell__�r" s @r r r ! s �� ��S� �36;r
r c �$ � � e Zd Zdef� fd� Z� xZS )�SystemRolePromptr c �& �� t �| � |d� y )N�system�r! r �r37 r r" s �r r zSystemRolePrompt.__init__: s �� �
�����*r
�r r r r& r r' r( s @r r* r* 9 s �� �+�s� +� +r
r* c �$ � � e Zd Zdef� fd� Z� xZS )�UserRolePromptr c �& �� t �| � |d� y )N�userr- r. s �r r zUserRolePrompt.__init__? s �� �
�����(r
r/ r( s @r r1 r1 > s �� �)�s� )� )r
r1 c �$ � � e Zd Zdef� fd� Z� xZS )�AssistantRolePromptr c �& �� t �| � |d� y )N� assistantr- r. s �r r z AssistantRolePrompt.__init__D s �� �
�����-r
r/ r( s @r r5 r5 C s �� �.�s� .� .r
r5 �__main__z%Hello {name}, you are {age} years old�John� )�name�age) r r r r* r1 r5 r r �printr r% r r r
r �<module>r> s� �� �2� 2�:;�� ;�0+�z� +�38)�Z� )�39.�*� .�40 �z��
�?�
@�F� �&�41 �42 �F��43 �443�4�
� E�
F�F� �&�45 �46 �V��47 �484�5� �&�49$�50$�51&�'�
r