Browse Source

chore: use singular style in middleware config class name (#5502)

tags/0.6.12
Bowen Liang 1 year ago
parent
commit
29ca6815ae
No account linked to committer's email address

+ 1
- 0
.gitignore View File

!.idea/vcs.xml !.idea/vcs.xml
!.idea/icon.png !.idea/icon.png
.ideaDataSources/ .ideaDataSources/
*.iml


api/.env api/.env
api/storage/* api/storage/*

+ 34
- 34
api/configs/middleware/__init__.py View File

from configs.middleware.storage.azure_blob_storage_config import AzureBlobStorageConfig from configs.middleware.storage.azure_blob_storage_config import AzureBlobStorageConfig
from configs.middleware.storage.google_cloud_storage_config import GoogleCloudStorageConfig from configs.middleware.storage.google_cloud_storage_config import GoogleCloudStorageConfig
from configs.middleware.storage.tencent_cos_storage_config import TencentCloudCOSStorageConfig 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( STORAGE_TYPE: str = Field(
description='storage type,' description='storage type,'
' default to `local`,' ' default to `local`,'
) )




class VectorStoreConfigs(BaseModel):
class VectorStoreConfig(BaseModel):
VECTOR_STORE: Optional[str] = Field( VECTOR_STORE: Optional[str] = Field(
description='vector store type', description='vector store type',
default=None, default=None,
) )




class KeywordStoreConfigs(BaseModel):
class KeywordStoreConfig(BaseModel):
KEYWORD_STORE: str = Field( KEYWORD_STORE: str = Field(
description='keyword store type', description='keyword store type',
default='jieba', default='jieba',
) )




class DatabaseConfigs:
class DatabaseConfig:
DB_HOST: str = Field( DB_HOST: str = Field(
description='db host', description='db host',
default='localhost', default='localhost',
} }




class CeleryConfigs(DatabaseConfigs):
class CeleryConfig(DatabaseConfig):
CELERY_BACKEND: str = Field( CELERY_BACKEND: str = Field(
description='Celery backend, available values are `database`, `redis`', description='Celery backend, available values are `database`, `redis`',
default='database', default='database',


class MiddlewareConfig( class MiddlewareConfig(
# place the configs in alphabet order # place the configs in alphabet order
CeleryConfigs,
DatabaseConfigs,
KeywordStoreConfigs,
CeleryConfig,
DatabaseConfig,
KeywordStoreConfig,
RedisConfig, RedisConfig,


# configs of storage and storage providers # configs of storage and storage providers
StorageConfigs,
StorageConfig,
AliyunOSSStorageConfig, AliyunOSSStorageConfig,
AzureBlobStorageConfig, AzureBlobStorageConfig,
GoogleCloudStorageConfig, GoogleCloudStorageConfig,
S3StorageConfig, S3StorageConfig,


# configs of vdb and vdb providers # 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 pass

api/configs/middleware/vdb/chroma_configs.py → api/configs/middleware/vdb/chroma_config.py View File

from pydantic import BaseModel, Field, PositiveInt from pydantic import BaseModel, Field, PositiveInt




class ChromaConfigs(BaseModel):
class ChromaConfig(BaseModel):
""" """
Chroma configs Chroma configs
""" """

api/configs/middleware/vdb/milvus_configs.py → api/configs/middleware/vdb/milvus_config.py View File

from pydantic import BaseModel, Field, PositiveInt from pydantic import BaseModel, Field, PositiveInt




class MilvusConfigs(BaseModel):
class MilvusConfig(BaseModel):
""" """
Milvus configs Milvus configs
""" """

api/configs/middleware/vdb/opensearch_configs.py → api/configs/middleware/vdb/opensearch_config.py View File

from pydantic import BaseModel, Field, PositiveInt from pydantic import BaseModel, Field, PositiveInt




class OpenSearchConfigs(BaseModel):
class OpenSearchConfig(BaseModel):
""" """
OpenSearch configs OpenSearch configs
""" """

api/configs/middleware/vdb/oracle_configs.py → api/configs/middleware/vdb/oracle_config.py View File

from pydantic import BaseModel, Field, PositiveInt from pydantic import BaseModel, Field, PositiveInt




class OracleConfigs(BaseModel):
class OracleConfig(BaseModel):
""" """
ORACLE configs ORACLE configs
""" """

api/configs/middleware/vdb/pgvector_configs.py → api/configs/middleware/vdb/pgvector_config.py View File

from pydantic import BaseModel, Field, PositiveInt from pydantic import BaseModel, Field, PositiveInt




class PGVectorConfigs(BaseModel):
class PGVectorConfig(BaseModel):
""" """
PGVector configs PGVector configs
""" """

api/configs/middleware/vdb/pgvectors_configs.py → api/configs/middleware/vdb/pgvectors_config.py View File

from pydantic import BaseModel, Field, PositiveInt from pydantic import BaseModel, Field, PositiveInt




class PGVectoRSConfigs(BaseModel):
class PGVectoRSConfig(BaseModel):
""" """
PGVectoRS configs PGVectoRS configs
""" """

api/configs/middleware/vdb/qdrant_configs.py → api/configs/middleware/vdb/qdrant_config.py View File

from pydantic import BaseModel, Field, NonNegativeInt, PositiveInt from pydantic import BaseModel, Field, NonNegativeInt, PositiveInt




class QdrantConfigs(BaseModel):
class QdrantConfig(BaseModel):
""" """
Qdrant configs Qdrant configs
""" """

api/configs/middleware/vdb/relyt_configs.py → api/configs/middleware/vdb/relyt_config.py View File

from pydantic import BaseModel, Field, PositiveInt from pydantic import BaseModel, Field, PositiveInt




class RelytConfigs(BaseModel):
class RelytConfig(BaseModel):
""" """
Relyt configs Relyt configs
""" """

api/configs/middleware/vdb/tencent_vector_configs.py → api/configs/middleware/vdb/tencent_vector_config.py View File

from pydantic import BaseModel, Field, PositiveInt from pydantic import BaseModel, Field, PositiveInt




class TencentVectorDBConfigs(BaseModel):
class TencentVectorDBConfig(BaseModel):
""" """
Tencent Vector configs Tencent Vector configs
""" """

api/configs/middleware/vdb/tidb_vector_configs.py → api/configs/middleware/vdb/tidb_vector_config.py View File

from pydantic import BaseModel, Field, PositiveInt from pydantic import BaseModel, Field, PositiveInt




class TiDBVectorConfigs(BaseModel):
class TiDBVectorConfig(BaseModel):
""" """
TiDB Vector configs TiDB Vector configs
""" """

api/configs/middleware/vdb/weaviate_configs.py → api/configs/middleware/vdb/weaviate_config.py View File

from pydantic import BaseModel, Field, PositiveInt from pydantic import BaseModel, Field, PositiveInt




class WeaviateConfigs(BaseModel):
class WeaviateConfig(BaseModel):
""" """
Weaviate configs Weaviate configs
""" """

Loading…
Cancel
Save