File size: 615 Bytes
55fd638
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
from pydantic_settings import BaseSettings
from typing import Optional


class Settings(BaseSettings):
    # Model settings
    MODEL_PATH: str = "./models/smartbugs-expert"
    HF_MODEL_ID: Optional[str] = None          # e.g. "your-hf-username/smart-contract-auditor"
    HF_TOKEN: Optional[str] = None
    USE_PATTERN_FALLBACK: bool = True

    # Generation settings
    MAX_NEW_TOKENS: int = 300
    TEMPERATURE: float = 0.7
    MAX_INPUT_LENGTH: int = 1024

    # Server
    HOST: str = "0.0.0.0"
    PORT: int = 8000

    class Config:
        env_file = ".env"


settings = Settings()