github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-11 13:58:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(model): Support database model registry (#1656)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-06-24 19:07:10 +08:00
committed by GitHub
parent c57ee0289b
commit 47d205f676
35 changed files with 2014 additions and 792 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
dbgpt/app/initialization/db_model_initialization.py
View File

@@ -1,11 +1,14 @@
"""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.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
from dbgpt.storage.chat_history.chat_history_db import (
@@ -24,4 +27,6 @@ _MODELS = [
ConnectConfigEntity,
ChatHistoryEntity,
ChatHistoryMessageEntity,
ModelInstanceEntity,
FlowServeEntity,
]