�
��jV � �Z � S r SSKJr SSKJr SSKJr SSKJr SSK J
r
" S S\5 r g )
z�Pier custom agent: apply a host-produced patch, commit, then exit.
Used after host-side adapters (tk/codex/pi/omp/fx). The DeepSWE collect hook
is `git diff <base> HEAD`, so the applied patch must be committed.
� )� annotations)�Path)� BaseAgent)�BaseEnvironment)� AgentContextc �z ^ � \ rS rSrSrSS.S
U 4S jjjr\S S j5 rS S jrS S jr S
S jr
Sr U =r $ )�ApplyHostPatchAgent� FN)�
patch_filec �2 >� Xl [ TU ] " U0 UD6 g �N)r �super�__init__)�selfr �args�kwargs� __class__s ��L/Users/undivisible/projects/rotary/scripts/bench/deepswe/apply_host_patch.pyr � ApplyHostPatchAgent.__init__ s �� �$��
���$�)�&�)� c � � g)Nzapply-host-patch� r r r �name�ApplyHostPatchAgent.name s � �!r c � � g)Nz1.0.0r )r s r �version�ApplyHostPatchAgent.version s � �r c � � # � g 7fr
r )r � environments r �setup�ApplyHostPatchAgent.setup s � � ��s �c � �Z # � U R ( d g [ U R 5 nUR 5 ( a UR 5 R S:X a g SnUR [
U5 U5 I S h v�N UR SU SU SU S35 I S h v�N g N( N7f)Nr z/tmp/host.patchzPcd /app && git config --global --add safe.directory /app && (git apply --binary z || git apply -p1 --binary z || patch -p1 < zr) && git add -A && git -c user.email=bench@rotary -c user.name=rotary-bench commit -m 'host adapter patch' || true)r r �exists�stat�st_size� upload_file�str�exec)r � instructionr �context�src�dests r �run�ApplyHostPatchAgent.run s� � � � ��� ��4�?�?�#���z�z�|�|�s�x�x�z� 1� 1�Q� 6� � ���%�%�c�#�h��5�5�5�� � �
#�#'�&�(C�D�6�IY�Z^�Y_� `5�
5�
�
�
� 6�
�s$ �A<B+�>B'�?"B+�!B)�"B+�)B+)r z
str | None)�returnr' )r r r/ �None)r) r' r r r* r r/ r0 )
�__name__�
__module__� __qualname__�__firstlineno__�
SUPPORTS_ATIFr � staticmethodr r r r- �__static_attributes__�
__classcell__)r s @r r r sc �� ��M�7;� *� *� �"� �"���
��
� %�
� �
�
�
�
r r N) �__doc__�
__future__r �pathlibr �pier.agents.baser �pier.environments.baser �pier.models.agent.contextr r r r r �<module>r? s' ���
#� � &� 2� 2�$
�)� $
r