File size: 950 Bytes
a08284c 05c86f0 a08284c 05c86f0 a08284c 05c86f0 a08284c 05c86f0 a08284c 05c86f0 a08284c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
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(', ')
|