8
8
from configs .middleware .storage .azure_blob_storage_config import AzureBlobStorageConfig
9
9
from configs .middleware .storage .google_cloud_storage_config import GoogleCloudStorageConfig
10
10
from configs .middleware .storage .tencent_cos_storage_config import TencentCloudCOSStorageConfig
11
- from configs .middleware .vdb .chroma_configs import ChromaConfigs
12
- from configs .middleware .vdb .milvus_configs import MilvusConfigs
13
- from configs .middleware .vdb .opensearch_configs import OpenSearchConfigs
14
- from configs .middleware .vdb .oracle_configs import OracleConfigs
15
- from configs .middleware .vdb .pgvector_configs import PGVectorConfigs
16
- from configs .middleware .vdb .pgvectors_configs import PGVectoRSConfigs
17
- from configs .middleware .vdb .qdrant_configs import QdrantConfigs
18
- from configs .middleware .vdb .relyt_configs import RelytConfigs
19
- from configs .middleware .vdb .tencent_vector_configs import TencentVectorDBConfigs
20
- from configs .middleware .vdb .tidb_vector_configs import TiDBVectorConfigs
21
- from configs .middleware .vdb .weaviate_configs import WeaviateConfigs
22
-
23
-
24
- class StorageConfigs (BaseModel ):
11
+ from configs .middleware .vdb .chroma_config import ChromaConfig
12
+ from configs .middleware .vdb .milvus_config import MilvusConfig
13
+ from configs .middleware .vdb .opensearch_config import OpenSearchConfig
14
+ from configs .middleware .vdb .oracle_config import OracleConfig
15
+ from configs .middleware .vdb .pgvector_config import PGVectorConfig
16
+ from configs .middleware .vdb .pgvectors_config import PGVectoRSConfig
17
+ from configs .middleware .vdb .qdrant_config import QdrantConfig
18
+ from configs .middleware .vdb .relyt_config import RelytConfig
19
+ from configs .middleware .vdb .tencent_vector_config import TencentVectorDBConfig
20
+ from configs .middleware .vdb .tidb_vector_config import TiDBVectorConfig
21
+ from configs .middleware .vdb .weaviate_config import WeaviateConfig
22
+
23
+
24
+ class StorageConfig (BaseModel ):
25
25
STORAGE_TYPE : str = Field (
26
26
description = 'storage type,'
27
27
' default to `local`,'
@@ -35,21 +35,21 @@ class StorageConfigs(BaseModel):
35
35
)
36
36
37
37
38
- class VectorStoreConfigs (BaseModel ):
38
+ class VectorStoreConfig (BaseModel ):
39
39
VECTOR_STORE : Optional [str ] = Field (
40
40
description = 'vector store type' ,
41
41
default = None ,
42
42
)
43
43
44
44
45
- class KeywordStoreConfigs (BaseModel ):
45
+ class KeywordStoreConfig (BaseModel ):
46
46
KEYWORD_STORE : str = Field (
47
47
description = 'keyword store type' ,
48
48
default = 'jieba' ,
49
49
)
50
50
51
51
52
- class DatabaseConfigs :
52
+ class DatabaseConfig :
53
53
DB_HOST : str = Field (
54
54
description = 'db host' ,
55
55
default = 'localhost' ,
@@ -130,7 +130,7 @@ def SQLALCHEMY_ENGINE_OPTIONS(self) -> dict[str, Any]:
130
130
}
131
131
132
132
133
- class CeleryConfigs ( DatabaseConfigs ):
133
+ class CeleryConfig ( DatabaseConfig ):
134
134
CELERY_BACKEND : str = Field (
135
135
description = 'Celery backend, available values are `database`, `redis`' ,
136
136
default = 'database' ,
@@ -155,31 +155,31 @@ def BROKER_USE_SSL(self) -> bool:
155
155
156
156
class MiddlewareConfig (
157
157
# place the configs in alphabet order
158
- CeleryConfigs ,
159
- DatabaseConfigs ,
160
- KeywordStoreConfigs ,
158
+ CeleryConfig ,
159
+ DatabaseConfig ,
160
+ KeywordStoreConfig ,
161
161
RedisConfig ,
162
162
163
163
# configs of storage and storage providers
164
- StorageConfigs ,
164
+ StorageConfig ,
165
165
AliyunOSSStorageConfig ,
166
166
AzureBlobStorageConfig ,
167
167
GoogleCloudStorageConfig ,
168
168
TencentCloudCOSStorageConfig ,
169
169
S3StorageConfig ,
170
170
171
171
# configs of vdb and vdb providers
172
- VectorStoreConfigs ,
173
- ChromaConfigs ,
174
- MilvusConfigs ,
175
- OpenSearchConfigs ,
176
- OracleConfigs ,
177
- PGVectorConfigs ,
178
- PGVectoRSConfigs ,
179
- QdrantConfigs ,
180
- RelytConfigs ,
181
- TencentVectorDBConfigs ,
182
- TiDBVectorConfigs ,
183
- WeaviateConfigs ,
172
+ VectorStoreConfig ,
173
+ ChromaConfig ,
174
+ MilvusConfig ,
175
+ OpenSearchConfig ,
176
+ OracleConfig ,
177
+ PGVectorConfig ,
178
+ PGVectoRSConfig ,
179
+ QdrantConfig ,
180
+ RelytConfig ,
181
+ TencentVectorDBConfig ,
182
+ TiDBVectorConfig ,
183
+ WeaviateConfig ,
184
184
):
185
185
pass
0 commit comments