2024-05-21 22:48:41 +08:00
|
|
|
# TODO: use pydantic for other modules in Qlib
|
|
|
|
|
# from pydantic_settings import BaseSettings
|
|
|
|
|
import os
|
|
|
|
|
from typing import Union
|
|
|
|
|
|
|
|
|
|
from dotenv import load_dotenv
|
|
|
|
|
|
|
|
|
|
# make sure that env variable is loaded while calling Config()
|
|
|
|
|
load_dotenv(verbose=True, override=True)
|
|
|
|
|
|
|
|
|
|
from pydantic_settings import BaseSettings
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class FincoSettings(BaseSettings):
|
|
|
|
|
use_azure: bool = True
|
|
|
|
|
max_retry: int = 10
|
|
|
|
|
retry_wait_seconds: int = 1
|
|
|
|
|
continuous_mode: bool = False
|
|
|
|
|
dump_chat_cache: bool = False
|
|
|
|
|
use_chat_cache: bool = False
|
|
|
|
|
dump_embedding_cache: bool = False
|
|
|
|
|
use_embedding_cache: bool = False
|
|
|
|
|
prompt_cache_path: str = os.getcwd() + "/prompt_cache.db"
|
|
|
|
|
session_cache_folder_location: str = os.getcwd() + "/session_cache_folder/"
|
|
|
|
|
max_past_message_include: int = 10
|
|
|
|
|
|
|
|
|
|
log_llm_chat_content: bool = True
|
|
|
|
|
|
|
|
|
|
# Chat configs
|
|
|
|
|
chat_openai_api_key: str = ""
|
|
|
|
|
chat_azure_api_base: str = ""
|
|
|
|
|
chat_azure_api_version: str = ""
|
|
|
|
|
chat_model: str = ""
|
|
|
|
|
chat_max_tokens: int = 3000
|
|
|
|
|
chat_temperature: float = 0.5
|
|
|
|
|
chat_stream: bool = True
|
|
|
|
|
chat_seed: Union[int, None] = None
|
|
|
|
|
chat_frequency_penalty: float = 0.0
|
|
|
|
|
chat_presence_penalty: float = 0.0
|
|
|
|
|
|
|
|
|
|
default_system_prompt: str = "You are an AI assistant who helps to answer user's questions about finance."
|
|
|
|
|
|
|
|
|
|
# Embedding configs
|
|
|
|
|
embedding_openai_api_key: str = ""
|
|
|
|
|
embedding_azure_api_base: str = ""
|
|
|
|
|
embedding_azure_api_version: str = ""
|
|
|
|
|
embedding_model: str = ""
|
