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

17
dbgpt/model/cluster/apiserver/tests/test_api.py
View File

@@ -1,29 +1,28 @@
import importlib.metadata as metadata
import pytest
import pytest_asyncio
from aioresponses import aioresponses
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from httpx import AsyncClient, HTTPError
import importlib.metadata as metadata
from dbgpt.component import SystemApp
from dbgpt.util.openai_utils import chat_completion_stream, chat_completion
from dbgpt.model.cluster.apiserver.api import (
api_settings,
initialize_apiserver,
ModelList,
UsageInfo,
ChatCompletionResponse,
ChatCompletionResponseChoice,
ChatCompletionResponseStreamChoice,
ChatCompletionStreamResponse,
ChatMessage,
ChatCompletionResponseChoice,
DeltaMessage,
ModelList,
UsageInfo,
api_settings,
initialize_apiserver,
)
from dbgpt.model.cluster.tests.conftest import _new_cluster
from dbgpt.model.cluster.worker.manager import _DefaultWorkerManagerFactory
from dbgpt.util.openai_utils import chat_completion, chat_completion_stream
app = FastAPI()
app.add_middleware(