jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

22
api/configs/middleware/__init__.py
View file

@ -3,6 +3,16 @@ from typing import Optional
from pydantic import BaseModel, Field
from configs.middleware.redis_configs import RedisConfigs
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.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):
@ -38,6 +48,18 @@ class MiddlewareConfigs(
KeywordStoreConfigs,
RedisConfigs,
StorageConfigs,
# configs of vdb and vdb providers
VectorStoreConfigs,
ChromaConfigs,
MilvusConfigs,
OpenSearchConfigs,
PGVectorConfigs,
PGVectoRSConfigs,
QdrantConfigs,
RelytConfigs,
TencentVectorDBConfigs,
TiDBVectorConfigs,
WeaviateConfigs,
):
pass