feat:rag graph conponent

This commit is contained in:
aries_ckt
2023-10-12 11:00:21 +05:00
parent e2a1990696
commit eb2c220d22
2 changed files with 5 additions and 0 deletions

View File

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

View File

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