diff --git a/packages/dbgpt-ext/src/dbgpt_ext/storage/graph_store/tugraph_store.py b/packages/dbgpt-ext/src/dbgpt_ext/storage/graph_store/tugraph_store.py index 7560c3250..961b88ed0 100644 --- a/packages/dbgpt-ext/src/dbgpt_ext/storage/graph_store/tugraph_store.py +++ b/packages/dbgpt-ext/src/dbgpt_ext/storage/graph_store/tugraph_store.py @@ -7,11 +7,11 @@ import os from dataclasses import dataclass, field from typing import List -from dbgpt.core.awel.flow import register_resource, ResourceCategory, Parameter +from dbgpt.core.awel.flow import Parameter, ResourceCategory, register_resource from dbgpt.storage.graph_store.base import GraphStoreBase, GraphStoreConfig from dbgpt.storage.graph_store.graph import GraphElemType -from dbgpt_ext.datasource.conn_tugraph import TuGraphConnector from dbgpt.util.i18n_utils import _ +from dbgpt_ext.datasource.conn_tugraph import TuGraphConnector logger = logging.getLogger(__name__) diff --git a/packages/dbgpt-ext/src/dbgpt_ext/storage/knowledge_graph/community_summary.py b/packages/dbgpt-ext/src/dbgpt_ext/storage/knowledge_graph/community_summary.py index 1bd45bd95..6871a2d4d 100644 --- a/packages/dbgpt-ext/src/dbgpt_ext/storage/knowledge_graph/community_summary.py +++ b/packages/dbgpt-ext/src/dbgpt_ext/storage/knowledge_graph/community_summary.py @@ -172,7 +172,7 @@ logger = logging.getLogger(__name__) optional=True, default=None, ), - ] + ], ) class CommunitySummaryKnowledgeGraph(BuiltinKnowledgeGraph): """Community summary knowledge graph class.""" @@ -304,7 +304,6 @@ class CommunitySummaryKnowledgeGraph(BuiltinKnowledgeGraph): """Get the knowledge graph config.""" return self._config - @property def embeddings(self) -> Embeddings: """Get the knowledge graph config.""" diff --git a/packages/dbgpt-ext/src/dbgpt_ext/storage/knowledge_graph/knowledge_graph.py b/packages/dbgpt-ext/src/dbgpt_ext/storage/knowledge_graph/knowledge_graph.py index 3184af298..ad235c8e3 100644 --- a/packages/dbgpt-ext/src/dbgpt_ext/storage/knowledge_graph/knowledge_graph.py +++ b/packages/dbgpt-ext/src/dbgpt_ext/storage/knowledge_graph/knowledge_graph.py @@ -3,7 +3,7 @@ import asyncio import logging import os -from typing import List, Optional, Any +from typing import Any, List, Optional from dbgpt.core import Chunk, Embeddings, LLMClient from dbgpt.core.awel.flow import Parameter, ResourceCategory, register_resource @@ -147,7 +147,6 @@ GRAPH_PARAMETERS = [ optional=True, default=None, ), - ], ) class BuiltinKnowledgeGraph(KnowledgeGraphBase): diff --git a/packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/chroma_store.py b/packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/chroma_store.py index 6d2918cc7..a518e87be 100644 --- a/packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/chroma_store.py +++ b/packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/chroma_store.py @@ -12,8 +12,9 @@ from dbgpt.core import Chunk, Embeddings from dbgpt.core.awel.flow import Parameter, ResourceCategory, register_resource from dbgpt.storage.vector_store.base import ( _COMMON_PARAMETERS, + _VECTOR_STORE_COMMON_PARAMETERS, VectorStoreBase, - VectorStoreConfig, _VECTOR_STORE_COMMON_PARAMETERS, + VectorStoreConfig, ) from dbgpt.storage.vector_store.filters import FilterOperator, MetadataFilters from dbgpt.util import string_utils diff --git a/packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/elastic_store.py b/packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/elastic_store.py index aa8bd0559..cce3ea65e 100644 --- a/packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/elastic_store.py +++ b/packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/elastic_store.py @@ -11,8 +11,9 @@ from dbgpt.core import Chunk, Embeddings from dbgpt.core.awel.flow import Parameter, ResourceCategory, register_resource from dbgpt.storage.vector_store.base import ( _COMMON_PARAMETERS, + _VECTOR_STORE_COMMON_PARAMETERS, VectorStoreBase, - VectorStoreConfig, _VECTOR_STORE_COMMON_PARAMETERS, + VectorStoreConfig, ) from dbgpt.storage.vector_store.filters import MetadataFilters from dbgpt.util import string_utils diff --git a/packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/milvus_store.py b/packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/milvus_store.py index 48e045c3d..01203b4d9 100644 --- a/packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/milvus_store.py +++ b/packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/milvus_store.py @@ -12,8 +12,9 @@ from dbgpt.core import Chunk, Embeddings from dbgpt.core.awel.flow import Parameter, ResourceCategory, register_resource from dbgpt.storage.vector_store.base import ( _COMMON_PARAMETERS, + _VECTOR_STORE_COMMON_PARAMETERS, VectorStoreBase, - VectorStoreConfig, _VECTOR_STORE_COMMON_PARAMETERS, + VectorStoreConfig, ) from dbgpt.storage.vector_store.filters import FilterOperator, MetadataFilters from dbgpt.util import string_utils diff --git a/packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/oceanbase_store.py b/packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/oceanbase_store.py index c1400fbd8..4abedd4a0 100644 --- a/packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/oceanbase_store.py +++ b/packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/oceanbase_store.py @@ -16,8 +16,9 @@ from dbgpt.core import Chunk, Embeddings from dbgpt.core.awel.flow import Parameter, ResourceCategory, register_resource from dbgpt.storage.vector_store.base import ( _COMMON_PARAMETERS, + _VECTOR_STORE_COMMON_PARAMETERS, VectorStoreBase, - VectorStoreConfig, _VECTOR_STORE_COMMON_PARAMETERS, + VectorStoreConfig, ) from dbgpt.storage.vector_store.filters import FilterOperator, MetadataFilters from dbgpt.util.i18n_utils import _ diff --git a/packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/pgvector_store.py b/packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/pgvector_store.py index 22ec34eb7..01d266c44 100644 --- a/packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/pgvector_store.py +++ b/packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/pgvector_store.py @@ -8,8 +8,9 @@ from dbgpt.core import Chunk, Embeddings from dbgpt.core.awel.flow import Parameter, ResourceCategory, register_resource from dbgpt.storage.vector_store.base import ( _COMMON_PARAMETERS, + _VECTOR_STORE_COMMON_PARAMETERS, VectorStoreBase, - VectorStoreConfig, _VECTOR_STORE_COMMON_PARAMETERS, + VectorStoreConfig, ) from dbgpt.storage.vector_store.filters import MetadataFilters from dbgpt.util.i18n_utils import _ diff --git a/packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/weaviate_store.py b/packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/weaviate_store.py index 74c798d45..e120a8421 100644 --- a/packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/weaviate_store.py +++ b/packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/weaviate_store.py @@ -9,8 +9,9 @@ from dbgpt.core import Chunk, Embeddings from dbgpt.core.awel.flow import Parameter, ResourceCategory, register_resource from dbgpt.storage.vector_store.base import ( _COMMON_PARAMETERS, + _VECTOR_STORE_COMMON_PARAMETERS, VectorStoreBase, - VectorStoreConfig, _VECTOR_STORE_COMMON_PARAMETERS, + VectorStoreConfig, ) from dbgpt.storage.vector_store.filters import MetadataFilters from dbgpt.util.i18n_utils import _