highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-06-23 06:21:00 +02:00
Code Issues Packages Projects Releases Wiki Activity

Committed LLM updates to db (#1899)

Browse Source
This commit is contained in:
hagen-danswer 2024-07-22 10:30:24 -07:00 committed by GitHub
parent 1276732409
commit 48a7fe80b1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
backend/danswer/db/llm.py
View File

@ -14,7 +14,7 @@ from danswer.server.manage.llm.models import FullLLMProvider
from danswer.server.manage.llm.models import LLMProviderUpsertRequest 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, llm_provider_id: int,
group_ids: list[int] | None, group_ids: list[int] | None,
db_session: Session, db_session: Session,
@ -34,7 +34,6 @@ def update_group_llm_provider_relationships(
for group_id in group_ids for group_id in group_ids
] ]
db_session.add_all(new_relationships) db_session.add_all(new_relationships)
db_session.commit()
def upsert_cloud_embedding_provider( def upsert_cloud_embedding_provider(
@ -83,12 +82,14 @@ def upsert_llm_provider(
db_session.flush() db_session.flush()
# Make sure the relationship table stays up to date # 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, llm_provider_id=existing_llm_provider.id,
group_ids=llm_provider.groups, group_ids=llm_provider.groups,
db_session=db_session, db_session=db_session,
) )
db_session.commit()
return FullLLMProvider.from_model(existing_llm_provider) return FullLLMProvider.from_model(existing_llm_provider)