Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 14 additions & 0 deletions rdagent/.env
Original file line number Diff line number Diff line change
@@ -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
5 changes: 3 additions & 2 deletions rdagent/components/coder/model_coder/conf.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand All @@ -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
Expand Down
11 changes: 9 additions & 2 deletions rdagent/log/ui/app.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down
2 changes: 1 addition & 1 deletion rdagent/scenarios/qlib/docker/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -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
Loading