diff --git a/src/core/vector_db.py b/src/core/vector_db.py index d1504e0acf..bd88619ec0 100644 --- 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,