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

feat(core): APP use new SDK component (#1050)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-01-10 10:39:04 +08:00
committed by GitHub
parent e11b72c724
commit fa8b5b190c
242 changed files with 2768 additions and 2163 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
dbgpt/model/cluster/controller/controller.py
View File

@@ -1,19 +1,17 @@
from abc import ABC, abstractmethod
import logging
from abc import ABC, abstractmethod
from typing import List
from fastapi import APIRouter, FastAPI
from dbgpt.component import BaseComponent, ComponentType, SystemApp
from dbgpt.model.base import ModelInstance
from dbgpt.model.parameter import ModelControllerParameters
from dbgpt.model.cluster.registry import EmbeddedModelRegistry, ModelRegistry
from dbgpt.model.parameter import ModelControllerParameters
from dbgpt.util.api_utils import _api_remote as api_remote
from dbgpt.util.api_utils import _sync_api_remote as sync_api_remote
from dbgpt.util.parameter_utils import EnvArgumentParser
from dbgpt.util.api_utils import (
_api_remote as api_remote,
_sync_api_remote as sync_api_remote,
)
from dbgpt.util.utils import setup_logging, setup_http_service_logging
from dbgpt.util.utils import setup_http_service_logging, setup_logging
logger = logging.getLogger(__name__)

5
dbgpt/model/cluster/controller/tests/test_registry.py
View File

@@ -1,7 +1,8 @@
import pytest
import asyncio
from datetime import datetime, timedelta
import asyncio
import pytest
from dbgpt.model.base import ModelInstance
from dbgpt.model.cluster.registry import EmbeddedModelRegistry