o pj@sdZddlmZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z ddlmZmZdd lmZmZed Zeejd Zed d dZd9ddZd:ddZddd;ddZdd(d)Zd?d.d/Z d@d1d2Z!dAd4d5Z"dBd7d8Z#dS)Cz1Centralized OpenAI client for the platform layer.) annotationsN)Path)Any)OpenAI)convert_from_bytes)get_platform_loggerlog_structured)consumerecord_observed_usageaiplatformsecrets openai_keys gpt-4o-minicCs,|pi}t|pidtr|pidpidnd}|r"t|nd}|r7t||d|d|p/iddd t||d }t|d }t|||d } z |jj j di| } WnLt y} z@t t d |||t|tro|pkidndt| jt| ddt| ddt| ddd |r|rt||d| d|piddd d} ~ wwd} | jr| jdjr| jdjjpd} t| |d} t| }|rt||d|dd|d|t|tr|pidnd|dd|dddd t t d||||dt|tr|pidndd| || t| ddt| dddS) zBExecute an OpenAI request and return normalized output plus usage.tenant tenant_idNr ai_tokensprovider_reservation request_id)kindr)rapp_idmetricamountmetadatar)api_key)promptmodeloptionsai_provider_request_failed status_codecodeparam)rrrrexception_type http_status provider_codeprovider_paramprovider_reservation_releasecontentr total_tokens#provider_reservation_reconciliation input_tokens output_tokens)rrrr.r/ai_usage)rrrtokensr _request_idid)outputusage raw_outputprovider_request_idprovider_response_id) isinstancegetdict_estimated_capacityr _resolve_api_keyr_build_request_kwargschat completionscreate Exceptionrloggertype__name__getattrr choicesmessager+_normalize_output_extract_usage)contextrrrrr reservationrclientrequest_kwargsresponseexcr+r4r5r9r9#bridge_platform/ai/openai_client.pyrun_llms0             rSrdict[str, Any]returnintcCs:|drtdt|dpdStdt|dpdS)zFConservative reservation; actual usage is recorded after the response. file_bytescapacity_reservation_tokensiPi)r;maxrV)rr9r9rRr=fs r=rrstrr str | Nonec Cstddt|pdd}tddt|pdd}tddt|p(dd}|rA|rAttd||dnd}tt|d}ttd }t d |}t d pet d } |po|po|po|po| } | syt d || S)Nz [^A-Z0-9]+_r)z [^a-z0-9_]+z [^a-z0-9_-]+z_-tenantsz.keyzdev.keyOPENAI_API_KEY_OPENAI_API_KEY_DEVOPENAI_API_KEYz"Missing OpenAI API key for app_id=) resubr[upperstriplower_read_key_file _KEYS_DIRosgetenv RuntimeError) rr normalizedsafe_app safe_tenant tenant_keyapp_keydev_key env_app_key env_dev_keyrr9r9rRr>ms   & r>pathrcCsDz |jdd}WntyYdStyYdSw|p!dS)Nutf-8)encoding) read_textreFileNotFoundErrorOSError)rtvaluer9r9rRrg~s  rgrrcCs|dt||ddgd}|d}|dkrddi|d<|S|dkrB|d}t|ts0td dt|d p8d d |d d|d<|S)Nuser)rr)roler+)rmessagesresponse_format json_objectrE json_schemaz4json_schema response format requires a schema object schema_namestructured_outputT)namestrictschema)rEr)_build_contentr;r:r< ValueErrorr[)rrrrOr~rr9r9rRr?s*       r?list[dict[str, Any]]cCspd|dg}|d}|r6t||dpdd\}}ddl}||d}|d d d |d |id |S)Ntext)rErrWmimetypezapplication/octet-streamrWrrru image_urlurlzdata:z;base64,)rEr)r;_prepare_visual_inputbase64 b64encodedecodeappend)rrr+rWrrb64r9r9rRrs    rrWbytesrtuple[bytes, str]cCs|dkr t|dfS||fS)Nzapplication/pdfz image/png)_pdf_first_page_to_pngrr9r9rRrs r pdf_bytescCsJt|ddddd}|stdddlm}|}|dj|dd |S) NpngrX)dpifmt first_page last_pagez(PDF conversion failed: no pages renderedr)BytesIOPNG)format)rrkiorsavegetvalue)rpagesrbufr9r9rRrs rr+cCs:|ddvrzt|WStyd|iYSw|S)Nr~>rr_raw)r;jsonloadsrCr*r9r9rRrJs   rJdict[str, int]cCs\t|dd}|durddddStt|ddpdtt|ddp!dtt|ddp*ddS)Nr5r)r.r/r, prompt_tokenscompletion_tokensr,)rGrV)rPr5r9r9rRrKs  rK)rN)rrTrUrV)rr[rr\rUr[)rtrrUr\)rr[rr[rrTrUrT)rr[rrTrUr)rWrrr[rUr)rrrUr)r+r[rrT)rUr)$__doc__ __future__rrrirbpathlibrtypingropenair pdf2imager'bridge_platform.logging.platform_loggerrrbridge_platform.quotas.servicer r rD__file__resolveparents _PROJECT_ROOTrhrSr=r>rgr?rrrrJrKr9r9r9rRs0       O