Files
NexQuant/rdagent/components/coder/CoSTEER/config.py
T
TPTBusiness 68ea969c32 feat: Add parallel run system with API key distribution
- Add predix_parallel.py: Run multiple factor experiments concurrently
  * python predix_parallel.py --runs 5 --api-keys 2 -m openrouter
  * Round-robin API key distribution across available keys
  * Rich live dashboard with per-run status, elapsed time, exit codes
  * Graceful shutdown (Ctrl+C kills all children cleanly)

- Add --run-id parameter to predix.py for isolated single runs
  * Separate log files: fin_quant_run{N}.log
  * Separate results: results/runs/run{N}/
  * Separate workspace: RD-Agent_workspace_run{N}/
  * Separate databases per run

- Modify CoSTEER and FactorRunner for PARALLEL_RUN_ID isolation
  * _save_intermediate_results uses run-specific directories
  * _save_result_to_database and _write_run_log isolated per run
  * _ensure_results_dirs creates run-specific paths

- Reduce max_loop from 10 to 3 for faster iterations
- Add docs/parallel_runs.md with full documentation

Tests: 103 passed
2026-04-04 09:39:12 +02:00

43 lines
1.2 KiB
Python

from typing import Union
from rdagent.core.conf import ExtendedBaseSettings
class CoSTEERSettings(ExtendedBaseSettings):
"""CoSTEER settings, this setting is supposed not to be used directly!!!"""
class Config:
env_prefix = "CoSTEER_"
coder_use_cache: bool = False
"""Indicates whether to use cache for the coder"""
max_loop: int = 3
"""Maximum number of task implementation loops (reduced from 10 for faster iterations)"""
fail_task_trial_limit: int = 20
v1_query_former_trace_limit: int = 3
v1_query_similar_success_limit: int = 3
v2_query_component_limit: int = 1
v2_query_error_limit: int = 1
v2_query_former_trace_limit: int = 3
v2_add_fail_attempt_to_latest_successful_execution: bool = False
v2_error_summary: bool = False
v2_knowledge_sampler: float = 1.0
knowledge_base_path: Union[str, None] = None
"""Path to the knowledge base"""
new_knowledge_base_path: Union[str, None] = None
"""Path to the new knowledge base"""
enable_filelock: bool = False
filelock_path: Union[str, None] = None
max_seconds_multiplier: int = 10**6
CoSTEER_SETTINGS = CoSTEERSettings()