ProHelper / settings.py
DarForm's picture
Upload folder using huggingface_hub
097caae verified
import configparser
class SettingManager:
def __init__(self):
#Accessing default config
self.config=configparser.ConfigParser()
self.config.read("config.ini")
#getting default params
self.ai_assisted_search=self.config.getboolean('RAG','default_ai_assisted_search')
self.max_new_token=self.config.getint("Settings", "MAX_NEW_TOKENS")
self.top_p=self.config.getfloat("Settings", "TOP_P")
self.defaultLLM=self.config.getint('Settings', 'DEFAULT_LLM')
self.temperature=self.config.getfloat("Settings", "TEMPERATURE")
self.repetition_penalty=self.config.getfloat("Settings", "REPETITION_PENALITY")
self.system_prompt=self.config.get("Settings", "default_prompt")
self.n_doc_return=self.config.getint('RAG','default_returned_docs')
self.available_search_methods=self.config.get('RAG','methods').split(',')
self.search_method=self.config.get('RAG','default_search_method')
self.default_ai_search_prompt=self.config.get("RAG", "default_ai_search_prompt")
self.RAG_max_new_token=self.config.getint("RAG", "RAG_MAX_NEW_TOKENS")
self.RAG_top_p=self.config.getfloat("RAG", "RAG_TOP_P")
self.RAG_temperature=self.config.getfloat("RAG", "RAG_TEMPERATURE")
self.RAG_repetition_penalty=self.config.getfloat("RAG", "RAG_REPETITION_PENALITY")
#Loading available LLM
self.listLLM=self.get_llm()
self.listLLMMap=self.get_llm_map()
#Function used to get a list of available LLM
def get_llm(self):
llm_section = 'LLM'
if llm_section in self.config:
return [self.config.get(llm_section, llm) for llm in self.config[llm_section]]
else:
return []
#Function used to get a list of available LLM
def get_llm_prompts(self):
prompt_section = 'Prompt_map'
if prompt_section in self.config:
return [self.config.get(prompt_section, llm) for llm in self.config[prompt_section]]
else:
return []
#Function used to get the list of llm Map
def get_llm_map(self):
llm_map_section = 'LLM_Map'
if llm_map_section in self.config:
return [self.config.get(llm_map_section, llm) for llm in self.config[llm_map_section]]
else:
return []