From 4907d2271950fb2f45c56c21e6d641b616c02ad7 Mon Sep 17 00:00:00 2001 From: pablodanswer Date: Sun, 3 Nov 2024 19:22:07 -0800 Subject: [PATCH] naming --- backend/ee/danswer/server/tenants/api.py | 4 ++-- backend/ee/danswer/server/tenants/provisioning.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/backend/ee/danswer/server/tenants/api.py b/backend/ee/danswer/server/tenants/api.py index e412338e514..8e79c0b37b1 100644 --- a/backend/ee/danswer/server/tenants/api.py +++ b/backend/ee/danswer/server/tenants/api.py @@ -27,9 +27,9 @@ from ee.danswer.server.tenants.models import ImpersonateRequest from ee.danswer.server.tenants.models import ProductGatingRequest from ee.danswer.server.tenants.provisioning import add_users_to_tenant +from ee.danswer.server.tenants.provisioning import configure_default_api_keys from ee.danswer.server.tenants.provisioning import ensure_schema_exists from ee.danswer.server.tenants.provisioning import run_alembic_migrations -from ee.danswer.server.tenants.provisioning import setup_api_keys from ee.danswer.server.tenants.provisioning import user_owns_a_tenant from shared_configs.configs import MULTI_TENANT from shared_configs.contextvars import CURRENT_TENANT_ID_CONTEXTVAR @@ -68,7 +68,7 @@ def create_tenant( with get_session_with_tenant(tenant_id) as db_session: setup_danswer(db_session, tenant_id) - setup_api_keys(db_session) + configure_default_api_keys(db_session) add_users_to_tenant([email], tenant_id) diff --git a/backend/ee/danswer/server/tenants/provisioning.py b/backend/ee/danswer/server/tenants/provisioning.py index 1f4ace08c91..9106821b5a5 100644 --- a/backend/ee/danswer/server/tenants/provisioning.py +++ b/backend/ee/danswer/server/tenants/provisioning.py @@ -60,7 +60,7 @@ def run_alembic_migrations(schema_name: str) -> None: raise -def setup_api_keys(db_session: Session) -> None: +def configure_default_api_keys(db_session: Session) -> None: open_provider = LLMProviderUpsertRequest( name="OpenAI", provider="OpenAI",