github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-02 01:27:14 +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

23
dbgpt/model/cluster/tests/conftest.py
View File

@@ -1,21 +1,22 @@
from contextlib import asynccontextmanager, contextmanager
from typing import Dict, Iterator, List, Tuple
import pytest
import pytest_asyncio
from contextlib import contextmanager, asynccontextmanager
from typing import List, Iterator, Dict, Tuple
from dbgpt.model.parameter import ModelParameters, ModelWorkerParameters, WorkerType
from dbgpt.core import ModelOutput, ModelMetadata
from dbgpt.model.cluster.worker_base import ModelWorker
from dbgpt.core import ModelMetadata, ModelOutput
from dbgpt.model.base import ModelInstance
from dbgpt.model.cluster.registry import EmbeddedModelRegistry, ModelRegistry
from dbgpt.model.cluster.worker.manager import (
WorkerManager,
ApplyFunction,
DeregisterFunc,
LocalWorkerManager,
RegisterFunc,
DeregisterFunc,
SendHeartbeatFunc,
ApplyFunction,
WorkerManager,
)
from dbgpt.model.base import ModelInstance
from dbgpt.model.cluster.registry import ModelRegistry, EmbeddedModelRegistry
from dbgpt.model.cluster.worker_base import ModelWorker
from dbgpt.model.parameter import ModelParameters, ModelWorkerParameters, WorkerType
@pytest.fixture