j32SrSSKrSSKJr SSKJrJrJrJrJ r J r SSK r SSK J r SSKJr SSKJr Sq\ "S \ S 9rS rS \S \4Sjr"SS\\5r"SS5r"SS5rg)zN Agent Factory for creating PydanticAI agents with different model providers. N)Enum)AnyDictListOptionalTypeTypeVar) BaseModel)Agent) get_settingsFOutputT)bound)zsonnet-5zopus-4-7zopus-4-8zopus-5fablemythosmodel_idreturnc`^UR5m[U4Sj[55(+$)z:Whether the model accepts an explicit temperature setting.c3,># UH oT;v M g7fN).0markerlowereds 2/home/mande/repo/quber/src/quber/agents/factory.py 0supports_sampling_temperature..#sM3LW$3Ls)loweranyNO_SAMPLING_MODEL_MARKERS)rrs @rsupports_sampling_temperaturer s$nnGM3LMM MMcH\rSrSrSrSrSrSrSrSr Sr S r S r S r S rS rSrSrg) ModelProvider&zSupported model providers. anthropicopenaigroqgeminivertexaibedrockollamaxaimistraldeepseekcoherecerebrasrN)__name__ __module__ __qualname____firstlineno____doc__ ANTHROPICOPENAIGROQGEMINIVERTEXAIBEDROCKOLLAMAXAIMISTRALDEEPSEEKCOHERECEREBRAS__static_attributes__rr!rr#r#&sA$I F D FHG F CGH FHr!r#c\rSrSrSr\R S\RS\RS\RS\RS\RS\RS\RS \RS \RS \R S \R"S 0 r\R S\RS\RS\RS\RS\RS\RS\RS\RS\RS\R S\R"S0 r\R S\RS\RS\RS\RS\RS\RS \RS!\RS"\R S#\R"S$0 r\S%\\\\44S&j5rS'rg)( ModelConfig7zModel configuration mappings.zclaude-haiku-4-5z gpt-4.1-minizllama-3.3-70b-versatilezgemini-2.5-flashz(anthropic.claude-haiku-4-5-20251001-v1:0zllama3.2z grok-4-fastzmistral-small-latestz deepseek-chatzcommand-r-plus-08-2024z gpt-oss-120bANTHROPIC_API_KEYOPENAI_API_KEY GROQ_API_KEYGEMINI_API_KEYGOOGLE_APPLICATION_CREDENTIALSAWS_ACCESS_KEY_IDN XAI_API_KEYMISTRAL_API_KEYDEEPSEEK_API_KEY CO_API_KEYCEREBRAS_API_KEYr%r&r'googlez google-cloudr*r,r-r.r/r0rc^SSKJn U"5n0nURR5H6upEUS3m[ U4SjU55nU(dM(XcUR 'M8 U$)aModel names per provider, taken from pydantic-ai's model registry. pydantic-ai maintains the authoritative set of recognized model ids and refreshes it on each release, so this tracks model availability with the pinned pydantic-ai version rather than a hand-kept file. The registry returns "provider:model" ids; we group them under our own provider values via ``REGISTRY_PREFIXES`` and strip the prefix. Returns: Dictionary mapping provider value to a sorted list of model names. r)known_model_names:c3n># UH*oRT5(dMU[T5Sv M, g7fr) startswithlen)rnameheads rr/ModelConfig.get_known_models..s)Y%$??SWCX-DT-%s55)pydantic_ai.modelsrSREGISTRY_PREFIXESitemssortedvalue)clsrSnames by_providerproviderprefixmodelsrYs @rget_known_modelsModelConfig.get_known_modelsrse 9!#,. # 5 5 ; ; = HXQ r!r)r1r2r3r4r5r#r6r7r8r9r:r;r<r=r>r?r@rADEFAULT_MODELS API_KEY_VARSr\ classmethodrstrrrfrBrr!rrDrD7s' !3n50 2Ij=56 N" !4.N. @2d=0 2l 2 L* hFhy5y h  c49n!5r!rDc\rSrSrSrSS\4SjjrSS\\S\ S\ \ S \ \ S \ S \ \ \44 S jjrS \ \ 4S jrS\ S \ S \ 4SjrSS\ \ S \\ \\ 44SjjrS \\ 4SjrSrg) AgentFactoryz Factory for creating PydanticAI agents with different model providers. This factory provides a unified interface for creating agents that can work with various LLM providers, automatically handling API key configuration and model selection. enable_logfirecJXl[5RRnU(aqU(aj[(dW[ R "SUSS9 [ R"5 [ R"5 [ R"5 SqSUl gSUl g)z^ Initialize the Agent Factory. Args: enable_logfire: Whether to enable logfire telemetry quberF) service_nametokenconsoleTN) ror obs logfire_token_logfire_configuredlogfire configureinstrument_anthropicinstrument_openaiinstrument_pydantic_ailogfire_configured)selfrorvs r__init__AgentFactory.__init__s{-$**88 m&&!!!('! ,,.))+..0&*#&*D #&+D #r!N output_type system_promptrcmodel agent_kwargsrc DU(aKSU;aEURSS5upg[H'nURUR5:XdM#UnUn O UceUR 5nUcR[ SSR [RR5V s/sH o(dM U PM sn 5-5e[RRU5n U (a)[R"U 5(d[ SU 35eUc[RUnURX45n UR(a9[ R""SURUS9 [%U 4XS .UD6n SSS5 U $[%U 4XS .UD6n U $s sn f!,(df  W $=f) a Create a PydanticAI agent with the specified configuration. Args: output_type: The Pydantic model for structured output system_prompt: System prompt for the agent provider: Model provider to use (auto-detected if not specified) model: Specific model to use (uses default if not specified) **agent_kwargs: Additional kwargs to pass to Agent constructor Returns: Configured PydanticAI Agent instance Raises: ValueError: If no valid API key is found or invalid configuration rTNz%No API key found. Please set one of: z, zAPI key not found: create_agent)rcr)rr)splitr#r_rdetect_provider ValueErrorjoinrDrivaluesgetosgetenvrhformat_model_stringr}rxspanr ) r~rrrcrr provider_str model_namepv api_key_var model_stragents rrAgentFactory.create_agents2 SE\',{{3': $L"77l0022 H&E #  ++-H ;iiK,D,D,K,K,M S,MqQR,M STU"..228< ryy552;-@A A =..x8E,,X=   " "nx~~USin[namnT  )jj]ijE 7!T,TS  s F (F "F Fcl[R[R[R[R/nU[Vs/sH o"U;dM UPM sn-nUHJn[ R RU5nU(dM+[R"U5(dMHUs $ gs snf)z Auto-detect available provider based on environment variables. Returns: First available provider or None if no API keys found N) r#r6r7r8r9rDrirrr)r~priorityrorderedrckey_vars rrAgentFactory.detect_providers  # #        LA8:KaLLH!..228 list of models zNo models found)rDrfr_rrhr#)r~rc known_models provider_namereresultrs rlist_available_models"AgentFactory.list_available_models"s#335 $NNM!%%mR8F%4488CTUV!* *AGGM!%%mR8F%4488/n[Hn[RRU5nUc3U[R:XaUR UR 5 MVMX[R"U5(dMuUR UR 5 M U$)z` Get list of providers with configured API keys. Returns: List of available provider names ) r#rDrirr<appendr_rr)r~ availablercrs rget_available_providers$AgentFactory.get_available_providersCs} %H%2266x@K"}333$$X^^44;''  0&r!)ror})T)NNr)r1r2r3r4r5boolrrr rkrr#rr rrrrrrlistrrBrr!rrmrms,t,:-1# C']CC=) C } C  C sG| CJ-!82+M+#+#+.h}.EQUVY[_`c[dVdQeBcr!rm)r5renumrtypingrrrrrr rxpydanticr pydantic_air quber.settingsr rwr rrkrr r#rDrmrr!rrs{ ;;' )9 - NCNDN C"QQhIIr!