Rahmath1/self_improving_agent
0
1�
2 �h�i�3 � � � d Z ddlmZmZmZmZmZ ddlmZ ddl m4Z m Z
G d� de � Z G d� de
� ZeZ G d� d e� Zy5) z�6Data models for the Advance Agent Environment.7 8The Advance_agent environment is a simple test environment that echoes back messages.9� )�List�Dict�Any�Optional�Union)� BaseModel)�Action� Observationc � � e Zd ZU dZeed< y)�AdvanceAgentObservation�10clean_data� action_typeN)�__name__�11__module__� __qualname__r �str�__annotations__� � �^C:\Users\rahma\OneDrive\Documents\GitHub\Streamlit\Project2\Self_Agent\Advance_agent\models.pyr r ' s � �#�K��#r r c � � e Zd ZU g Zee ed< g Zee ed< i Z eee12f ed< g Z ee ed< dZ eed< dZ
eed< dZeed< y )13r � dataset_head�columns�stats�history� �task� objective�14difficultyN)r r r r r r r r r r r r r r r r r r r r / s^ � � "$�L�$�t�*�#�!#�G�$�s�)�#�#%�E�$�s�C�x�.�%�!#�G�$�s�)�#�!#�D�#�#�!#�I�#�#�!#�J�#�#r c �0 � e Zd ZU eed< eed< dZeed< y)�Reward�score�feedbackF�15is_penaltyN)r r r �floatr r r$ �boolr r r r! r! C s � ����O� �J�� r r! N)�__doc__�typingr r r r r �pydanticr � openenv.corer �16BaseActionr17 �BaseObservationr r! r r r �<module>r- sC ���, 4� 3� � M�$�j� $� $�o� $� &� � �Y� r