yuviertdh/promptshieldenv
0
1+
2 �2�i� � �| � R t ^ RIHtHt ^ RIt^ RIHt ^ RIHtH t ]! RR7 t3]! 4 t ]4P R]R7 R 4 t
]5P R64 R R l4 t]7P R
4 R 4 t]8P R4 R 4 t]9P R4 R 4 t]10P R4 R 4 tR# )z*11FastAPI application for PromptShieldEnv.12)�FastAPI�
HTTPExceptionN)�PromptShieldEnvironment)� Observation�ActionzPromptShieldEnv API)�titlez/reset)�response_modelc � �6 " � \ P 4 p V # 5i)z7Starts a new episode and returns the first observation.)�env�reset)�obss �,E:\New folder\AI 2.0\promptshieldenv\main.py� reset_envr
s � � � �)�)�+�C� �J�s �z/stepc �$ � V ^8� d QhR\ /# )� �action)r )�formats "r
� __annotate__r s � � � �6� � c � �j " � \ P V P 4 4 w rr4RVRVRVRV/# 5i)zNTakes an action and returns the next observation, reward, done flag, and info.� observation�reward�done�info)r13 �step�14model_dump)r r r r r s & r
�step_envr sA � � � "�h�h�v�'8�'8�':� ;��C���s��&����� � �s �13z/statec � �2 " � \ P 4 # 5i)z"Returns current environment state.)r15 �state� r r
� get_stater s � � � �9�9�;��s �z/tasksc � �Z " � RRRRR/RRRR/RRRR /.R16\ P ! 4 /# 5i) z(Returns list of tasks and action schema.�tasks�name�direct_injection_detection�17difficulty�easy�indirect_and_hidden_attacks�medium�jailbreak_and_adversarial�hard�
action_schema)r �model_json_schemar r r
� get_tasksr- # sP � � � �
�1�<�� H�
�2�L�(� K�
�0�,�� G�18�19 ��1�1�3�
� �s �)+z/graderc � � " � \ P 4 p V R, ' g \ RRR7 h\ P 4 pV# 5i)z-Returns grader scores after the episode ends.�is_donei� z Episode not finished completely.�� status_code�detail)r20 r r �evaluate_grader)r �scoress r
�21get_graderr5 / s> � � �
�I�I�K�E� �� � ���4V�W�W�
�
�
�
"�F� �M�s �AA z /baselinec � �� " � ^ RI p R p V P V4 G Rj x�L22 pRVP /# L \ P d p\ RRTP 2R7 hRp?ii ; i5i)zTRuns the baseline agent and returns the scores by shelling out in a separate thread.Nc �: � \ P ! R R.RRRR7 # )z2C:\Users\Admin\AppData\Local\Python\bin\python.exez inference.pyT)�capture_output�text�check)�23subprocess�runr r r
�
run_inference�,run_baseline_endpoint.<locals>.run_inference= s$ � ��~�~�
B�N� S� ��� 24� 25r �outputi� zBaseline failed: r0 )�asyncio� to_thread�stdoutr; �CalledProcessErrorr �stderr)r@ r= �result�es r
�run_baseline_endpointrG 8 sl � � � �26�T� �(�(� �7�7���&�-�-�(�(� 8�� � (� (� T���6G����z�4R�S�S��T�s1 �A+�4 �2�4 �A+�4 �A(� A#�#A(�(A+)�__doc__�fastapir r r; � environmentr �modelsr r �appr27 �postr r �getr r- r5 rG r r r
�<module>rO s� ��� +� � /� &�
�)�*�� � �����(�;��/�� 0��28 ���'��� �� ������ �� ����� � � � ���)��� �� ���+��T� �Tr