highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-04-09 20:39:29 +02:00
Code Issues Packages Projects Releases Wiki Activity

Make seeded model default

Browse Source
This commit is contained in:
Weves 2024-06-26 16:00:23 -07:00 committed by Chris Weaver
parent 4ba6e45128
commit b5aa7370a2
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/ee/danswer/server/seeding.py
View File

@ -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: