summary / settings.py
xsestech's picture
Changed generation prompts
975cc83 verified
raw
history blame
764 Bytes
import enum
from pydantic_settings import BaseSettings
class LogLevel(str, enum.Enum):
"""Possible log levels."""
NOTSET = "NOTSET"
DEBUG = "DEBUG"
INFO = "INFO"
WARNING = "WARNING"
ERROR = "ERROR"
FATAL = "FATAL"
class Settings(BaseSettings):
temperature: float = 0.2
model: str = "llama3-8b-8192"
log_level: LogLevel = LogLevel.INFO
segment_time: int = 1500
max_tokens: int = 375
system_prompt: str = ("You are a helpful, respectful and honest assistant."
"You task will be to summarize video transcripts. "
"You will receive them in parts, so don't indicate beggining."
"Answer only in russian.")
app_settings = Settings()