mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-16 06:30:02 +00:00
feat:rag graph conponent
This commit is contained in:
@@ -47,6 +47,7 @@ class ComponentType(str, Enum):
|
||||
WORKER_MANAGER_FACTORY = "dbgpt_worker_manager_factory"
|
||||
MODEL_CONTROLLER = "dbgpt_model_controller"
|
||||
EXECUTOR_DEFAULT = "dbgpt_thread_pool_default"
|
||||
RAG_GRAPH_DEFAULT = "dbgpt_rag_engine_default"
|
||||
|
||||
|
||||
class BaseComponent(LifeCycle, ABC):
|
||||
|
@@ -28,6 +28,10 @@ def initialize_components(
|
||||
|
||||
system_app.register_instance(controller)
|
||||
|
||||
# Register global default RAGGraphFactory
|
||||
from pilot.graph_engine.graph_factory import DefaultRAGGraphFactory
|
||||
system_app.register(DefaultRAGGraphFactory)
|
||||
|
||||
_initialize_embedding_model(
|
||||
param, system_app, embedding_model_name, embedding_model_path
|
||||
)
|
||||
|
Reference in New Issue
Block a user