From 48a7fe80b1f250200cd5b80bc3ae2bf5b387e973 Mon Sep 17 00:00:00 2001 From: hagen-danswer Date: Mon, 22 Jul 2024 10:30:24 -0700 Subject: [PATCH] Committed LLM updates to db (#1899) --- backend/danswer/db/llm.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/backend/danswer/db/llm.py b/backend/danswer/db/llm.py index 07ef44fff..b22329f13 100644 --- a/backend/danswer/db/llm.py +++ b/backend/danswer/db/llm.py @@ -14,7 +14,7 @@ from danswer.server.manage.llm.models import FullLLMProvider from danswer.server.manage.llm.models import LLMProviderUpsertRequest -def update_group_llm_provider_relationships( +def update_group_llm_provider_relationships__no_commit( llm_provider_id: int, group_ids: list[int] | None, db_session: Session, @@ -34,7 +34,6 @@ def update_group_llm_provider_relationships( for group_id in group_ids ] db_session.add_all(new_relationships) - db_session.commit() def upsert_cloud_embedding_provider( @@ -83,12 +82,14 @@ def upsert_llm_provider( db_session.flush() # Make sure the relationship table stays up to date - update_group_llm_provider_relationships( + update_group_llm_provider_relationships__no_commit( llm_provider_id=existing_llm_provider.id, group_ids=llm_provider.groups, db_session=db_session, ) + db_session.commit() + return FullLLMProvider.from_model(existing_llm_provider)