highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-09-26 11:58:28 +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
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.engine import get_session_context_manager
from danswer.db.llm import fetch_existing_llm_providers 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.db.llm import upsert_llm_provider
from danswer.server.manage.llm.models import LLMProviderUpsertRequest from danswer.server.manage.llm.models import LLMProviderUpsertRequest
from danswer.utils.logger import setup_logger from danswer.utils.logger import setup_logger
@@ -42,8 +43,11 @@ def _seed_llms(
return return
logger.info("Seeding LLMs") logger.info("Seeding LLMs")
for llm_upsert_request in llm_upsert_requests: seeded_providers = [
upsert_llm_provider(db_session, llm_upsert_request) 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: def get_seed_config() -> SeedConfiguration | None: