mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-07-30 23:28:35 +00:00
chore:fmt
This commit is contained in:
parent
d095091525
commit
4f2903a6b7
@ -7,11 +7,11 @@ import os
|
|||||||
from dataclasses import dataclass, field
|
from dataclasses import dataclass, field
|
||||||
from typing import List
|
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.base import GraphStoreBase, GraphStoreConfig
|
||||||
from dbgpt.storage.graph_store.graph import GraphElemType
|
from dbgpt.storage.graph_store.graph import GraphElemType
|
||||||
from dbgpt_ext.datasource.conn_tugraph import TuGraphConnector
|
|
||||||
from dbgpt.util.i18n_utils import _
|
from dbgpt.util.i18n_utils import _
|
||||||
|
from dbgpt_ext.datasource.conn_tugraph import TuGraphConnector
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
@ -172,7 +172,7 @@ logger = logging.getLogger(__name__)
|
|||||||
optional=True,
|
optional=True,
|
||||||
default=None,
|
default=None,
|
||||||
),
|
),
|
||||||
]
|
],
|
||||||
)
|
)
|
||||||
class CommunitySummaryKnowledgeGraph(BuiltinKnowledgeGraph):
|
class CommunitySummaryKnowledgeGraph(BuiltinKnowledgeGraph):
|
||||||
"""Community summary knowledge graph class."""
|
"""Community summary knowledge graph class."""
|
||||||
@ -304,7 +304,6 @@ class CommunitySummaryKnowledgeGraph(BuiltinKnowledgeGraph):
|
|||||||
"""Get the knowledge graph config."""
|
"""Get the knowledge graph config."""
|
||||||
return self._config
|
return self._config
|
||||||
|
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def embeddings(self) -> Embeddings:
|
def embeddings(self) -> Embeddings:
|
||||||
"""Get the knowledge graph config."""
|
"""Get the knowledge graph config."""
|
||||||
|
@ -3,7 +3,7 @@
|
|||||||
import asyncio
|
import asyncio
|
||||||
import logging
|
import logging
|
||||||
import os
|
import os
|
||||||
from typing import List, Optional, Any
|
from typing import Any, List, Optional
|
||||||
|
|
||||||
from dbgpt.core import Chunk, Embeddings, LLMClient
|
from dbgpt.core import Chunk, Embeddings, LLMClient
|
||||||
from dbgpt.core.awel.flow import Parameter, ResourceCategory, register_resource
|
from dbgpt.core.awel.flow import Parameter, ResourceCategory, register_resource
|
||||||
@ -147,7 +147,6 @@ GRAPH_PARAMETERS = [
|
|||||||
optional=True,
|
optional=True,
|
||||||
default=None,
|
default=None,
|
||||||
),
|
),
|
||||||
|
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
class BuiltinKnowledgeGraph(KnowledgeGraphBase):
|
class BuiltinKnowledgeGraph(KnowledgeGraphBase):
|
||||||
|
@ -12,8 +12,9 @@ from dbgpt.core import Chunk, Embeddings
|
|||||||
from dbgpt.core.awel.flow import Parameter, ResourceCategory, register_resource
|
from dbgpt.core.awel.flow import Parameter, ResourceCategory, register_resource
|
||||||
from dbgpt.storage.vector_store.base import (
|
from dbgpt.storage.vector_store.base import (
|
||||||
_COMMON_PARAMETERS,
|
_COMMON_PARAMETERS,
|
||||||
|
_VECTOR_STORE_COMMON_PARAMETERS,
|
||||||
VectorStoreBase,
|
VectorStoreBase,
|
||||||
VectorStoreConfig, _VECTOR_STORE_COMMON_PARAMETERS,
|
VectorStoreConfig,
|
||||||
)
|
)
|
||||||
from dbgpt.storage.vector_store.filters import FilterOperator, MetadataFilters
|
from dbgpt.storage.vector_store.filters import FilterOperator, MetadataFilters
|
||||||
from dbgpt.util import string_utils
|
from dbgpt.util import string_utils
|
||||||
|
@ -11,8 +11,9 @@ from dbgpt.core import Chunk, Embeddings
|
|||||||
from dbgpt.core.awel.flow import Parameter, ResourceCategory, register_resource
|
from dbgpt.core.awel.flow import Parameter, ResourceCategory, register_resource
|
||||||
from dbgpt.storage.vector_store.base import (
|
from dbgpt.storage.vector_store.base import (
|
||||||
_COMMON_PARAMETERS,
|
_COMMON_PARAMETERS,
|
||||||
|
_VECTOR_STORE_COMMON_PARAMETERS,
|
||||||
VectorStoreBase,
|
VectorStoreBase,
|
||||||
VectorStoreConfig, _VECTOR_STORE_COMMON_PARAMETERS,
|
VectorStoreConfig,
|
||||||
)
|
)
|
||||||
from dbgpt.storage.vector_store.filters import MetadataFilters
|
from dbgpt.storage.vector_store.filters import MetadataFilters
|
||||||
from dbgpt.util import string_utils
|
from dbgpt.util import string_utils
|
||||||
|
@ -12,8 +12,9 @@ from dbgpt.core import Chunk, Embeddings
|
|||||||
from dbgpt.core.awel.flow import Parameter, ResourceCategory, register_resource
|
from dbgpt.core.awel.flow import Parameter, ResourceCategory, register_resource
|
||||||
from dbgpt.storage.vector_store.base import (
|
from dbgpt.storage.vector_store.base import (
|
||||||
_COMMON_PARAMETERS,
|
_COMMON_PARAMETERS,
|
||||||
|
_VECTOR_STORE_COMMON_PARAMETERS,
|
||||||
VectorStoreBase,
|
VectorStoreBase,
|
||||||
VectorStoreConfig, _VECTOR_STORE_COMMON_PARAMETERS,
|
VectorStoreConfig,
|
||||||
)
|
)
|
||||||
from dbgpt.storage.vector_store.filters import FilterOperator, MetadataFilters
|
from dbgpt.storage.vector_store.filters import FilterOperator, MetadataFilters
|
||||||
from dbgpt.util import string_utils
|
from dbgpt.util import string_utils
|
||||||
|
@ -16,8 +16,9 @@ from dbgpt.core import Chunk, Embeddings
|
|||||||
from dbgpt.core.awel.flow import Parameter, ResourceCategory, register_resource
|
from dbgpt.core.awel.flow import Parameter, ResourceCategory, register_resource
|
||||||
from dbgpt.storage.vector_store.base import (
|
from dbgpt.storage.vector_store.base import (
|
||||||
_COMMON_PARAMETERS,
|
_COMMON_PARAMETERS,
|
||||||
|
_VECTOR_STORE_COMMON_PARAMETERS,
|
||||||
VectorStoreBase,
|
VectorStoreBase,
|
||||||
VectorStoreConfig, _VECTOR_STORE_COMMON_PARAMETERS,
|
VectorStoreConfig,
|
||||||
)
|
)
|
||||||
from dbgpt.storage.vector_store.filters import FilterOperator, MetadataFilters
|
from dbgpt.storage.vector_store.filters import FilterOperator, MetadataFilters
|
||||||
from dbgpt.util.i18n_utils import _
|
from dbgpt.util.i18n_utils import _
|
||||||
|
@ -8,8 +8,9 @@ from dbgpt.core import Chunk, Embeddings
|
|||||||
from dbgpt.core.awel.flow import Parameter, ResourceCategory, register_resource
|
from dbgpt.core.awel.flow import Parameter, ResourceCategory, register_resource
|
||||||
from dbgpt.storage.vector_store.base import (
|
from dbgpt.storage.vector_store.base import (
|
||||||
_COMMON_PARAMETERS,
|
_COMMON_PARAMETERS,
|
||||||
|
_VECTOR_STORE_COMMON_PARAMETERS,
|
||||||
VectorStoreBase,
|
VectorStoreBase,
|
||||||
VectorStoreConfig, _VECTOR_STORE_COMMON_PARAMETERS,
|
VectorStoreConfig,
|
||||||
)
|
)
|
||||||
from dbgpt.storage.vector_store.filters import MetadataFilters
|
from dbgpt.storage.vector_store.filters import MetadataFilters
|
||||||
from dbgpt.util.i18n_utils import _
|
from dbgpt.util.i18n_utils import _
|
||||||
|
@ -9,8 +9,9 @@ from dbgpt.core import Chunk, Embeddings
|
|||||||
from dbgpt.core.awel.flow import Parameter, ResourceCategory, register_resource
|
from dbgpt.core.awel.flow import Parameter, ResourceCategory, register_resource
|
||||||
from dbgpt.storage.vector_store.base import (
|
from dbgpt.storage.vector_store.base import (
|
||||||
_COMMON_PARAMETERS,
|
_COMMON_PARAMETERS,
|
||||||
|
_VECTOR_STORE_COMMON_PARAMETERS,
|
||||||
VectorStoreBase,
|
VectorStoreBase,
|
||||||
VectorStoreConfig, _VECTOR_STORE_COMMON_PARAMETERS,
|
VectorStoreConfig,
|
||||||
)
|
)
|
||||||
from dbgpt.storage.vector_store.filters import MetadataFilters
|
from dbgpt.storage.vector_store.filters import MetadataFilters
|
||||||
from dbgpt.util.i18n_utils import _
|
from dbgpt.util.i18n_utils import _
|
||||||
|
Loading…
Reference in New Issue
Block a user