openenv/coding_env
21
1�
2 ���hb � � � S r SSKJr SSKJrJr SSKJr SSKJ r SSK3J r J r J
r
\( a SSKJr " S S4\\ \ 4 5 rg ) a� 5CodingEnv6---------7Client-side wrapper for the Coding environment server.8Talks HTTP to a single base_url exposing: /reset and /step.9 10- users instantiate CodingEnv with a base_url provided by the higher-level11 vector/orchestration layer.12- Environment authors ship the Docker image that serves the HTTP API.13 14(Seeds, episode IDs, request IDs, capabilities can be added later in the payloads.)15� )� annotations)�
TYPE_CHECKING�Optional)�
HTTPEnvClient)�16StepResult� )�17CodeAction�CodeObservation� CodeState)�ContainerProviderc �2 � \ rS rSrSS jrSS jrS S jrSrg)18� CodingEnv� c � � SUR 0$ )N�code)r )�self�actions �J/Users/zwentz/Development/OpenEnv/src/envs/coding_env/coding_env_client.py�
_step_payload�CodingEnv._step_payload s � �
�F�K�K�19� 20� c 21 � � [ S0 US D6n[ UUR S5 [ UR SS5 5 S9$ )N� observation�reward�doneF)r r r � )r22 r �get�bool)r �payload�obss r �
_parse_result�CodingEnv._parse_result$ sC � � �7��
� 6�7�����;�;�x�(��g�k�k�&�%�0�1�23� 24r c �v � [ UR S5 UR SS5 UR SS5 S9$ )z�25Parse server response into CodeState object.26 27Args:28 payload: JSON response from /state endpoint29 30Returns:31 CodeState object with episode_id, step_count, and last_exit_code32�33episode_id�34step_countr �last_exit_code)r$ r% r& )r r )r r s r � _parse_state�CodingEnv._parse_state- s: � � � �{�{�<�0� �{�{�<��3�"�;�;�'7��;�35� 36r r N)r r �return�dict)r r* r) zStepResult[CodeObservation])r r* r) r )�__name__�37__module__� __qualname__�__firstlineno__r r! r' �__static_attributes__r r r r r s � �38� 39�40r r N)�__doc__�41__future__r �typingr r �core.http_env_clientr �42core.typesr �modelsr r43 r �core.containers.runtimer r r r r �<module>r7 s: �� � #� *� .� !� :� :��9� 44�
�j�/� 9�:� 45r