feat(core): Add common schemas

2025-09-15 05:59:59 +00:00 · 2024-03-21 11:23:24 +08:00
parent ab3e8e54a1
commit b4b810d68f
18 changed files with 188 additions and 123 deletions
--- a/dbgpt/serve/flow/service/service.py
+++ b/dbgpt/serve/flow/service/service.py
@@ -485,9 +485,7 @@ async def _chat_with_dag_task(
            if OpenAIStreamingOutputOperator and isinstance(
                task, OpenAIStreamingOutputOperator
            ):
-                from fastchat.protocol.openai_api_protocol import (
-                    ChatCompletionResponseStreamChoice,
-                )
+                from dbgpt.core.schema.api import ChatCompletionResponseStreamChoice

                previous_text = ""
                async for output in await task.call_stream(request):