highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-10-11 13:46:07 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix indexing model server port for warmup (#2767)

Browse Source
This commit is contained in:
Chris Weaver
2024-10-10 21:24:34 -07:00
committed by GitHub
parent 1581d35476
commit 35d32ea3b0
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
backend/danswer/background/update.py
View File

@@ -48,8 +48,8 @@ from danswer.utils.logger import setup_logger
from danswer.utils.variable_functionality import global_version from danswer.utils.variable_functionality import global_version
from danswer.utils.variable_functionality import set_is_ee_based_on_env_variable 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_HOST
from shared_configs.configs import INDEXING_MODEL_SERVER_PORT
from shared_configs.configs import LOG_LEVEL from shared_configs.configs import LOG_LEVEL
from shared_configs.configs import MODEL_SERVER_PORT
logger = setup_logger() logger = setup_logger()
@@ -494,7 +494,7 @@ def update_loop(
embedding_model = EmbeddingModel.from_db_model( embedding_model = EmbeddingModel.from_db_model(
search_settings=search_settings, search_settings=search_settings,
server_host=INDEXING_MODEL_SERVER_HOST, 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) warm_up_bi_encoder(embedding_model=embedding_model)
logger.notice("First inference complete.") logger.notice("First inference complete.")