fh[ ZdZddlmZddlmZmZmZddlmZddl m Z m Z Gdde Z y) z/ Configuration management for the application. )Path)AnyDictOptional load_dotenv) BaseModelFieldceZdZUdZeddZeeed<eddZ eeed<eddZ eeed <edd Z eeed <edd Z eeed <dZ eed<dZeeed<dZeed<dZeed<edZeed<edZeed<dZeed<dZeed<GddZed$deeddfd Zd!edeefd"Zdeeeffd#Zy)%ConfigzApplication configuration.NANTHROPIC_API_KEY)defaultenvanthropic_api_keyOPENAI_API_KEYopenai_api_key GROQ_API_KEY groq_api_keyGEMINI_API_KEYgemini_api_keyPYDANTIC_LOGFIRE_TOKEN logfire_token anthropicdefault_provider default_modelmax_concurrent_tablestable_analysis_timeoutoutput output_dir experimentsexperiments_dirINFO log_levelTenable_logfireceZdZdZdZy) Config.Configz.envzutf-8N)__name__ __module__ __qualname__env_fileenv_file_encoding0/home/mande/repo/quber/src/quber/utils/config.pyr r($s #r/r,returncL|rt||St|S)z Load configuration from environment and .env file. Args: env_file: Optional path to .env file Returns: Config instance r)clsr,s r0loadz Config.load(s(   ! u  Mu r/providerc|j|j|j|jd}|j |j S)z Get API key for a specific provider. Args: provider: Provider name Returns: API key or None )ropenaigroqgemini)rrrrgetlower)selfr5key_maps r0 get_api_keyzConfig.get_api_key<sG//))%%))   {{8>>+,,r/c,|jdhdS)zConvert config to dictionary.T>rrrrr) exclude_noneexclude) model_dump)r<s r0to_dictzConfig.to_dictNs   r/)N)r)r*r+__doc__r rrstr__annotations__rrrrrrrintrrr!r#r%r&boolr classmethodr4r>rrrCr.r/r0r r s6$(-T?R'Sx}S$)$rNs)&&%M YM r/