AuditAgent / app /config.py
Parsa2025AI's picture
backend
55fd638 verified
raw
history blame contribute delete
615 Bytes
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()