mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-14 05:31:40 +00:00
feat: (0.6)New UI (#1855)
Co-authored-by: 夏姜 <wenfengjiang.jwf@digital-engine.com> Co-authored-by: aries_ckt <916701291@qq.com> Co-authored-by: wb-lh513319 <wb-lh513319@alibaba-inc.com> Co-authored-by: csunny <cfqsunny@163.com>
This commit is contained in:
@@ -1,13 +1,15 @@
|
||||
"""Import all models to make sure they are registered with SQLAlchemy.
|
||||
"""
|
||||
|
||||
from dbgpt.app.knowledge.chunk_db import DocumentChunkEntity
|
||||
from dbgpt.app.knowledge.document_db import KnowledgeDocumentEntity
|
||||
from dbgpt.app.openapi.api_v1.feedback.feed_back_db import ChatFeedBackEntity
|
||||
from dbgpt.datasource.manages.connect_config_db import ConnectConfigEntity
|
||||
from dbgpt.model.cluster.registry_impl.db_storage import ModelInstanceEntity
|
||||
from dbgpt.serve.agent.db.my_plugin_db import MyPluginEntity
|
||||
from dbgpt.serve.agent.db.plugin_hub_db import PluginHubEntity
|
||||
from dbgpt.serve.agent.app.recommend_question.recommend_question import (
|
||||
RecommendQuestionEntity,
|
||||
)
|
||||
from dbgpt.serve.agent.hub.db.my_plugin_db import MyPluginEntity
|
||||
from dbgpt.serve.agent.hub.db.plugin_hub_db import PluginHubEntity
|
||||
from dbgpt.serve.flow.models.models import ServeEntity as FlowServeEntity
|
||||
from dbgpt.serve.prompt.models.models import ServeEntity as PromptManageEntity
|
||||
from dbgpt.serve.rag.models.models import KnowledgeSpaceEntity
|
||||
@@ -29,4 +31,5 @@ _MODELS = [
|
||||
ChatHistoryMessageEntity,
|
||||
ModelInstanceEntity,
|
||||
FlowServeEntity,
|
||||
RecommendQuestionEntity,
|
||||
]
|
||||
|
@@ -65,3 +65,13 @@ def register_serve_apps(system_app: SystemApp, cfg: Config):
|
||||
# Register serve app
|
||||
system_app.register(DatasourceServe)
|
||||
# ################################ AWEL Flow Serve Register End ########################################
|
||||
|
||||
# ################################ Chat Feedback Serve Register End ########################################
|
||||
from dbgpt.serve.feedback.serve import (
|
||||
SERVE_CONFIG_KEY_PREFIX as Feedback_SERVE_CONFIG_KEY_PREFIX,
|
||||
)
|
||||
from dbgpt.serve.feedback.serve import Serve as FeedbackServe
|
||||
|
||||
# Register serve feedback
|
||||
system_app.register(FeedbackServe)
|
||||
# ################################ Chat Feedback Register End ########################################
|
||||
|
Reference in New Issue
Block a user