| from configparser import ConfigParser | |
| class LoadConfig: | |
| def __init__(self, config_file='src/ui/config.ini'): | |
| self.config = ConfigParser() | |
| self.config.read(config_file) | |
| def get_llm_options(self): | |
| """ | |
| Get the list of LLM options from the configuration file. | |
| """ | |
| return self.config.get('DEFAULT', 'LLM_options').split(', ') | |
| def get_use_case(self): | |
| """ | |
| Get the use case from config file. | |
| """ | |
| return self.config.get('DEFAULT', 'USE_CASE').split(', ') | |
| def get_title(self): | |
| """ | |
| Get the title from config file. | |
| """ | |
| return self.config.get('DEFAULT', 'Title') | |
| def get_groq_models(self): | |
| """ | |
| Get groq models from the config file. | |
| """ | |
| return self.config.get('DEFAULT', 'GROQ_MODEL').split(', ') | |
| def get_openai_models(self): | |
| """ | |
| Get OpenAI models from config file. | |
| """ | |
| return self.config.get('DEFAULT', 'OPENAI_MODEL').split(', ') | |