mirror of https://github.com/langgenius/dify.git
69 lines
2.1 KiB
Python
69 lines
2.1 KiB
Python
|
from holo_search_sdk.types import BaseQuantizationType, DistanceType, TokenizerType
|
||
|
|
from pydantic import Field
|
||
|
|
from pydantic_settings import BaseSettings
|
||
|
|
|
||
|
|
|
||
|
|
class HologresConfig(BaseSettings):
|
||
|
|
"""
|
||
|
|
Configuration settings for Hologres vector database.
|
||
|
|
|
||
|
|
Hologres is compatible with PostgreSQL protocol.
|
||
|
|
access_key_id is used as the PostgreSQL username,
|
||
|
|
and access_key_secret is used as the PostgreSQL password.
|
||
|
|
"""
|
||
|
|
|
||
|
|
HOLOGRES_HOST: str | None = Field(
|
||
|
|
description="Hostname or IP address of the Hologres instance.",
|
||
|
|
default=None,
|
||
|
|
)
|
||
|
|
|
||
|
|
HOLOGRES_PORT: int = Field(
|
||
|
|
description="Port number for connecting to the Hologres instance.",
|
||
|
|
default=80,
|
||
|
|
)
|
||
|
|
|
||
|
|
HOLOGRES_DATABASE: str | None = Field(
|
||
|
|
description="Name of the Hologres database to connect to.",
|
||
|
|
default=None,
|
||
|
|
)
|
||
|
|
|
||
|
|
HOLOGRES_ACCESS_KEY_ID: str | None = Field(
|
||
|
|
description="Alibaba Cloud AccessKey ID, also used as the PostgreSQL username.",
|
||
|
|
default=None,
|
||
|
|
)
|
||
|
|
|
||
|
|
HOLOGRES_ACCESS_KEY_SECRET: str | None = Field(
|
||
|
|
description="Alibaba Cloud AccessKey Secret, also used as the PostgreSQL password.",
|
||
|
|
default=None,
|
||
|
|
)
|
||
|
|
|
||
|
|
HOLOGRES_SCHEMA: str = Field(
|
||
|
|
description="Schema name in the Hologres database.",
|
||
|
|
default="public",
|
||
|
|
)
|
||
|
|
|
||
|
|
HOLOGRES_TOKENIZER: TokenizerType = Field(
|
||
|
|
description="Tokenizer for full-text search index (e.g., 'jieba', 'ik', 'standard', 'simple').",
|
||
|
|
default="jieba",
|
||
|
|
)
|
||
|
|
|
||
|
|
HOLOGRES_DISTANCE_METHOD: DistanceType = Field(
|
||
|
|
description="Distance method for vector index (e.g., 'Cosine', 'Euclidean', 'InnerProduct').",
|
||
|
|
default="Cosine",
|
||
|
|
)
|
||
|
|
|
||
|
|
HOLOGRES_BASE_QUANTIZATION_TYPE: BaseQuantizationType = Field(
|
||
|
|
description="Base quantization type for vector index (e.g., 'rabitq', 'sq8', 'fp16', 'fp32').",
|
||
|
|
default="rabitq",
|
||
|
|
)
|
||
|
|
|
||
|
|
HOLOGRES_MAX_DEGREE: int = Field(
|
||
|
|
description="Max degree (M) parameter for HNSW vector index.",
|
||
|
|
default=64,
|
||
|
|
)
|
||
|
|
|
||
|
|
HOLOGRES_EF_CONSTRUCTION: int = Field(
|
||
|
|
description="ef_construction parameter for HNSW vector index.",
|
||
|
|
default=400,
|
||
|
|
)
|