
    RjUB                     ~    d Z ddlmZmZmZmZ ddlmZ ddlm	Z	m
Z
  G d de          ZddlmZ  ede           d	S )
u   OpenAI Responses API (Codex) transport.

Delegates to the existing adapter functions in agent/codex_responses_adapter.py.
This transport owns format conversion and normalization — NOT client lifecycle,
streaming, or the _run_codex_stream() call path.
    )AnyDictListOptional)ProviderTransport)NormalizedResponseToolCallc                   |   e Zd ZU dZdZee         ed<   edefd            Z	de
eef         defdZdee
eef                  defd	Zd
ee
eef                  defdZ	 ddedee
eef                  d
eee
eef                           de
eef         fdZdedefdZdedefdZdddededefdZdedefdZdS )ResponsesApiTransportzyTransport for api_mode='codex_responses'.

    Wraps the functions extracted into codex_responses_adapter.py (PR 1).
    N_last_issuer_kindreturnc                     dS )Ncodex_responses )selfs    ;/home/ubuntu/.hermes/hermes-agent/agent/transports/codex.pyapi_modezResponsesApiTransport.api_mode   s          paramsc           	         ddl m}  |t          |                    d                    t          |                    d                    t          |                    d                    |                    d                    S )z>Classify the current Responses endpoint from transport params.r   )_classify_responses_issueris_xai_responsesis_github_responsesis_codex_backendbase_url)r   r   r   r   )agent.codex_responses_adapterr   boolget)r   r   r   s      r   _resolve_issuer_kindz*ResponsesApiTransport._resolve_issuer_kind   s    LLLLLL))!&**-?"@"@AA $VZZ0E%F%F G G!&**-?"@"@AAZZ
++	
 
 
 	
r   messagesc           
          ddl m} |                     |          }|| _         ||t	          |                    d                    t	          |                    dd                    |          S )z:Convert OpenAI chat messages to Responses API input items.r   )!_chat_messages_to_responses_inputr   replay_encrypted_reasoningTr   r#   current_issuer_kind)r   r"   r   r   r   r   )r   r    kwargsr"   issuers        r   convert_messagesz&ResponsesApiTransport.convert_messages)   s    SSSSSS**622!'00!&**-?"@"@AA'+

7>>( ( !'
 
 
 	
r   toolsc                 $    ddl m}  ||          S )zBConvert OpenAI tool schemas to Responses API function definitions.r   )_responses_tools)r   r+   )r   r)   r+   s      r   convert_toolsz#ResponsesApiTransport.convert_tools7   s%    BBBBBB&&&r   modelc                 	   ddl m}m} ddlm} |                    dd          }|}	|sg|re|d                             d          dk    rFt          |d                             d          pd                                          }|d	d
         }	|s|}|                    dd          }
|                    dd          }|                    dd          }t          |                    dd                    }| 	                    |          }|| _
        d}d}|                    d          }|rLt          |t                    r7|                    d          du rd}n|                    d          r|d         }ddi}|                    ||          } ||          }|| ||	|||          dd}|r||d<   d|d<   d|d<   |                    d          }|
s	|s|r||d<   |r%|r#ddlm} |rd gng |d!<    ||          rd|i|d"<   n=|r2|
r|                    d#          }|||d"<   n|dd$|d"<   |rd gng |d!<   n	|
s|sg |d!<   |                    d%          }|r|                    |           |r|                    d&d
           |                    d'|                    d'                    }t          |t"          t$          f          rRt          |t                    s=dt%          |          cxk     rt%          d(          k     rn nt%          |          |d'<   n|                    d'd
           |r|                    d          }t          |p|pd                                          }|rl|                    d)          }i }t          |t                    r1|                    d* |                                D                        ||d<   ||d+<   ||d)<   |                    d,          }||s||d-<   |r|r|                    d)          }i }t          |t                    r1|                    d. |                                D                        ||d/<   ||d)<   |                    d0          }i } t          |t                    r|                     |           |                     d|           | |d0<   |S )1u  Build Responses API kwargs.

        Calls convert_messages and convert_tools internally.

        params:
            instructions: str — system prompt (extracted from messages[0] if not given)
            reasoning_config: dict | None — {effort, enabled}
            session_id: str | None — used for prompt_cache_key + xAI conv header
            max_tokens: int | None — max_output_tokens
            timeout: float | None — per-request timeout forwarded to the SDK
            request_overrides: dict | None — extra kwargs merged in
            provider: str | None — provider name for backend-specific logic
            base_url: str | None — endpoint URL
            base_url_hostname: str | None — hostname for backend detection
            is_github_responses: bool — Copilot/GitHub models backend
            is_codex_backend: bool — chatgpt.com/backend-api/codex
            is_xai_responses: bool — xAI/Grok backend
            github_reasoning_extra: dict | None — Copilot reasoning params
        r   )r"   r+   )DEFAULT_AGENT_IDENTITYinstructions rolesystemcontent   Nr   Fr   r   r#   Tmediumreasoning_configenabledeffortminimallowr$   )r-   r0   inputstorer)   autotool_choiceparallel_tool_calls
session_idprompt_cache_key)grok_supports_reasoning_effortzreasoning.encrypted_contentinclude	reasoninggithub_reasoning_extra)r9   summaryrequest_overridesservice_tiertimeoutinfextra_headersc                 V    i | ]&\  }}|r|	t          |          t          |          'S Nstr.0keyvalues      r   
<dictcomp>z6ResponsesApiTransport.build_kwargs.<locals>.<dictcomp>   sD        *U" (-'8  HHc%jj'8'8'8r   zx-client-request-id
max_tokensmax_output_tokensc                 V    i | ]&\  }}|r|	t          |          t          |          'S rN   rO   rQ   s      r   rU   z6ResponsesApiTransport.build_kwargs.<locals>.<dictcomp>   sD       &C $)#4 C#e**#4#4#4r   zx-grok-conv-id
extra_body)r   r"   r+   	run_agentr/   r   rP   stripr   r   r   
isinstancedictagent.model_metadatarC   updatepopintfloatitems
setdefault)!r   r-   r    r)   r   r"   r+   r/   r0   payload_messagesr   r   r   r#   issuer_kindreasoning_effortreasoning_enabledr7   _effort_clampresponse_toolsr&   rA   rC   github_reasoningrH   rJ   rB   cache_scope_idexisting_extra_headersmerged_extra_headersrV   existing_extra_bodymerged_extra_bodys!                                    r   build_kwargsz"ResponsesApiTransport.build_kwargs<   s]   4	
 	
 	
 	
 	
 	
 	
 	

 	544444zz."55# 	0 0HQKOOF33x??"8A;??9#=#=#CDDJJLL#+ABB<  	21L$jj)>FF!::&8%@@!::&8%@@%)JJ3T::&
 &
" //77!, $ !::&899 	>
+;T B B 	>##I..%77$)!!!%%h// >#3H#= "E*(,,-=?OPP))%00 (66 !1+E$/	   

 

  	1,F7O$*F=!,0F()ZZ--
 # 	4+; 	4
 	4)3F%& 	#!1 	#KKKKKK 4NU.//SU 9 .-e44 C'/1A&B{# 	#" #)::.F#G#G #/*:F;'1Af&U&U{#7QY233WY y!! % 	#-= 	# "F9"JJ':;; 	-MM+,,,  	-JJ~t,,, **Y

9(=(=>>we--	(w--	( E'NN1111U5\\11111 %gF9JJy$''' 	?%zz*<== !1!EZ!E2FFLLNNN ?)/O)D)D&79$4d;; (// .D.J.J.L.L     6D$\2>L$%:;*>'ZZ--
!*:!*4F&' 	5
 	5%+ZZ%@%@"35 0$77 $++ *@*F*F*H*H     6@ !12&:F?# #)**\":":02-t44 >!(()<===(();ZHHH#4F< r   responsec                    ddl m} |                    d          p| j        } |||          \  }}d}|r	|j        rg }|j        D ]}i }	t          |d          r|j        r
|j        |	d<   t          |d          r|j        r
|j        |	d<   |                    t          t          |d          r|j
        nt          |d	          r|j        j        ndt          |d	          r|j        j        nt          |d
d          t          |d	          r|j        j        nt          |dd          |	pd                     i }	|r!t          |d          r|j        r
|j        |	d<   |r!t          |d          r|j        r
|j        |	d<   |r!t          |d          r|j        r
|j        |	d<   t%          |r|j        nd||pd|rt          |d          r|j        ndd|	pd          S )z=Normalize Codex Responses API response to NormalizedResponse.r   )_normalize_codex_responserf   )rf   Ncall_idresponse_item_ididfunctionnamer1   	argumentsz{})rw   ry   rz   provider_datacodex_reasoning_itemscodex_message_itemsreasoning_detailsstoprE   )r4   
tool_callsfinish_reasonrE   usager{   )r   rt   r   r   r   hasattrru   rv   appendr	   rw   rx   ry   getattrrz   r|   r}   r~   r   r4   rE   )
r   rr   r&   rt   rf   msgr   r   tcr{   s
             r   normalize_responsez(ResponsesApiTransport.normalize_response  s   	
 	
 	
 	
 	
 	
 jj//I43I66x[YYY]
 	3> 	Jn   "2y)) :bj :/1zM),2122 Lr7J L8:8KM"45!!( 'D 1 1nruuGTVXbLcLc8m8H8Him-4R-D-Da))'RTV\^`JaJa7>r:7N7Nrbk33T[\^`kmqTrTr"/"74	# # #      	O73 788 	OS=V 	O585NM12 	K73 566 	K3;R 	K363JM/0 	G73 344 	G9N 	G141FM-.!#&0CKKD!'16'*RwsK/H/HRcmmd'/4
 
 
 	
r   c                 b    |dS t          |dd          }t          |t                    r|sdS dS )u  Check Codex Responses API response has valid output structure.

        Returns True only if response.output is a non-empty list.
        Does NOT check output_text fallback — the caller handles that
        with diagnostic logging for stream backfill recovery.
        NFoutputT)r   r\   list)r   rr   r   s      r   validate_responsez'ResponsesApiTransport.validate_response=  sC     58T22&$'' 	v 	5tr   Fallow_stream
api_kwargsr   c                (    ddl m}  |||          S )zValidate and sanitize Codex API kwargs before the call.

        Normalizes input items, strips unsupported fields, validates structure.
        r   )_preflight_codex_api_kwargsr   )r   r   )r   r   r   r   s       r   preflight_kwargsz&ResponsesApiTransport.preflight_kwargsK  s-    
 	NMMMMM**:LQQQQr   
raw_reasonc                 <    ddddd}|                     |d          S )aK  Map Codex response.status to OpenAI finish_reason.

        Codex uses response.status ('completed', 'incomplete') +
        response.incomplete_details.reason for granular mapping.
        This method handles the simple status string; the caller
        should check incomplete_details separately for 'max_output_tokens'.
        r   length)	completed
incompletefailed	cancelled)r   )r   r   _MAPs      r   map_finish_reasonz'ResponsesApiTransport.map_finish_reasonS  s2      "	
 
 xx
F+++r   rN   )__name__
__module____qualname____doc__r   r   rP   __annotations__propertyr   r   r   r   r   r(   r,   rq   r   r   r   r   r]   r   r   r   r   r   r   r      s          (,x}+++!# ! ! ! X!
4S> 
c 
 
 
 

d38n)= 
C 
 
 
 
'4S#X#7 'C ' ' ' ' 15	O OO tCH~&O T#s(^,-	O 
c3hO O O Ob.
3 .
=O .
 .
 .
 .
`# $     IN R R R3 R RRV R R R R,C ,C , , , , , ,r   r   )register_transportr   N)r   typingr   r   r   r   agent.transports.baser   agent.transports.typesr   r	   r   agent.transportsr   r   r   r   <module>r      s     - , , , , , , , , , , , 3 3 3 3 3 3 ? ? ? ? ? ? ? ?S, S, S, S, S,- S, S, S,n
 0 / / / / /  $&; < < < < <r   