mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-07-23 20:26:15 +00:00
fix:rag workflow error fix
This commit is contained in:
parent
eccbe2ba94
commit
d095091525
@ -22,7 +22,6 @@ from dbgpt_ext.storage.knowledge_graph.community.community_store import Communit
|
|||||||
from dbgpt_ext.storage.knowledge_graph.knowledge_graph import (
|
from dbgpt_ext.storage.knowledge_graph.knowledge_graph import (
|
||||||
GRAPH_PARAMETERS,
|
GRAPH_PARAMETERS,
|
||||||
BuiltinKnowledgeGraph,
|
BuiltinKnowledgeGraph,
|
||||||
BuiltinKnowledgeGraphConfig,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
@ -146,14 +145,34 @@ logger = logging.getLogger(__name__)
|
|||||||
description=_("Community Summary Knowledge Graph."),
|
description=_("Community Summary Knowledge Graph."),
|
||||||
parameters=[
|
parameters=[
|
||||||
Parameter.build_from(
|
Parameter.build_from(
|
||||||
_("Community Summary Knowledge Graph Config."),
|
_("Graph Store Config"),
|
||||||
"config",
|
"config",
|
||||||
BuiltinKnowledgeGraphConfig,
|
GraphStoreConfig,
|
||||||
description=_("Community Summary Knowledge Graph Config."),
|
description=_("graph store config."),
|
||||||
|
),
|
||||||
|
Parameter.build_from(
|
||||||
|
_("Graph Store Name"),
|
||||||
|
"name",
|
||||||
|
str,
|
||||||
|
optional=True,
|
||||||
|
default="dbgpt",
|
||||||
|
description=_("Graph Store Name"),
|
||||||
|
),
|
||||||
|
Parameter.build_from(
|
||||||
|
_("LLM Client"),
|
||||||
|
"llm_client",
|
||||||
|
LLMClient,
|
||||||
|
description=_("llm client for extract graph triplets."),
|
||||||
|
),
|
||||||
|
Parameter.build_from(
|
||||||
|
_("LLM Model Name"),
|
||||||
|
"llm_model",
|
||||||
|
str,
|
||||||
|
description=_("kg extract llm model 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."""
|
||||||
|
@ -3,7 +3,6 @@
|
|||||||
import asyncio
|
import asyncio
|
||||||
import logging
|
import logging
|
||||||
import os
|
import os
|
||||||
from dataclasses import dataclass, field
|
|
||||||
from typing import List, Optional, Any
|
from typing import List, Optional, Any
|
||||||
|
|
||||||
from dbgpt.core import Chunk, Embeddings, LLMClient
|
from dbgpt.core import Chunk, Embeddings, LLMClient
|
||||||
@ -11,7 +10,7 @@ from dbgpt.core.awel.flow import Parameter, ResourceCategory, register_resource
|
|||||||
from dbgpt.rag.transformer.keyword_extractor import KeywordExtractor
|
from dbgpt.rag.transformer.keyword_extractor import KeywordExtractor
|
||||||
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 Graph
|
from dbgpt.storage.graph_store.graph import Graph
|
||||||
from dbgpt.storage.knowledge_graph.base import KnowledgeGraphBase, KnowledgeGraphConfig
|
from dbgpt.storage.knowledge_graph.base import KnowledgeGraphBase
|
||||||
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 _
|
||||||
from dbgpt_ext.rag.transformer.triplet_extractor import TripletExtractor
|
from dbgpt_ext.rag.transformer.triplet_extractor import TripletExtractor
|
||||||
|
Loading…
Reference in New Issue
Block a user