You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
37 lines
2.1 KiB
37 lines
2.1 KiB
import json |
|
import os |
|
|
|
|
|
class Config: |
|
def __init__(self): |
|
config_path = os.path.join(os.path.dirname(__file__), 'config.json') |
|
if not os.path.exists(config_path): |
|
raise FileNotFoundError(f'config.json not found at {config_path}, ' |
|
f'please copy config.example.json to config.json and modify it.') |
|
with open(config_path, 'r') as f: |
|
self.config = json.load(f) |
|
self.language = self.config.get('language', 'Chinese') |
|
self.open_ai_key = self.config.get('open_ai_key') |
|
self.open_ai_proxy = self.config.get('open_ai_proxy') |
|
self.open_ai_chat_model = self.config.get('open_ai_chat_model', 'gpt-3.5-turbo') |
|
if not self.open_ai_key: |
|
raise ValueError('open_ai_key is not set') |
|
self.temperature = self.config.get('temperature', 0.1) |
|
if self.temperature < 0 or self.temperature > 1: |
|
raise ValueError('temperature must be between 0 and 1, less is more conservative, more is more creative') |
|
self.use_stream = self.config.get('use_stream', False) |
|
self.use_postgres = self.config.get('use_postgres', False) |
|
if not self.use_postgres: |
|
self.index_path = self.config.get('index_path', './temp') |
|
os.makedirs(self.index_path, exist_ok=True) |
|
self.postgres_url = self.config.get('postgres_url') |
|
if self.use_postgres and self.postgres_url is None: |
|
raise ValueError('postgres_url is not set') |
|
self.modes = self.config.get('mode', 'webui').split(',') |
|
for mode in self.modes: |
|
if mode not in ['console', 'api', 'webui','train_model','tg_bot','console_assistant']: |
|
raise ValueError('mode must be console or api or webui') |
|
self.api_port = self.config.get('api_port', 9531) |
|
self.api_host = self.config.get('api_host', 'localhost') |
|
self.webui_port = self.config.get('webui_port', 7860) |
|
self.webui_host = self.config.get('webui_host', '0.0.0.0')
|
|
|