From 35d32ea3b0be0ee7687e9c1c06329b8f33214f0a Mon Sep 17 00:00:00 2001 From: Chris Weaver <25087905+Weves@users.noreply.github.com> Date: Thu, 10 Oct 2024 21:24:34 -0700 Subject: [PATCH] Fix indexing model server port for warmup (#2767) --- backend/danswer/background/update.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/backend/danswer/background/update.py b/backend/danswer/background/update.py index f7a00687c43..58c31236eae 100755 --- a/backend/danswer/background/update.py +++ b/backend/danswer/background/update.py @@ -48,8 +48,8 @@ from danswer.utils.variable_functionality import global_version from danswer.utils.variable_functionality import set_is_ee_based_on_env_variable from shared_configs.configs import INDEXING_MODEL_SERVER_HOST +from shared_configs.configs import INDEXING_MODEL_SERVER_PORT from shared_configs.configs import LOG_LEVEL -from shared_configs.configs import MODEL_SERVER_PORT logger = setup_logger() @@ -494,7 +494,7 @@ def update_loop( embedding_model = EmbeddingModel.from_db_model( search_settings=search_settings, server_host=INDEXING_MODEL_SERVER_HOST, - server_port=MODEL_SERVER_PORT, + server_port=INDEXING_MODEL_SERVER_PORT, ) warm_up_bi_encoder(embedding_model=embedding_model) logger.notice("First inference complete.")