refactor: extract vdb configs into pydantic-setting based dify configs (#5426)

This commit is contained in:
Bowen Liang 2024-06-20 16:24:10 +08:00 • committed by GitHub
commit 65d34ebb96
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
12 changed files with 377 additions and 84 deletions

View file

@ -0,0 +1,29 @@
from typing import Optional
from pydantic import BaseModel, Field, PositiveInt
class WeaviateConfigs(BaseModel):
"""
Weaviate configs
"""
WEAVIATE_ENDPOINT: Optional[str] = Field(
description='Weaviate endpoint URL',
default=None,
)
WEAVIATE_API_KEY: Optional[str] = Field(
description='Weaviate API key',
default=None,
)
WEAVIATE_GRPC_ENABLED: bool = Field(
description='whether to enable gRPC for Weaviate connection',
default=True,
)
WEAVIATE_BATCH_SIZE: PositiveInt = Field(
description='Weaviate batch size',
default=100,
)