admin/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

feat: VECTOR_STORE supports seekdb (#29658)

Browse Source
This commit is contained in:
longbingljw
2025-12-16 11:35:04 +08:00
committed by GitHub
parent a232da564a
commit 4cc6652424
5 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
api/configs/middleware/__init__.py
View File

@@ -107,7 +107,7 @@ class KeywordStoreConfig(BaseSettings):
class DatabaseConfig(BaseSettings): class DatabaseConfig(BaseSettings):
# Database type selector # Database type selector
DB_TYPE: Literal["postgresql", "mysql", "oceanbase"] = Field( DB_TYPE: Literal["postgresql", "mysql", "oceanbase", "seekdb"] = Field(
description="Database type to use. OceanBase is MySQL-compatible.", description="Database type to use. OceanBase is MySQL-compatible.",
default="postgresql", default="postgresql",
) )

1
api/controllers/console/datasets/datasets.py
View File

@@ -223,6 +223,7 @@ def _get_retrieval_methods_by_vector_type(vector_type: str | None, is_mock: bool
VectorType.COUCHBASE, VectorType.COUCHBASE,
VectorType.OPENGAUSS, VectorType.OPENGAUSS,
VectorType.OCEANBASE, VectorType.OCEANBASE,
VectorType.SEEKDB,
VectorType.TABLESTORE, VectorType.TABLESTORE,
VectorType.HUAWEI_CLOUD, VectorType.HUAWEI_CLOUD,
VectorType.TENCENT, VectorType.TENCENT,

2
api/core/rag/datasource/vdb/vector_factory.py
View File

@@ -163,7 +163,7 @@ class Vector:
from core.rag.datasource.vdb.lindorm.lindorm_vector import LindormVectorStoreFactory from core.rag.datasource.vdb.lindorm.lindorm_vector import LindormVectorStoreFactory
return LindormVectorStoreFactory return LindormVectorStoreFactory
case VectorType.OCEANBASE: case VectorType.OCEANBASE | VectorType.SEEKDB:
from core.rag.datasource.vdb.oceanbase.oceanbase_vector import OceanBaseVectorFactory from core.rag.datasource.vdb.oceanbase.oceanbase_vector import OceanBaseVectorFactory
return OceanBaseVectorFactory return OceanBaseVectorFactory

1
api/core/rag/datasource/vdb/vector_type.py
View File

@@ -27,6 +27,7 @@ class VectorType(StrEnum):
UPSTASH = "upstash" UPSTASH = "upstash"
TIDB_ON_QDRANT = "tidb_on_qdrant" TIDB_ON_QDRANT = "tidb_on_qdrant"
OCEANBASE = "oceanbase" OCEANBASE = "oceanbase"
SEEKDB = "seekdb"
OPENGAUSS = "opengauss" OPENGAUSS = "opengauss"
TABLESTORE = "tablestore" TABLESTORE = "tablestore"
HUAWEI_CLOUD = "huawei_cloud" HUAWEI_CLOUD = "huawei_cloud"

2
api/libs/helper.py
View File

@@ -184,7 +184,7 @@ def timezone(timezone_string):
def convert_datetime_to_date(field, target_timezone: str = ":tz"): def convert_datetime_to_date(field, target_timezone: str = ":tz"):
if dify_config.DB_TYPE == "postgresql": if dify_config.DB_TYPE == "postgresql":
return f"DATE(DATE_TRUNC('day', {field} AT TIME ZONE 'UTC' AT TIME ZONE {target_timezone}))" return f"DATE(DATE_TRUNC('day', {field} AT TIME ZONE 'UTC' AT TIME ZONE {target_timezone}))"
elif dify_config.DB_TYPE == "mysql": elif dify_config.DB_TYPE in ["mysql", "oceanbase", "seekdb"]:
return f"DATE(CONVERT_TZ({field}, 'UTC', {target_timezone}))" return f"DATE(CONVERT_TZ({field}, 'UTC', {target_timezone}))"
else: else:
raise NotImplementedError(f"Unsupported database type: {dify_config.DB_TYPE}") raise NotImplementedError(f"Unsupported database type: {dify_config.DB_TYPE}")