yjWUdZddlmZddlmZddlmZddlmZddl m Z ddl m Z ddlmZdd lmZmZmZdd lmZd d lmZd d lmZd dlmZd dlmZd dlmZd dl m!Z!m"Z"m#Z#m$Z$d dl m%Z%ed d;dZ&ed ddZ)edd?d Z*edd@d!Z+iZ,d"e-d#<ej.j/r%eej.j/ej.j0d$%e,d&<ej.j1r$eej.j1ej.j2'e,d(<ej.j3rHej.j4re j5ej.j4)nd*Z6e j7ej.j3e6+e,d,<dAd2Z8dBd5Z9dCd7Z:dDd9Z;gd:ZDs w : : ::r'c&t||dS)zHAnthropic client for a specific (base_url, api_key) pair. Cached by key.rrrr=s r(get_anthropic_override_clientr@Ls 'He L L LLr'c`|rtj|nd}tj||S)zEGemini client for a specific (base_url, api_key) pair. Cached by key.r1Nr2)r5r6rr7)r!r r3s r(get_gemini_override_clientrBUs8 BJS;*H====tL <l C C CCr'z$dict[ModelTransport, ProviderClient]CLIENTSrr anthropicr+openair1Nr2geminiproviderr model_configr rcd|j%|jt|}||S|jpt |}|j}|st d|d|dkrt ||S|dkrt||S|dkrt||St|dS)uResolve the provider client for a ModelConfig. Fast path: no overrides → reuse the module-level default client from CLIENTS (the test-mockable seam). Otherwise route through the cached override factories. NzMissing API key for z model configrDrErF) r r!rCgetrrr@r>rBr)rGrHexisting_clientr r!s r(client_for_model_configrL{s# (=(E!++h//  &" ""I&?&I&IG$H R!"P"P"P"PQQQ;,Xw???8)(G<<<8)(G<<<r'clientrc|dkrt|S|dkrt|S|dkrt|St|dS)zGWrap a raw provider SDK client in the matching ProviderBackend adapter.rDrErFN)rrrr)rGrMs r(backend_for_providerrOsc ;'''8V$$$8V$$$r'rcn|dkrtS|dkrtStS)zJProvider-appropriate HistoryAdapter for assistant/tool message formatting.rDrF)rrr)rGs r(history_adapter_for_providerrQs<;&(((8#%%%  ! !!r'configcVt|j|}t|j|S)uHigh-level one-shot backend factory: ModelConfig → ProviderBackend. Delegates client resolution to ``client_for_model_config``, which owns the CLIENTS fast-path and the missing-API-key validation. Both the production path (via ``honcho_llm_call_inner``) and the live-test path (via this function) now construct clients through the same helper, so validation behavior stays consistent. )rL transportrO)rRrMs r( get_backendrUs)%V%5v > >F  0& 9 99r') rCrOrLr)r@rUr9rBr.r>rQ)rr)rr )rr/)r!r;r r;rr )r!r;r r;rr)r!r;r r;rr/)rGr rHr rr)rGr rMrrr)rGr rr)rRr rr)=__doc__ __future__r functoolsrtypingrrDrgoogler google.genair r5rEr src.configr r rsrc.exceptionsrbackendrbackends.anthropicrbackends.geminirbackends.openair credentialsrhistory_adaptersrrrrrr)r.r9r>r@rBrC__annotations__r#r$r%r,r-r8r4r6r3r7rLrOrQrU__all__r&r'r(rfs#"""""$$$$$$------<<<<<<<<<<......$$$$$$000000************222222 "!!!!! 1 1 1XXXX 3;;;; 3MMMM 3DDDD132222 <!)> .0GK  <#  +-GH  <  < '  )EFFFF  %  +!GH :    """" : : : :   r'