-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfig.py
28 lines (21 loc) · 955 Bytes
/
config.py
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
import configparser
import os
config = configparser.ConfigParser()
config.read('../config.ini')
print("Current working directory:", os.getcwd())
files_read = config.read('config.ini')
print("Files read:", files_read)
# print(config.sections())
environment = 'PRODUCTION' # OR 'DEVELOPMENT'
ollama_url = config[environment]['ollama_url']
chroma_host = config[environment]['chroma_host']
chroma_port = int(config[environment]['chroma_port'])
emb_model = config['DEFAULT']['emb_model']
ll_model_big = config['DEFAULT']['ll_model_big']
ll_model = config['DEFAULT']['ll_model']
ll_model_small = config['DEFAULT']['ll_model_small']
# ll_model_large_ctx = config['DEFAULT']['ll_model_large_ctx']
# ll_model_large_ctx_70b = config['DEFAULT']['ll_model_large_ctx_70b']
# ll_model_llama31_70b_instruct_q8 = config['DEFAULT']['ll_model_llama31_70b_instruct_q8']
# question1 = config['DEFAULT']['question1']
# collect_name = config['DEFAULT']['collect_name']