tjdify/api/configs/middleware/vdb/weaviate_config.py

from typing import Optional

from pydantic import Field, PositiveInt
from pydantic_settings import BaseSettings


class WeaviateConfig(BaseSettings):
    """
    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,
    )