fix: fix_dotenv_error (#257)

* fix_dotenv_error

* format with isort

* Update rdagent/app/cli.py

---------

Co-authored-by: you-n-g <you-n-g@users.noreply.github.com>
This commit is contained in:
Linlang
2024-09-10 19:38:35 +08:00
committed by GitHub
parent df1f5e9390
commit 92c46a66b6
5 changed files with 7 additions and 24 deletions
+6 -3
View File
@@ -5,11 +5,16 @@ This will
- make rdagent a nice entry and
- autoamtically load dotenv
"""
from dotenv import load_dotenv
load_dotenv(".env")
# 1) Make sure it is at the beginning of the script so that it will load dotenv before initializing BaseSettings.
# 2) The ".env" argument is necessary to make sure it loads `.env` from the current directory.
import subprocess
from importlib.resources import path as rpath
import fire
from dotenv import load_dotenv
from rdagent.app.data_mining.model import main as med_model
from rdagent.app.general_model.general_model import (
@@ -20,8 +25,6 @@ from rdagent.app.qlib_rd_loop.factor_from_report import main as fin_factor_repor
from rdagent.app.qlib_rd_loop.model import main as fin_model
from rdagent.app.utils.info import collect_info
load_dotenv()
def ui(port=80, log_dir="", debug=False):
"""
+1 -6
View File
@@ -1,9 +1,3 @@
from dotenv import load_dotenv
from rdagent.scenarios.general_model.scenario import GeneralModelScenario
load_dotenv(override=True)
import fire
from rdagent.components.coder.model_coder.task_loader import (
@@ -13,6 +7,7 @@ from rdagent.components.document_reader.document_reader import (
extract_first_page_screenshot_from_pdf,
)
from rdagent.log import rdagent_logger as logger
from rdagent.scenarios.general_model.scenario import GeneralModelScenario
from rdagent.scenarios.qlib.developer.model_coder import QlibModelCoSTEER
-5
View File
@@ -2,13 +2,8 @@ from dataclasses import field
from pathlib import Path
from typing import Optional
from dotenv import load_dotenv
from pydantic_settings import BaseSettings
# Load environment variables
load_dotenv(verbose=True, override=True)
DIRNAME = Path("./")
-6
View File
@@ -2,12 +2,6 @@ from __future__ import annotations
from pathlib import Path
from dotenv import load_dotenv
from pydantic_settings import BaseSettings
# make sure that env variable is loaded while calling Config()
load_dotenv(verbose=True, override=True)
from pydantic_settings import BaseSettings
-4
View File
@@ -2,15 +2,11 @@ from __future__ import annotations
from pathlib import Path
from dotenv import load_dotenv
from pydantic_settings import BaseSettings
# TODO: use pydantic for other modules in Qlib
# from pydantic_settings import BaseSettings
# make sure that env variable is loaded while calling Config()
load_dotenv(verbose=True, override=True)
class RDAgentSettings(BaseSettings):
# TODO: (xiao) I think LLMSetting may be a better name.