fixed timeouts

sweepai · Jun 23, 2023 · c08b276 · c08b276
1 parent f5cb511
commit c08b276
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/core/vector_db.py b/src/core/vector_db.py
@@ -230,7 +230,7 @@ def get_deeplake_vs_from_repo(
         logger.error("No documents found in repository")
         return deeplake_vs
 
-@stub.function(image=image, secrets=secrets, timeout=15 * 60, shared_volumes={DISKCACHE_DIR: model_volume}, timeout=timeout)
+@stub.function(image=image, secrets=secrets, shared_volumes={DISKCACHE_DIR: model_volume}, timeout=timeout)
 def init_index(
     repo_name: str,
     installation_id: int,
@@ -239,7 +239,7 @@ def init_index(
     pass
 
 
-@stub.function(image=image, secrets=secrets, timeout=30 * 60, shared_volumes={DISKCACHE_DIR: model_volume}, timeout=timeout)
+@stub.function(image=image, secrets=secrets, shared_volumes={DISKCACHE_DIR: model_volume}, timeout=timeout)
 def update_index(
     repo_name,
     installation_id: int,