File size: 1,695 Bytes
d2bfe97 | 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 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 | """
config.py
Central configuration for the YouTube Transcript Toolkit.
Author: algorembrant
"""
# ---------------------------------------------------------------------------
# Model settings
# ---------------------------------------------------------------------------
# claude-haiku-4-5 is used by default for speed.
# Switch to claude-sonnet-4-6 for higher quality at the cost of latency.
DEFAULT_MODEL = "claude-haiku-4-5-20251001"
QUALITY_MODEL = "claude-sonnet-4-6"
MAX_TOKENS = 8192 # Maximum tokens to request from the model
CHUNK_SIZE = 60_000 # Characters per chunk for very long transcripts
# ---------------------------------------------------------------------------
# Transcript defaults
# ---------------------------------------------------------------------------
DEFAULT_LANGUAGES = ["en"]
# ---------------------------------------------------------------------------
# Summary modes
# ---------------------------------------------------------------------------
SUMMARY_MODES = {
"brief": {
"label": "Brief",
"description": "3-5 sentence executive summary",
},
"detailed": {
"label": "Detailed",
"description": "Comprehensive multi-section breakdown",
},
"bullets": {
"label": "Bullet Points",
"description": "Key takeaways as a structured bullet list",
},
"outline": {
"label": "Outline",
"description": "Hierarchical topic outline",
},
}
# ---------------------------------------------------------------------------
# Output formats
# ---------------------------------------------------------------------------
OUTPUT_FORMATS = ["text", "json", "srt", "vtt"]
|