diff --git a/server/docker-compose.yml b/server/docker-compose.yml index f34a8e0a..374130fa 100644 --- a/server/docker-compose.yml +++ b/server/docker-compose.yml @@ -7,6 +7,7 @@ services: - 1250:1250 environment: LLM_URL: "${LLM_URL}" + MIN_TRANSCRIPT_LENGTH: "${MIN_TRANSCRIPT_LENGTH}" volumes: - model-cache:/root/.cache diff --git a/server/reflector/processors/transcript_topic_detector.py b/server/reflector/processors/transcript_topic_detector.py index 43bf9762..297c55a5 100644 --- a/server/reflector/processors/transcript_topic_detector.py +++ b/server/reflector/processors/transcript_topic_detector.py @@ -1,6 +1,7 @@ from reflector.llm import LLM, LLMTaskParams from reflector.processors.base import Processor from reflector.processors.types import TitleSummary, Transcript +from reflector.settings import settings class TranscriptTopicDetectorProcessor(Processor): @@ -12,7 +13,9 @@ class TranscriptTopicDetectorProcessor(Processor): OUTPUT_TYPE = TitleSummary TASK = "topic" - def __init__(self, min_transcript_length: int = 750, **kwargs): + def __init__( + self, min_transcript_length: int = int(settings.MIN_TRANSCRIPT_LENGTH), **kwargs + ): super().__init__(**kwargs) self.transcript = None self.min_transcript_length = min_transcript_length diff --git a/server/reflector/settings.py b/server/reflector/settings.py index 249661db..f42ff0a4 100644 --- a/server/reflector/settings.py +++ b/server/reflector/settings.py @@ -97,5 +97,8 @@ class Settings(BaseSettings): # Cache directory for all model storage CACHE_DIR: str = "./data" + # Min transcript length to generate topic + summary + MIN_TRANSCRIPT_LENGTH: int = 1800 + settings = Settings()