From b5aa7370a2e27041104898930635528cde570067 Mon Sep 17 00:00:00 2001 From: Weves Date: Wed, 26 Jun 2024 16:00:23 -0700 Subject: [PATCH] Make seeded model default --- backend/ee/danswer/server/seeding.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/backend/ee/danswer/server/seeding.py b/backend/ee/danswer/server/seeding.py index 9047d7c2f..9245c20ca 100644 --- a/backend/ee/danswer/server/seeding.py +++ b/backend/ee/danswer/server/seeding.py @@ -5,6 +5,7 @@ from sqlalchemy.orm import Session from danswer.db.engine import get_session_context_manager from danswer.db.llm import fetch_existing_llm_providers +from danswer.db.llm import update_default_provider from danswer.db.llm import upsert_llm_provider from danswer.server.manage.llm.models import LLMProviderUpsertRequest from danswer.utils.logger import setup_logger @@ -42,8 +43,11 @@ def _seed_llms( return logger.info("Seeding LLMs") - for llm_upsert_request in llm_upsert_requests: + seeded_providers = [ upsert_llm_provider(db_session, llm_upsert_request) + for llm_upsert_request in llm_upsert_requests + ] + update_default_provider(db_session, seeded_providers[0].id) def get_seed_config() -> SeedConfiguration | None: