mirror of
https://github.com/microsoft/qlib.git
synced 2026-07-11 06:46:56 +08:00
73d51f05b4
* Update setup.py and config * WIP * init_workspace and CMDTask * Delete test_sumarize.py
32 lines
1.3 KiB
Python
32 lines
1.3 KiB
Python
# TODO: use pydantic for other modules in Qlib
|
|
from pydantic import BaseSettings
|
|
from qlib.finco.utils import Singleton
|
|
|
|
import os
|
|
|
|
|
|
class Config(Singleton):
|
|
"""
|
|
This config is for fast demo purpose.
|
|
Please use BaseSettings insetead in the future
|
|
"""
|
|
|
|
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 = 800 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 = int(os.getenv("MAX_RETRY")) if os.getenv("MAX_RETRY") is not None else None
|
|
|
|
self.continous_mode = (
|
|
os.getenv("CONTINOUS_MODE") == "True" if os.getenv("CONTINOUS_MODE") is not None else False
|
|
)
|
|
self.debug_mode = os.getenv("DEBUG_MODE") == "True" if os.getenv("DEBUG_MODE") is not None else False
|
|
self.workspace = os.getenv("WORKSPACE") if os.getenv("WORKSPACE") is not None else "./finco_workspace"
|