This commit is contained in:
pablodanswer
2024-09-19 16:07:03 -07:00
parent 14871a62ad
commit 3cc99cf79a
6 changed files with 40 additions and 6 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

@@ -13,6 +13,8 @@ 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 FullLLMProviderSnapshot
from danswer.server.manage.llm.models import LLMProviderCreationRequest
from danswer.server.manage.llm.models import LLMProviderUpdateRequest
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
@@ -60,6 +62,38 @@ def upsert_cloud_embedding_provider(
return CloudEmbeddingProvider.from_request(existing_provider) return CloudEmbeddingProvider.from_request(existing_provider)
def update_llm_provider(
llm_provider_update: LLMProviderUpdateRequest,
db_session: Session,
) -> FullLLMProviderSnapshot:
existing_llm_provider = db_session.scalar(
select(LLMProviderModel).where(
LLMProviderModel.name == llm_provider_update.name
)
)
# if llm_provider_update.api_key_set:
if not existing_llm_provider:
raise ValueError(
f"LLM Provider with name {llm_provider_update.name} does not exist"
)
return FullLLMProviderSnapshot.from_full_llm_provider(
FullLLMProvider.from_model(existing_llm_provider)
)
def create_llm_provider(
llm_provider_creation: LLMProviderCreationRequest,
db_session: Session,
) -> FullLLMProviderSnapshot:
new_llm_provider = LLMProviderModel(name=llm_provider_creation.name)
db_session.add(new_llm_provider)
db_session.commit()
return FullLLMProviderSnapshot.from_full_llm_provider(
FullLLMProvider.from_model(new_llm_provider)
)
def upsert_llm_provider( def upsert_llm_provider(
llm_provider: LLMProviderUpsertRequest, llm_provider: LLMProviderUpsertRequest,
db_session: Session, db_session: Session,

View File

@@ -73,11 +73,11 @@ class LLMProviderUpsertRequest(LLMProvider):
model_names: list[str] | None = None model_names: list[str] | None = None
class LLMProviderCreationRequest(LLMProvider): class LLMProviderUpdateRequest(LLMProvider):
api_key_set: bool api_key_set: bool
class LLMProviderUpdateRequest(LLMProvider): class LLMProviderCreationRequest(LLMProvider):
pass pass

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