diff --git a/scraibe/localai_client.py b/scraibe/localai_client.py index 714c5a7..dc873e2 100644 --- a/scraibe/localai_client.py +++ b/scraibe/localai_client.py @@ -48,7 +48,7 @@ class LocalAIClient: api_url: Optional[str] = None, api_key: Optional[str] = None, model: Optional[str] = None, - timeout: float = 600.0, + timeout: float = 3600.0, ): """ Args: diff --git a/scraibe/summarizer.py b/scraibe/summarizer.py index e2d72bd..a61c1fb 100644 --- a/scraibe/summarizer.py +++ b/scraibe/summarizer.py @@ -43,7 +43,7 @@ class SummarizerClient: api_url: Optional[str] = None, api_key: Optional[str] = None, model: Optional[str] = None, - timeout: float = 600.0, + timeout: float = 3600.0, ): self.api_url = (api_url or os.getenv("SUMMARIZER_API_URL")).strip().rstrip("/") self.api_key = api_key or os.getenv("SUMMARIZER_API_KEY") or None diff --git a/scraibe/tasks.py b/scraibe/tasks.py index 78d18fc..df605f3 100644 --- a/scraibe/tasks.py +++ b/scraibe/tasks.py @@ -251,6 +251,8 @@ def send_error_email(to: str, error_message: str, task_id: str): name="scraibe.tasks.process_transcription_task", bind=True, max_retries=1, + task_time_limit=14400, # 4 hours + task_soft_time_limit=13500, # warn at 3h45m ) def process_transcription_task( self,