ڎjK2dZddlmZddlmZddlmZmZmZmZm Z m Z ddl m Z m Z ddlmZdd lmZmZdd lmZdd lmZdd lmZdd lmZmZddlmZmZmZm Z ddl!m"Z"e rddl#m$Z$ddl#m%Z%ddl&m'Z'neZ$eZ%eZ'e(ddhZ)GddeZ*dS)z3The OpenAI Responses API chat model implementation.) OrderedDict)datetime)LiteralAnyAsyncGeneratorList TYPE_CHECKINGType) BaseModelField) _generate_id) ChatModelBase_TOOL_CHOICE_LITERAL_MODES) ChatResponse) ChatUsage)OpenAICredential) FormatterBaseOpenAIResponseFormatter)Msg ThinkingBlock ToolCallBlock TextBlock) ToolChoice)Response)ResponseStreamEvent) AsyncStream modalitiesaudioceZdZUdZGddeZdZeded< d&d e d e d dde de de de dedzdee efdzddffd ZedeeedffdZ d'de deedeedzdedzdedeeedfzf dZded d!deedffd"Zded d#defd$Zdeedzdedzdeeedze ezdzffd%Z xZ!S)(OpenAIResponseModelaHThe OpenAI Responses API chat model. Compared with the Chat Completions API, the Responses API provides first-class streaming events for reasoning / thinking, text output and function-call arguments, which makes it a natural fit for models that expose chain-of-thought reasoning (e.g. ``o3``, ``o4-mini``). ceZdZUdZeddddZedzed<edd d Ze ed <edd d Z e ddzed<edddddZ e dzed<dS)OpenAIResponseModel.Parametersz1The parameters for the OpenAI Response API model.Nz Max Tokensz$The maximum number of output tokens.r)defaulttitle descriptiongt max_tokensFThinkingzWhether to enable reasoning for reasoning models (e.g. o3, o4-mini, gpt-5.5). Use reasoning_effort to control the depth of reasoning.)r%r&r'thinking_enablezReasoning EffortzControls the depth of reasoning for reasoning models (e.g. o3, o4-mini, gpt-5.5). Supported values are model-dependent and may include: none, minimal, low, medium, high, xhigh.)noneminimallowmediumhighxhighreasoning_effort Temperaturez#The temperature for the LLM output.r)r%r&r'gele temperature)__name__ __module__ __qualname____doc__r r)int__annotations__r+boolr2rr6floatVC:\Projects\DSCons\.venv\Lib\site-packages\agentscope\model\_openai_response\_model.py Parametersr$(s??!&> " " " C$J   !&2 ! ! !     E$'   G H4 O %*E= % % % UT\     r@rBopenai_responsetypeNTr ?@  credentialmodel parametersz%OpenAIResponseModel.Parameters | Nonestream max_retries retry_delay context_size formatter client_kwargsreturnc Zt|||p||||||p t|_| pi|_ddl} | jd|jj |jj |jj d|j|_ dS)aInitialize the OpenAI Responses API chat model. Args: credential (`OpenAICredential`): The OpenAI credential used to authenticate API calls. model (`str`): The OpenAI model name, e.g. ``o3`` or ``o4-mini``. parameters (`OpenAIResponseModel.Parameters | None`, defaults to `None`): The OpenAI Responses API parameters. When ``None``, the default parameters will be used. stream (`bool`, defaults to `True`): Whether to enable streaming output. max_retries (`int`, defaults to `3`): The maximum number of retries for the OpenAI Responses API. retry_delay (`float`, defaults to `1.0`): Seconds to sleep between retry attempts. context_size (`int`, defaults to `200000`): The model context size used for context compression. formatter (`FormatterBase | None`, defaults to `None`): The formatter that converts ``Msg`` objects to the format required by the OpenAI Responses API. When ``None``, an ``OpenAIResponseFormatter`` instance will be used. client_kwargs (`dict[str, Any] | None`, defaults to `None`): Extra keyword arguments forwarded to ``openai.AsyncClient`` (e.g. ``timeout``, ``default_headers``, ``http_client``). )rGrHrIrJrKrLrMrN)api_key organizationbase_urlr?)super__init__rBrrNrOopenai AsyncClientrGrRget_secret_valuerSrTclient) selfrGrHrIrJrKrLrMrNrOrW __class__s rArVzOpenAIResponseModel.__init__TsN !!6T__%6%6##%    #?&=&?&?*0b *<&*<+ O+<<>>5_-+ +  + +  r@.c>ddl}|j|j|j|jfS)Nr)rWAPIConnectionErrorAPITimeoutErrorRateLimitErrorInternalServerError)clsrWs rA_get_retryable_exceptionsz-OpenAIResponseModel._get_retryable_exceptionss/   %  "  !  &   r@ model_namemessagestools tool_choicegenerate_kwargscK|j|d{V}|||jd}|jj|jj|d<|jj|jj|d<|jjr|jjrd|jji|d<|d| D| ||\}} |||d<| | |d <tj } |j jjd i|d{V} |jr|| | S|| | S) a#Call the OpenAI Responses API. Args: model_name (`str`): The model name to use for this call. messages (`list`): A list of input items for the Responses API. tools (`list[dict]`, default `None`): The tools JSON schemas. tool_choice (`ToolChoice | None`, optional): Controls which (if any) tool is called by the model. **generate_kwargs (`Any`): Extra keyword arguments forwarded to the API. Returns: `ChatResponse | AsyncGenerator[ChatResponse, None]`: A ``ChatResponse`` when streaming is disabled, or an async generator of ``ChatResponse`` objects when streaming is enabled. N)rHinputrJmax_output_tokensr6effort reasoningc,i|]\}}|tv||Sr?)_RESPONSES_UNSUPPORTED_KWARGS).0kvs rA z1OpenAIResponseModel._call_api..s4   Aq9991999r@rfrgr?)rNformatrJrIr)r6r+r2updateitems _format_toolsrnowrZ responsescreate_parse_stream_response_parse_completion_response) r[rdrerfrgrhformatted_messages api_kwargs fmt_toolsfmt_tool_choicestart_datetimeresponses rA _call_apizOpenAIResponseModel._call_apis8$(>#8#8#B#BBBBBBB 'k& & ? % 1.2o.HJ* + ? & 2(,(CJ} % O + 0  $/:'J{ #   +1133      &*%7%7{%K%K" ?  "+Jw   &(7J} %!5.5CC CCCCCCCC ; I..~xHH H..~xHHHr@rrz AsyncStream[ResponseStreamEvent]c 2Kd}t}t}t}t}|23d{V}|j} tgd|} | dkr| |j|n|| dkr| |j|nX| dkr@|j} t| ddd kr"| j t| d d pd f|| j <n| d kr6|j } | |vr*|| \} }| | ||jpd n| dkr|j }|jrn|j}t|dd}t|j|jt%j|z |rt|ddnd}t|dgD]B}t|dddkr+t|dd}|r| d ||C| js|r || _| WV6dS)aParse the OpenAI Responses API streaming response. Each event yields only the *delta* content produced by that event so that the base ``ChatModelBase.__call__`` can accumulate the full response. Because the Responses API only returns the upstream response id on the final ``response.completed`` event, we assign a locally-generated ``response_id`` up-front so every delta chunk carries a stable id. Args: start_datetime (`datetime`): The start datetime of the response generation. response (`AsyncStream[ResponseStreamEvent]`): The OpenAI Responses API async stream object. Yields: `ChatResponse`: Incremental ``ChatResponse`` objects with ``is_last=False``. The base class ``__call__`` accumulates them and emits the final ``is_last=True`` chunk. NF)contentis_lastidz%response.reasoning_summary_text.delta)block_idzresponse.output_text.deltazresponse.output_item.addedrD function_callnameunknownz&response.function_call_arguments.delta)rrrjzresponse.completedinput_tokens_details cached_tokensr input_tokens output_tokenstimecache_input_tokensoutputrmr)thinkingrreasoning_item_id)rrrDrappend_thinkingdelta append_textitemgetattrcall_idritem_idappend_tool_callrusagerrrrrx total_secondsr)r[rrr response_idtext_id thinking_idtool_call_mappingevent event_type delta_resrrrrrespudetails output_itemrs rAr{z*OpenAIResponseModel._parse_stream_responses4#''>> #~~'>> #.--#R R R R R R R %J$I DDD ))K(* ;;;%%ekG%DDDD;;;z4../AA fb11>Y2%dg. GGG-///$5g$>MGT..!(!#k/R/ 333~: A%a)?FFG%%&^&'o&lnn~=LLNN # ,7#+,,,    E&$+42#>#>K{FD99[HH,3Kt,L,L),%55)+)42C6   E "' e$88sHrc g}|jD]"}t|dd}|dkrit|dd}ddt|dgD}|s|r%|t d|| |d krSt|d gD]@}t|ddd kr)|t d |jA|dkr@|t|j|j t|ddpd$d} |j rn|j } t| dd} t| j | j tj|z | rt| ddnd} |d| d} t|dd} | r| | d<t#di| S)aParse the OpenAI Responses API non-streaming response. Args: start_datetime (`datetime`): The start datetime of the response generation. response (`Response`): The OpenAI Responses API response object. Returns: `ChatResponse`: A single ``ChatResponse`` with ``is_last=True``. rDNrmr c3ZK|]&}t|ddt|ddV'dS)textrN)r)rpss rA zAOpenAIResponseModel._parse_completion_response..wsT,,q&"--,Avr**,,,,,,r@summaryr)rDrrmessager output_textr)rDrr argumentsrz{})rrrjrrrrT)rrrr?)rrjoinappendrrrrrrrrrrrrxrr)r[rrcontent_blocksr item_typercombined_summarypartrrr resp_kwargsrs rAr|z.OpenAIResponseModel._parse_completion_response_sl"KMO# # Dfd33IK''$+D$$=$=!#&88,,$T9b99,,,$$ $'8"))%!+%5.?i''#D)R88DtVT22mCC&--%6 BBB o--%%!<!Y%dK<<D > Aa!7>>G^olnn~5DDFF  $7#$$$    E&' ' hd33  , +K **k***r@c|r|r|||d}|r d|D}|s|dfS|j}|tvr|d|dfS|jr|d|d|jDdfS||fS)aValidate and format tools and tool_choice for the Responses API. The full ``tools`` list is always sent unchanged to maximise prompt cache hits. When ``tool_choice.tools`` restricts the callable subset, the ``allowed_tools`` tool_choice format is used instead of filtering the schemas list. Args: tools (`list[dict] | None`, optional): The raw tool schemas. tool_choice (`ToolChoice | None`, optional): The tool choice configuration. Returns: `tuple[list[dict] | None, str | dict | None]`: A tuple of ``(formatted_tools, formatted_tool_choice)`` ready for the Responses API. Nc&g|]}ddi|dS)rDfunctionr?)rptools rA z5OpenAIResponseModel._format_tools..s4=A8tJ'78r@rrDr allowed_toolscg|]}d|dS)rrr?)rprs rArz5OpenAIResponseModel._format_tools..s0(66r@)rDmoderf)_validate_tool_choicerrrf)r[rfrgrrs rArwz!OpenAIResponseModel._format_toolss.  ;5 ;  & &{E : : :  EJI #d? " 1 1 1z4@@@ @   ' + 1 $r@)NTr rErFNN)NN)"r7r8r9r:r rBrDrr<rstrr=r;r>rdictrrV classmethodtupler Exceptionrclistrrrrrrr{r|rw __classcell__)r\s@rAr"r"s' ' ' ' ' Y' ' ' R(9D'# $888% ?C "*./3: : $: : < :  :  : : : !4': CH~,:  : : : : : : x %Y0D*E   [ $()- KIKIKIs)KIDzD KI  $& KI  KI |T'9: :KIKIKIKIZu  u 5u   d* + u u u u nR+ R+R+  R+R+R+R+h2DzD 2 $&2 tDzD #*t"33 4 22222222r@r"N)+r: collectionsrrtypingrrrrr r pydanticr r _utils._commonr_baserr_model_responser _model_usagerrGrrNrrrrrrrrropenai.types.responsesrrrWr frozensetror"r?r@rArs99######JJJJJJJJJJJJJJJJ%%%%%%%%******========******$$$$$$******????????CCCCCCCCCCCC//////::::::"""""""HK!* <*A B BFFFFF-FFFFFr@