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

feat: Run AWEL flow in CLI (#1341)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-03-27 12:50:05 +08:00
committed by GitHub
parent 340a9fbc35
commit 3a7a2cbbb8
42 changed files with 1454 additions and 422 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
dbgpt/model/utils/chatgpt_utils.py
View File

@@ -154,6 +154,9 @@ def _build_openai_client(init_params: OpenAIParameters) -> Tuple[str, ClientType
class OpenAIStreamingOutputOperator(TransformStreamAbsOperator[ModelOutput, str]):
"""Transform ModelOutput to openai stream format."""
incremental_output = True
output_format = "SSE"
metadata = ViewMetadata(
label=_("OpenAI Streaming Output Operator"),
name="openai_streaming_output_operator",
@@ -177,7 +180,7 @@ class OpenAIStreamingOutputOperator(TransformStreamAbsOperator[ModelOutput, str]
str,
is_list=True,
description=_(
"The model output after transform to openai stream format"
"The model output after transformed to openai stream format."
),
)
],