mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-07-28 16:07:46 +00:00
Xu Yang
1aa5cc2535
* feat: enhance timeout management and knowledge base handling in CoSTEER components * fix a little bug * fix small bug * fix a small bug * Update rdagent/scenarios/data_science/loop.py Co-authored-by: you-n-g <you-n-g@users.noreply.github.com> * add scale check * fix a small bug * fix CI * use dynamic chat_token_limit & remove repeated lines * fix CI * remove useless comment * fix small bug * update draft appendix * fix prompt * add code correctness as top priority --------- Co-authored-by: you-n-g <you-n-g@users.noreply.github.com>
43 lines
1.1 KiB
Python
43 lines
1.1 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 = 10
|
|
"""Maximum number of task implementation loops"""
|
|
|
|
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()
|