mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-07-27 23:47:46 +00:00
Xu Yang
515fb50ce2
* filter feature which is high correlation to former implemented features * use multiprocessing to calculate IC and some minor fix
30 lines
800 B
Python
30 lines
800 B
Python
from pathlib import Path
|
|
from typing import Union
|
|
|
|
from pydantic_settings import BaseSettings
|
|
|
|
|
|
class ModelImplSettings(BaseSettings):
|
|
class Config:
|
|
env_prefix = "MODEL_CODER_" # Use MODEL_CODER_ as prefix for environment variables
|
|
|
|
coder_use_cache: bool = False
|
|
|
|
cache_location: str = str(
|
|
(Path().cwd() / "git_ignore_folder" / "model_implementation_execution_cache").absolute(),
|
|
)
|
|
|
|
knowledge_base_path: Union[str, None] = None
|
|
new_knowledge_base_path: Union[str, None] = knowledge_base_path
|
|
|
|
max_loop: int = 10
|
|
|
|
query_former_trace_limit: int = 5
|
|
query_similar_success_limit: int = 5
|
|
fail_task_trial_limit: int = 20
|
|
|
|
enable_execution_cache: bool = True # whether to enable the execution cache
|
|
|
|
|
|
MODEL_IMPL_SETTINGS = ModelImplSettings()
|