jUdZddlmZddlZddlZddlmZejeZ ddddd Z d e d <dd Z dd Z y)z author: Yagnik Poshiya organization: Webbrains Technologies Private Limited Builds the correct LLMClient for a given provider name. API key env var convention: {PROVIDER_NAME_UPPERCASE}_LLM_API_KEY ) annotationsN) LLMClientgoogleopenai anthropic openrouter)rrrrzdict[str, str]_PROVIDER_ALIASESc|jd}tjj|xsdj }|st d|d|d|S)N _LLM_API_KEYz%API key not configured for provider 'z'. Set z in the RAG agent .env file.)upperosenvirongetstrip ValueError) provider_nameenv_keyvalues C/var/www/html/agent-engine/apps/agents/rag-agent/src/llm/factory.py_resolve_api_keyrsi$$&'|4G ZZ^^G $ * 1 1 3E 3M?C)7 9   Lc |jj}tj|}|;t d|ddj t tjt|}|dk(rddl m }|||S|dk(rdd l m }|||S|d k(rdd l m}|||S|d k(rdd lm}|||St d|d)NzUnsupported LLM provider: 'z'. Supported: z, rr)GoogleLLMClient)api_keymodel_identifierr)OpenAILLMClientr)AnthropicLLMClientr)OpenRouterLLMClient')rlowerr rrjoinsortedkeysrllm.google_llmrllm.openai_llmrllm.anthropic_llmrllm.openrouter_llmr) rrkeyresolvedrrrrrs rbuild_llm_clientr+%s     % % 'C $$S)H)-9))F+<+A+A+C$DEF H  x(G82wAQRR82wAQRR;8!'DTUU<:"7EUVV 2=/C DDr)rstrreturnr,)rr,rr,r-r)__doc__ __future__rloggingrllm.baser getLogger__name__loggerr __annotations__rr+rrr7so#    8 $ %>EEEEr