fix:rag workflow error fix

This commit is contained in:
aries_ckt 2025-03-23 19:17:43 +08:00
parent eccbe2ba94
commit d095091525
2 changed files with 25 additions and 7 deletions

View File

@ -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."""

View File

@ -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