| from pydantic import Field | |
| from pydantic_settings import BaseSettings | |
| class Config(BaseSettings): | |
| hf_token: str = Field(...) | |
| model_path: str = Field("InvestmentResearchAI/LLM-ADE-dev") | |
| headless: bool = Field(False, description="Run in headless mode.") | |
| chat_template: str = Field("chatml", description="Chat template for prompt formatting") | |
| num_fewshot: int | None = Field(None, description="Option to use json mode examples") | |
| load_in_4bit: str = Field("False", description="Option to load in 4bit with bitsandbytes") | |
| max_depth: int = Field(5, description="Maximum number of recursive iteration") | |
| config = Config(_env_file=".env") | |