github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-08 20:39:44 +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

6
dbgpt/model/cluster/worker/manager.py
View File

@@ -1059,6 +1059,10 @@ def initialize_worker_manager_in_client(
if not app:
raise Exception("app can't be None")
if system_app:
logger.info(f"Register WorkerManager {_DefaultWorkerManagerFactory.name}")
system_app.register(_DefaultWorkerManagerFactory, worker_manager)
worker_params: ModelWorkerParameters = _parse_worker_params(
model_name=model_name, model_path=model_path, controller_addr=controller_addr
)
@@ -1104,8 +1108,6 @@ def initialize_worker_manager_in_client(
if include_router and app:
# mount WorkerManager router
app.include_router(router, prefix="/api")
if system_app:
system_app.register(_DefaultWorkerManagerFactory, worker_manager)
def run_worker_manager(