<:i!vSrSSKJrJrJrJr SSKrSSKrSSKJ r \R"\ 5r "SS\ 5r g)uG OpenAI 提供商实现 支持 GPT-4o, GPT-4, GPT-3.5-turbo 等模型 )DictListAny GeneratorN)BaseAIProvidercj\rSrSrSr\S\4Sj5r\S\4Sj5rS\4Sjr S\4Sjr S\ \4Sjr S r SS \ \\\4S \ \S \S\S\\\44 SjjrSS \ \\\4S \ \S \S\S\\\\4S S 44 SjjrS\\\44SjrS \ \\\4S\S\S\S\\\44 SjrSrg )OpenAIProvideruOpenAI API 提供商returncg)Nopenaiselfs X/Users/yu22x/Desktop/ddd_副本5/ctf/app/services/ai_driver/providers/openai_provider.py provider_typeOpenAIProvider.provider_typecg)NOpenAIrrs r provider_nameOpenAIProvider.provider_namerrcg)Nzhttps://api.openai.com/v1rrs rget_default_base_url#OpenAIProvider.get_default_base_urls*rcg)Ngpt-4orrs rget_default_model OpenAIProvider.get_default_modelsrc /SQ$)N)rz gpt-4o-miniz gpt-4-turbozgpt-4z gpt-3.5-turborrs rget_available_models#OpenAIProvider.get_available_models"sQQrcSSKJn U"URURURS9$![ a [ S5ef=f)u获取 OpenAI 客户端r)r)api_keybase_urltimeoutu(请安装 openai 库: pip install openai)rrr&r'r( ImportError)rrs r _get_clientOpenAIProvider._get_client%sL J %     JHI I Js +.ANmessagestools temperature max_tokensc ^SSKnUR5nURURU5US.nU(aXHS'U(aUR U5US'SUS'Sn [ UR 5Hin URSS UR3U S:aS U S 3OS -5 URRR"S0UD6n URU 5s $ URSSUR S[U 535 U e![an U n [U 5R5m[U4S jS55n U (a^XR S- :aLU S-S-nURSS[!U 5R"SUS35 UR%U5 Sn A GM9URSS[U 535 eSn A ff=f)u执行聊天补全rN)modelr,r.r/r-auto tool_choiceinfou发送请求到 u (重试 )c3,># UH oT;v M g7f)Nr).0keyword error_strs r &OpenAIProvider.chat..Ws#HGi#7Hs) connectionr(rate502503504z bad gatewayzservice unavailablezgateway timeoutrwarninguAPI 调用失败 (u)，u秒后重试...erroruAPI 调用失败: uAPI 调用失败（已重试 u 次）: r)timer*r1_format_messages _format_toolsrange max_retries_logchat completionscreate_parse_response Exceptionstrloweranytype__name__sleep)rr,r-r.r/kwargsrEclientrequest_params last_errorattemptresponsee should_retry wait_timer:s @rrKOpenAIProvider.chat1s !!#ZZ--h7&  +5< ( &*&8&8&?N7 #,2N= ) T--.G  &$4TZZL"A_fij_jyQXPYYZE[pr"st!;;2299KNK++H55 /. ':4;K;K:LIVYZdVeUfgh%  FLLN "#H#   !1!1A!55%,q[A$5  )/A$q'BRBRASSWXaWbbq-rs 9-  '%7Ax#@A s ;A$D GBG0GGc +# UR5nURURU5USS.nU(aXGS'U(aURU5US'SUS'UR SSUR35 UR R R"S0UD6nS n /n UGHn U R(dM[U R5S :dM2U RS Rn U R(a XR- n S U RU S .v U R(GaU RGHn U R[U 5:a+U RU R=(d S S S S S.S.5 XRnU R(aU RUS'U R (dMU R R"(aU R R"US S'U R R$(dMUS S==U R R$- ss'GM U RS R&(dGMSU RS R&U U (aU OSS.v GM g![(a$nUR SS[+U535 eSnAff=f7f)u流式聊天补全T)r1r,r.streamr/r-r2r3r4u开始流式请求到 r6rcontent)rSrb accumulatedfunctionname argumentsidrSrdrirfrgfinishN)rS finish_reasonrb tool_callsrDu流式 API 调用失败: r)r*r1rFrGrJrKrLrMchoiceslendeltarbrlindexappendrirdrfrgrkrOrP)rr,r-r.r/rVrWrXracollected_contentcollected_tool_callschunkro tool_calltcr\s r chat_streamOpenAIProvider.chat_streamgsl!!#ZZ--h7&   +5< ( &*&8&8&?N7 #,2N= )6  IIf 6tzzlC D[[,,33EnEF " #% ===S%7!%;!MM!,22E}})]]:)$-',}}+<''').)9)9I(#6J2KK 4 ; ;*3,,*<",602571&="!#"6oo!FB(||+4<<4(111#,#5#5#:#:=F=O=O=T=TBzN6$:#,#5#5#?#?#?$&zN;$?9CUCUC_C_$_$?'*:,}}Q'555$,-2]]1-=-K-K'8BV*>\` M Z  IIg!:3q6(C D  sLAK AJJC-J AJ!AJ+.JK  K$KKK c0URSnURnUR=(d SURSSS.nUR(ao/US'URHZnUSR UR URURRURRS.S.5 M\ UR(aEURRURRURRS.US 'U$) u解析 OpenAI 响应rr6N)rbrkrlusagerlrerh) prompt_tokenscompletion_tokens total_tokensrz)rmmessagerbrkrlrqrirSrdrfrgrzr{r|r})rr[choicer~resultrvs rrNOpenAIProvider._parse_responses!!!$..,"#11     #%F< ((|$++%%GG " 0 0%'[[%:%:!-) >>!)!=!=%-^^%E%E ( ; ;F7O  r tool_call_id tool_name tool_resultc VUSUUUS./-nUR"U4SUR0UD6$)u继续对话，传入工具执行结果 Args: messages: 之前的消息列表 tool_call_id: 工具调用 ID tool_name: 工具名称 tool_result: 工具执行结果 Returns: AI 响应 tool)rolerrfrbr-)rK_tools)rr,rrrrVmessages_with_results rcontinue_with_tool_result(OpenAIProvider.continue_with_tool_resultsE( ((" , +  yy-KT[[KFKKrr)Ngffffff?N)rT __module__ __qualname____firstlineno____doc__propertyrPrrrr rr#r*rfloatintrrKrrwrNr__static_attributes__rrrr r s ss+c+3Rd3iR J!  4tCH~&4Dz4 4  4 c3h4r!  OtCH~&ODzO O  O 4S>4- .Ob!4S>!FLtCH~&LL L  L c3hLrr )rtypingrrrrloggingjsonbaser getLoggerrTloggerr rrrrs; .-   8 $eL^eLr