afulara/PythonicRAG-FastAPI-React
0
1�
2 ʱ�g� � �* � S SK r " S S5 r " S S\5 r " S S\5 r " S S \5 r " S3 S \5 r\S :X aS \" S
5 r\" \R SSS95 \" S
5 r\" \R SSS95 \" \R 5 5 gg)� Nc �& � \ rS rSrS rS rS rSrg)�4BasePrompt� c �F � Xl [ R " S5 U l g)z�5Initializes the BasePrompt object with a prompt template.6 7:param prompt: A string that can contain placeholders within curly braces8z \{([^}]+)\}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__�BasePrompt.__init__ s � � � ��9�10�>�2��
� c �� � U R R U R 5 nU R R " S0 U Vs0 s H o3UR US5 _M snD6$ s snf )z�11Formats 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� � )r16 �findallr �format�get)r �kwargs�matches�matchs r �
format_prompt�BasePrompt.format_prompt sT � � �-�-�'�'�� � �4���{�{�!�!�X�w�$W�w�e�F�J�J�u�b�,A�%A�w�$W�X�X��$Ws � A# c �L � U R R U R 5 $ )zf17Gets the list of input variable names from the prompt string.18 19:return: List of input variable names20)r21 r r )r s r �get_input_variables� BasePrompt.get_input_variables s � � �}�}�$�$�T�[�[�1�1r )r22 r N)�__name__�23__module__� __qualname__�__firstlineno__r
r r �__static_attributes__r r r r r s � �3�Y�2r r c �: ^ � \ rS rSrS\4U 4S jjrSS jrSrU =r$ )�24RolePrompt�! �rolec �0 >� [ TU ] U5 X l g)z�25Initializes the RolePrompt object with a prompt template and a role.26 27:param prompt: A string that can contain placeholders within curly braces28:param role: The role for the message ('system', 'user', or 'assistant')29N)�superr
r& )r r r& � __class__s �r r
�RolePrompt.__init__" s �� � ���� �� r c � � U( a U R U R " S0 UD6S.$ U R U R S.$ )z�30Creates 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 )r r r s r �create_message�RolePrompt.create_message, s7 � � � �I�I�$�2D�2D�2N�v�2N�O� O� � � �d�k�k�:�:r )r& )T) r r r r! �strr
r- r" �
__classcell__�r) s @r r$ r$ ! s �� ��S� �35;� 36;r r$ c �0 ^ � \ rS rSrS\4U 4S jjrSrU =r$ )�SystemRolePrompt�9 r c �& >� [ TU ] US5 g )N�system�r( r
�r r r) s �r r
�SystemRolePrompt.__init__: s �� �
�����*r r �r r r r! r/ r
r" r0 r1 s @r r3 r3 9 s �� �+�s� +� +r r3 c �0 ^ � \ rS rSrS\4U 4S jjrSrU =r$ )�UserRolePrompt�> r c �& >� [ TU ] US5 g )N�userr7 r8 s �r r
�UserRolePrompt.__init__? s �� �
�����(r r r: r1 s @r r< r< > s �� �)�s� )� )r r< c �0 ^ � \ rS rSrS\4U 4S jjrSrU =r$ )�AssistantRolePrompt�C r c �& >� [ TU ] US5 g )N� assistantr7 r8 s �r r
� AssistantRolePrompt.__init__D s �� �
�����-r r r: r1 s @r rB rB C s �� �.�s� .� .r rB �__main__z%Hello {name}, you are {age} years old�John� )�name�age) r r r$ r3 r< rB r r �printr r- r r r r �<module>rM s� �� �2� 2�:;�� ;�0+�z� +�37)�Z� )�38.�*� .�39 �z��
�?�
@�F� �&�40 �41 �F��42 �433�4�
� E�
F�F� �&�44 �45 �V��46 �474�5� �&�48$�49$�50&�'�
r