quick update to some naming

This commit is contained in:
pablodanswer
2024-09-13 16:52:25 -07:00
parent 4b2e4ca159
commit 7b91beb3b2
8 changed files with 28 additions and 9 deletions

View File

@@ -135,7 +135,7 @@ POSTGRES_PASSWORD = urllib.parse.quote_plus(
os.environ.get("POSTGRES_PASSWORD") or "password" os.environ.get("POSTGRES_PASSWORD") or "password"
) )
POSTGRES_HOST = os.environ.get("POSTGRES_HOST") or "localhost" POSTGRES_HOST = os.environ.get("POSTGRES_HOST") or "localhost"
POSTGRES_PORT = os.environ.get("POSTGRES_PORT") or "5432" POSTGRES_PORT = os.environ.get("POSTGRES_PORT") or "5433"
POSTGRES_DB = os.environ.get("POSTGRES_DB") or "postgres" POSTGRES_DB = os.environ.get("POSTGRES_DB") or "postgres"
# defaults to False # defaults to False

View File

@@ -12,6 +12,7 @@ from danswer.db.models import User__UserGroup
from danswer.server.manage.embedding.models import CloudEmbeddingProvider from danswer.server.manage.embedding.models import CloudEmbeddingProvider
from danswer.server.manage.embedding.models import CloudEmbeddingProviderCreationRequest from danswer.server.manage.embedding.models import CloudEmbeddingProviderCreationRequest
from danswer.server.manage.llm.models import FullLLMProvider from danswer.server.manage.llm.models import FullLLMProvider
from danswer.server.manage.llm.models import FullLLMProviderSnapshot
from danswer.server.manage.llm.models import LLMProviderUpsertRequest from danswer.server.manage.llm.models import LLMProviderUpsertRequest
from shared_configs.enums import EmbeddingProvider from shared_configs.enums import EmbeddingProvider
@@ -59,9 +60,12 @@ def upsert_cloud_embedding_provider(
return CloudEmbeddingProvider.from_request(existing_provider) return CloudEmbeddingProvider.from_request(existing_provider)
# from danswer.server.manage.llm.models import FullLLMProviderSnapshot
def upsert_llm_provider( def upsert_llm_provider(
llm_provider: LLMProviderUpsertRequest, db_session: Session llm_provider: LLMProviderUpsertRequest, db_session: Session
) -> FullLLMProvider: ) -> FullLLMProviderSnapshot:
existing_llm_provider = db_session.scalar( existing_llm_provider = db_session.scalar(
select(LLMProviderModel).where(LLMProviderModel.name == llm_provider.name) select(LLMProviderModel).where(LLMProviderModel.name == llm_provider.name)
) )
@@ -94,7 +98,9 @@ def upsert_llm_provider(
db_session.commit() db_session.commit()
return FullLLMProvider.from_model(existing_llm_provider) return FullLLMProviderSnapshot.from_full_llm_provider(
FullLLMProvider.from_model(existing_llm_provider)
)
def fetch_existing_embedding_providers( def fetch_existing_embedding_providers(

View File

@@ -92,7 +92,7 @@ class SearchSettingsSnapshot(SavedSearchSettings):
def from_saved_settings( def from_saved_settings(
cls, settings: SavedSearchSettings cls, settings: SavedSearchSettings
) -> "SearchSettingsSnapshot": ) -> "SearchSettingsSnapshot":
data = settings.dict(exclude={"rerank_api_key"}) data = settings.dict(exclude={"rerank_api_key", "api_key"})
data["rerank_api_key"] = obfuscate_api_key(settings.rerank_api_key) data["rerank_api_key"] = obfuscate_api_key(settings.rerank_api_key)
data["api_key"] = obfuscate_api_key(settings.api_key) data["api_key"] = obfuscate_api_key(settings.api_key)

View File

@@ -19,6 +19,7 @@ from danswer.llm.llm_provider_options import fetch_available_well_known_llms
from danswer.llm.llm_provider_options import WellKnownLLMProviderDescriptor from danswer.llm.llm_provider_options import WellKnownLLMProviderDescriptor
from danswer.llm.utils import test_llm from danswer.llm.utils import test_llm
from danswer.server.manage.llm.models import FullLLMProvider from danswer.server.manage.llm.models import FullLLMProvider
from danswer.server.manage.llm.models import FullLLMProviderSnapshot
from danswer.server.manage.llm.models import LLMProviderDescriptor from danswer.server.manage.llm.models import LLMProviderDescriptor
from danswer.server.manage.llm.models import LLMProviderUpsertRequest from danswer.server.manage.llm.models import LLMProviderUpsertRequest
from danswer.server.manage.llm.models import TestLLMRequest from danswer.server.manage.llm.models import TestLLMRequest
@@ -120,8 +121,10 @@ def put_llm_provider(
llm_provider: LLMProviderUpsertRequest, llm_provider: LLMProviderUpsertRequest,
_: User | None = Depends(current_admin_user), _: User | None = Depends(current_admin_user),
db_session: Session = Depends(get_session), db_session: Session = Depends(get_session),
) -> FullLLMProvider: ) -> FullLLMProviderSnapshot:
return upsert_llm_provider(llm_provider=llm_provider, db_session=db_session) return FullLLMProviderSnapshot.from_full_llm_provider(
upsert_llm_provider(llm_provider=llm_provider, db_session=db_session)
)
@admin_router.delete("/provider/{provider_id}") @admin_router.delete("/provider/{provider_id}")

View File

@@ -101,3 +101,13 @@ class FullLLMProvider(LLMProvider):
is_public=llm_provider_model.is_public, is_public=llm_provider_model.is_public,
groups=[group.id for group in llm_provider_model.groups], groups=[group.id for group in llm_provider_model.groups],
) )
class FullLLMProviderSnapshot(FullLLMProvider):
@classmethod
def from_full_llm_provider(
cls, settings: FullLLMProvider
) -> "FullLLMProviderSnapshot":
data = settings.dict(exclude={"api_key"})
data["api_key"] = obfuscate_api_key(settings.api_key)
return cls(**data)

View File

@@ -292,7 +292,7 @@ services:
- POSTGRES_USER=${POSTGRES_USER:-postgres} - POSTGRES_USER=${POSTGRES_USER:-postgres}
- POSTGRES_PASSWORD=${POSTGRES_PASSWORD:-password} - POSTGRES_PASSWORD=${POSTGRES_PASSWORD:-password}
ports: ports:
- "5432:5432" - "5433:5432"
volumes: volumes:
- db_volume:/var/lib/postgresql/data - db_volume:/var/lib/postgresql/data

View File

@@ -302,7 +302,7 @@ services:
- POSTGRES_USER=${POSTGRES_USER:-postgres} - POSTGRES_USER=${POSTGRES_USER:-postgres}
- POSTGRES_PASSWORD=${POSTGRES_PASSWORD:-password} - POSTGRES_PASSWORD=${POSTGRES_PASSWORD:-password}
ports: ports:
- "5432:5432" - "5433:5432"
volumes: volumes:
- db_volume:/var/lib/postgresql/data - db_volume:/var/lib/postgresql/data

View File

@@ -154,7 +154,7 @@ services:
- POSTGRES_USER=${POSTGRES_USER:-postgres} - POSTGRES_USER=${POSTGRES_USER:-postgres}
- POSTGRES_PASSWORD=${POSTGRES_PASSWORD:-password} - POSTGRES_PASSWORD=${POSTGRES_PASSWORD:-password}
ports: ports:
- "5432" - "5433"
volumes: volumes:
- db_volume:/var/lib/postgresql/data - db_volume:/var/lib/postgresql/data