diff --git a/rdagent/.env b/rdagent/.env new file mode 100644 index 000000000..ea80ae465 --- /dev/null +++ b/rdagent/.env @@ -0,0 +1,14 @@ +# CHAT MODEL: Using DeepSeek Official API +CHAT_MODEL=deepseek/deepseek-chat +DEEPSEEK_API_KEY=sk-fb7bbafeb5b8439084453d22e24548f1 + +# EMBEDDING MODEL: Using SiliconFlow for embedding since deepseek has no embedding model. +# Note: embedding requires litellm_proxy prefix +EMBEDDING_MODEL=litellm_proxy/BAAI/bge-m3 +LITELLM_PROXY_API_KEY=sk-pvxqhsrimhhvtkjqpyypvxipmuyrnzzirhkgokyyhqzmmmqj +LITELLM_PROXY_API_BASE=https://api.siliconflow.cn/v1 +MODEL_COSTEER_ENV_TYPE=docker +QLIB_DOCKER_ENABLE_GPU=true +QLIB_QUANT_TEST_END=2020-09-25 +QLIB_FACTOR_TEST_END=2020-09-25 +QLIB_MODEL_TEST_END=2020-09-25 diff --git a/rdagent/components/coder/model_coder/conf.py b/rdagent/components/coder/model_coder/conf.py index 1cb8d2d35..e7251aa9d 100644 --- a/rdagent/components/coder/model_coder/conf.py +++ b/rdagent/components/coder/model_coder/conf.py @@ -15,7 +15,7 @@ class ModelCoSTEERSettings(CoSTEERSettings): def get_model_env( conf_type: Optional[str] = None, - extra_volumes: dict = {}, + extra_volumes: Optional[dict] = None, running_timeout_period: int = 600, enable_cache: Optional[bool] = None, ) -> Env: @@ -27,7 +27,8 @@ def get_model_env( else: raise ValueError(f"Unknown env type: {conf.env_type}") - env.conf.extra_volumes = extra_volumes.copy() + if extra_volumes is not None: + env.conf.extra_volumes = extra_volumes.copy() env.conf.running_timeout_period = running_timeout_period if enable_cache is not None: env.conf.enable_cache = enable_cache diff --git a/rdagent/log/ui/app.py b/rdagent/log/ui/app.py index 605e97d22..c5f6a031f 100644 --- a/rdagent/log/ui/app.py +++ b/rdagent/log/ui/app.py @@ -1104,10 +1104,17 @@ def analyze_task_completion(): if isinstance(state.scenario, SIMILAR_SCENARIOS): st.header("R&D Loops♾️", divider="rainbow", anchor="_rdloops") if len(state.msgs) > 1: - r_options = list(state.msgs.keys()) + r_options = sorted(state.msgs.keys()) if 0 in r_options: r_options.remove(0) - round = st.radio("**Loops**", horizontal=True, options=r_options, index=state.lround - 1) + if r_options: + if state.lround in r_options: + r_index = r_options.index(state.lround) + else: + r_index = 0 + round = st.radio("**Loops**", horizontal=True, options=r_options, index=r_index) + else: + round = 1 else: round = 1 diff --git a/rdagent/scenarios/qlib/docker/Dockerfile b/rdagent/scenarios/qlib/docker/Dockerfile index 598f2d92c..684a98f3b 100644 --- a/rdagent/scenarios/qlib/docker/Dockerfile +++ b/rdagent/scenarios/qlib/docker/Dockerfile @@ -21,5 +21,5 @@ RUN python -m pip install -e . RUN pip install catboost RUN pip install xgboost -RUN pip install scipy==1.11.4 +RUN pip install scipy==1.12.0 RUN pip install tables