burtenshaw/openenv-benchmark-ws
0
1�
2 �v9i� � � � S r SSKJr SSKJr SSKJr SSKJr SSK J3r4 SSK J r J
r
" S S5\\ \
4 5 r " S S \6\ \
4 5 rg
)zd7Benchmark Environment Clients.8 9Provides HTTP and WebSocket clients for the Benchmark Environment.10� )�Dict)�11StepResult)�State)�
HTTPEnvClient)�WebSocketEnvClient� )�BenchmarkAction�BenchmarkObservationc �T � \ rS rSrSrS\S\4S jrS\S\\ 4S jr12S\S\ 4S jr S r
g13) � BenchmarkEnv� z*HTTP client for the Benchmark Environment.�action�returnc � � SUR 0$ �N� wait_seconds�r ��selfr s �+/Users/ben/code/OpenEnv/benchmark/client.py�
_step_payload�BenchmarkEnv._step_payload � � � �� 3� 3�4�4� �payloadc
� � UR S0 5 n[ UR SS5 UR SS5 UR SS5 UR SS5 UR S S5 UR S14S 5 UR S 5 UR S
0 5 S9n[ UUR S 5 UR S15S 5 S9$ �N� observation�host_url� �pidr � session_hash�waited_secondsg � timestamp�doneF�reward�metadata)r r! r" r# r$ r% r& r' )r r& r% ��getr16 r �r r �obs_datar s r �
_parse_result�BenchmarkEnv._parse_result � � ��;�;�}�b�1��*� �\�\�*�b�1�� � �U�A�&�!� � �n�b�9�#�<�<�(8�#�>� �l�l�;��4����V�U�+��;�;�x�(� �\�\�*�b�1� 17� � �#��;�;�x�(����V�U�+�18� 19r c �T � [ UR S5 UR SS5 S9$ �N�20episode_id�21step_countr )r1 r2 �r r) �r r s r � _parse_state�BenchmarkEnv._parse_state/ �) � �� �{�{�<�0� �{�{�<��3�22� 23r � N��__name__�24__module__� __qualname__�__firstlineno__�__doc__r r r r r25 r, r r5 �__static_attributes__r8 r r r r sD � �4�5�O� 5�� 5�26�T� 27�j�9M�.N� 28�$29�D� 30�U� 31r r c �T � \ rS rSrSrS\S\4S jrS\S\\ 4S jr32S\S\ 4S jr S r
g33) �BenchmarkEnvWS�6 z/WebSocket client for the Benchmark Environment.r r c � � SUR 0$ r r r s r r � BenchmarkEnvWS._step_payload9 r r r c
� � UR S0 5 n[ UR SS5 UR SS5 UR SS5 UR SS5 UR S S5 UR S34S 5 UR S 5 UR S
0 5 S9n[ UUR S 5 UR S35S 5 S9$ r r( r* s r r, � BenchmarkEnvWS._parse_result< r. r c �T � [ UR S5 UR SS5 S9$ r0 r3 r4 s r r5 �BenchmarkEnvWS._parse_stateN r7 r r8 Nr9 r8 r r rA rA 6 sD � �9�5�O� 5�� 5�36�T� 37�j�9M�.N� 38�$39�D� 40�U� 41r rA N)r> �typingr �openenv.core.client_typesr � openenv.core.env_server.typesr � openenv.core.http_env_clientr �openenv.core.ws_env_clientr �modelsr r42 r rA r8 r r �<module>rO sM ��� � 0� /� 6� 9� 9� 43�=� �2F�!F�G� 44�> 45�'��9M�(M�N� 46r