|
|
|
@@ -8,20 +8,20 @@ from configs.middleware.storage.amazon_s3_storage_config import S3StorageConfig |
|
|
|
from configs.middleware.storage.azure_blob_storage_config import AzureBlobStorageConfig |
|
|
|
from configs.middleware.storage.google_cloud_storage_config import GoogleCloudStorageConfig |
|
|
|
from configs.middleware.storage.tencent_cos_storage_config import TencentCloudCOSStorageConfig |
|
|
|
from configs.middleware.vdb.chroma_configs import ChromaConfigs |
|
|
|
from configs.middleware.vdb.milvus_configs import MilvusConfigs |
|
|
|
from configs.middleware.vdb.opensearch_configs import OpenSearchConfigs |
|
|
|
from configs.middleware.vdb.oracle_configs import OracleConfigs |
|
|
|
from configs.middleware.vdb.pgvector_configs import PGVectorConfigs |
|
|
|
from configs.middleware.vdb.pgvectors_configs import PGVectoRSConfigs |
|
|
|
from configs.middleware.vdb.qdrant_configs import QdrantConfigs |
|
|
|
from configs.middleware.vdb.relyt_configs import RelytConfigs |
|
|
|
from configs.middleware.vdb.tencent_vector_configs import TencentVectorDBConfigs |
|
|
|
from configs.middleware.vdb.tidb_vector_configs import TiDBVectorConfigs |
|
|
|
from configs.middleware.vdb.weaviate_configs import WeaviateConfigs |
|
|
|
|
|
|
|
|
|
|
|
class StorageConfigs(BaseModel): |
|
|
|
from configs.middleware.vdb.chroma_config import ChromaConfig |
|
|
|
from configs.middleware.vdb.milvus_config import MilvusConfig |
|
|
|
from configs.middleware.vdb.opensearch_config import OpenSearchConfig |
|
|
|
from configs.middleware.vdb.oracle_config import OracleConfig |
|
|
|
from configs.middleware.vdb.pgvector_config import PGVectorConfig |
|
|
|
from configs.middleware.vdb.pgvectors_config import PGVectoRSConfig |
|
|
|
from configs.middleware.vdb.qdrant_config import QdrantConfig |
|
|
|
from configs.middleware.vdb.relyt_config import RelytConfig |
|
|
|
from configs.middleware.vdb.tencent_vector_config import TencentVectorDBConfig |
|
|
|
from configs.middleware.vdb.tidb_vector_config import TiDBVectorConfig |
|
|
|
from configs.middleware.vdb.weaviate_config import WeaviateConfig |
|
|
|
|
|
|
|
|
|
|
|
class StorageConfig(BaseModel): |
|
|
|
STORAGE_TYPE: str = Field( |
|
|
|
description='storage type,' |
|
|
|
' default to `local`,' |
|
|
|
@@ -35,21 +35,21 @@ class StorageConfigs(BaseModel): |
|
|
|
) |
|
|
|
|
|
|
|
|
|
|
|
class VectorStoreConfigs(BaseModel): |
|
|
|
class VectorStoreConfig(BaseModel): |
|
|
|
VECTOR_STORE: Optional[str] = Field( |
|
|
|
description='vector store type', |
|
|
|
default=None, |
|
|
|
) |
|
|
|
|
|
|
|
|
|
|
|
class KeywordStoreConfigs(BaseModel): |
|
|
|
class KeywordStoreConfig(BaseModel): |
|
|
|
KEYWORD_STORE: str = Field( |
|
|
|
description='keyword store type', |
|
|
|
default='jieba', |
|
|
|
) |
|
|
|
|
|
|
|
|
|
|
|
class DatabaseConfigs: |
|
|
|
class DatabaseConfig: |
|
|
|
DB_HOST: str = Field( |
|
|
|
description='db host', |
|
|
|
default='localhost', |
|
|
|
@@ -130,7 +130,7 @@ class DatabaseConfigs: |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
class CeleryConfigs(DatabaseConfigs): |
|
|
|
class CeleryConfig(DatabaseConfig): |
|
|
|
CELERY_BACKEND: str = Field( |
|
|
|
description='Celery backend, available values are `database`, `redis`', |
|
|
|
default='database', |
|
|
|
@@ -155,13 +155,13 @@ class CeleryConfigs(DatabaseConfigs): |
|
|
|
|
|
|
|
class MiddlewareConfig( |
|
|
|
# place the configs in alphabet order |
|
|
|
CeleryConfigs, |
|
|
|
DatabaseConfigs, |
|
|
|
KeywordStoreConfigs, |
|
|
|
CeleryConfig, |
|
|
|
DatabaseConfig, |
|
|
|
KeywordStoreConfig, |
|
|
|
RedisConfig, |
|
|
|
|
|
|
|
# configs of storage and storage providers |
|
|
|
StorageConfigs, |
|
|
|
StorageConfig, |
|
|
|
AliyunOSSStorageConfig, |
|
|
|
AzureBlobStorageConfig, |
|
|
|
GoogleCloudStorageConfig, |
|
|
|
@@ -169,17 +169,17 @@ class MiddlewareConfig( |
|
|
|
S3StorageConfig, |
|
|
|
|
|
|
|
# configs of vdb and vdb providers |
|
|
|
VectorStoreConfigs, |
|
|
|
ChromaConfigs, |
|
|
|
MilvusConfigs, |
|
|
|
OpenSearchConfigs, |
|
|
|
OracleConfigs, |
|
|
|
PGVectorConfigs, |
|
|
|
PGVectoRSConfigs, |
|
|
|
QdrantConfigs, |
|
|
|
RelytConfigs, |
|
|
|
TencentVectorDBConfigs, |
|
|
|
TiDBVectorConfigs, |
|
|
|
WeaviateConfigs, |
|
|
|
VectorStoreConfig, |
|
|
|
ChromaConfig, |
|
|
|
MilvusConfig, |
|
|
|
OpenSearchConfig, |
|
|
|
OracleConfig, |
|
|
|
PGVectorConfig, |
|
|
|
PGVectoRSConfig, |
|
|
|
QdrantConfig, |
|
|
|
RelytConfig, |
|
|
|
TencentVectorDBConfig, |
|
|
|
TiDBVectorConfig, |
|
|
|
WeaviateConfig, |
|
|
|
): |
|
|
|
pass |