From c08b27659caadfa2981d88fd142467ce1521d736 Mon Sep 17 00:00:00 2001 From: Kevin Lu Date: Thu, 22 Jun 2023 18:09:10 -0700 Subject: [PATCH] fixed timeouts --- src/core/vector_db.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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,