mirror of
https://github.com/microsoft/qlib.git
synced 2026-07-06 12:30:57 +08:00
24 lines
991 B
Python
24 lines
991 B
Python
# TODO: use pydantic for other modules in Qlib
|
|
from pydantic import (BaseSettings)
|
|
|
|
import os
|
|
|
|
class Config():
|
|
_instance = None
|
|
def __new__(cls, *args, **kwargs):
|
|
if cls._instance is None:
|
|
cls._instance = super().__new__(cls, *args, **kwargs)
|
|
return cls._instance
|
|
|
|
def __init__(self):
|
|
self.use_azure = os.getenv("USE_AZURE") == "True"
|
|
self.temperature = 0.5 if os.getenv("TEMPERATURE") is None else float(os.getenv("TEMPERATURE"))
|
|
self.max_tokens = 8000 if os.getenv("MAX_TOKENS") is None else int(os.getenv("MAX_TOKENS"))
|
|
|
|
self.openai_api_key = os.getenv("OPENAI_API_KEY")
|
|
self.use_azure = os.getenv("USE_AZURE") == "True"
|
|
self.azure_api_base = os.getenv("AZURE_API_BASE")
|
|
self.azure_api_version = os.getenv("AZURE_API_VERSION")
|
|
self.model = os.getenv("MODEL") or ("gpt-35-turbo" if self.use_azure else "gpt-3.5-turbo")
|
|
|
|
self.max_retry = os.getenv("MAX_RETRY") |