o Tj@s"ddlmZddlZddlmZmZmZmZddlm Z m Z ddl Z ddl m Z ddlmZdd lmZdd lmZmZdd lmZdd lmZmZd dlmZddlmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'ddl(m)Z)e dddZ*d3ddZ+d4d d!Z,d5d'd(Z-d6d,d-Z.d7d1d2Z/dS)8) annotationsN) TYPE_CHECKINGListIterablecast)TypeVar assert_never)ResponsesPydanticFunctionTool)Omit)is_given) PYDANTIC_V1model_parse_json)construct_type_unchecked)is_basemodel_typeis_dataclass_like_type)type_to_response_format_param) Response ToolParam ParsedContentParsedResponseFunctionToolParamParsedResponseOutputItemParsedResponseOutputTextResponseFunctionToolCallParsedResponseOutputMessageResponseFormatTextConfigParamParsedResponseFunctionToolCall)ResponseFormat TextFormatT)defaulttype_typereturnrcCs\t|}t|s Jtt|}|ddksJd|dvsJdd|dd|dddS)Nr$ json_schemaschemaTname)r$strictr(r')rr rr )r#response_format_dictr+a/home/dinkstrade/pdmp-scanner/venv/lib/python3.10/site-packages/openai/lib/_parsing/_responses.pytype_to_text_format_param&s    r- text_formattype[TextFormatT] | Omit input_tools!Iterable[ToolParam] | Omit | Noneresponse!Response | ParsedResponse[object]ParsedResponse[TextFormatT]c Cs6g}|jpgD]}|jdkrTg}|jpgD](}|jdkr#||q|ttdti|dt|j ||j didq|ttdt i|d|idq|jd kro|tt i|d t ||d idq|jd ks|jd ks|jdks|jdks|jdks|jdks|jdks|jdks|jdks|jdks|jdks|jdks|jdks|jdks|jdks|jdks|jdks|jdks|jdks|jdks|jd ks|jd!ks|jd"ks|jd#ks|jd$ks|jd%ks|jd&kr||qtrt|q||qttd'ti|d(|idS))Nmessage output_textz+type[ParsedResponseOutputText[TextFormatT]]parsed)r.phase)r#valuez.type[ParsedResponseOutputMessage[TextFormatT]]content function_callparsed_arguments)r0r; computer_callfile_search_callweb_search_calltool_search_calltool_search_outputadditional_tools reasoningprogramprogram_output compactionmcp_callmcp_approval_requestmcp_approval_responseimage_generation_callcode_interpreter_calllocal_shell_calllocal_shell_call_output shell_callshell_call_outputapply_patch_callapply_patch_call_outputmcp_list_toolsexeccustom_tool_callfunction_call_outputcomputer_call_outputcustom_tool_call_outputz!type[ParsedResponse[TextFormatT]]output)rXr$r:appendrrrto_dict parse_texttextr8rrparse_function_tool_argumentsrrr)r.r0r2 output_listrX content_listitemr+r+r,parse_response5s                                  rar\strr8 str | NoneTextFormatT | NonecCsp|dur |dkr dSt|sdSt|rttt||St|r1tr)td|t | |Std|)a=Parse final-answer text, retaining legacy behavior for messages without a phase. Explicit phases other than `final_answer` are not structured results. Leave their text unparsed, including phases introduced by future API versions. Both completed responses and streamed text-done events use this rule. N final_answerz:Non BaseModel types are only supported with Pydantic v2 - z3Unable to automatically parse response format type ) r rrr!rrr TypeErrorpydantic TypeAdapter validate_json)r\r.r8r+r+r,r[sr[Iterable[ToolParam]r(FunctionToolParam | NonecCs0|D]}|ddkr|d|kr|SqdS)Nr$functionr()get)r0r(toolr+r+r,get_input_tool_by_names ror;9ParsedResponseFunctionToolCall | ResponseFunctionToolCallobjectcCsf|dust|s dSt||jd}|sdStt|}t|tr&t|j|j S| ds-dSt |j S)N)r0r(r)) r ror(rrq isinstancer rmodel argumentsrmjsonloads)r0r; input_toolrnr+r+r,r]s    r])r#r$r%r)r.r/r0r1r2r3r%r4)r\rbr.r/r8rcr%rd)r0rjr(rbr%rk)r0r1r;rpr%rq)0 __future__rrutypingrrrrtyping_extensionsrrrg_toolsr _typesr _utilsr _compatrr_modelsr _pydanticrr _completionsrtypes.responsesrrrrrrrrrrr#types.chat.completion_create_paramsr r!r-rar[ror]r+r+r+r,s.      4    ]