31 lines
977 B
Python
31 lines
977 B
Python
import yaml
|
|
from utils.path_tool import get_abs_path
|
|
|
|
|
|
def load_rag_config(config_path=get_abs_path("config/rag.yaml"), encoding="utf-8"):
|
|
with open(config_path, 'r', encoding=encoding) as f:
|
|
return yaml.safe_load(f)
|
|
|
|
|
|
def load_chroma_config(config_path=get_abs_path("config/chroma.yaml"), encoding="utf-8"):
|
|
with open(config_path, 'r', encoding=encoding) as f:
|
|
return yaml.safe_load(f)
|
|
|
|
|
|
def load_prompts_config(config_path=get_abs_path("config/prompts.yaml"), encoding="utf-8"):
|
|
with open(config_path, 'r', encoding=encoding) as f:
|
|
return yaml.safe_load(f)
|
|
|
|
|
|
def load_agent_config(config_path=get_abs_path("config/agent.yaml"), encoding="utf-8"):
|
|
with open(config_path, 'r', encoding=encoding) as f:
|
|
return yaml.safe_load(f)
|
|
|
|
rag_conf = load_rag_config()
|
|
chroma_conf = load_chroma_config()
|
|
prompts_conf = load_prompts_config()
|
|
agent_conf = load_agent_config()
|
|
|
|
if __name__ == '__main__':
|
|
print(rag_conf["chat_model_name"])
|