youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-03-11 02:07:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: remove streaming structured output from invoke_llm_with_pydantic_model

Browse Source 
Signed-off-by: Stream <Stream_2@qq.com>
This commit is contained in:
Stream
2026-01-29 23:27:51 +08:00
parent 6be800e14f
commit 749cebe60d
3 changed files with 11 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
api/tests/unit_tests/utils/structured_output_parser/test_structured_output_parser.py
View File

@ -492,7 +492,6 @@ def test_structured_output_with_pydantic_model_non_streaming():
model_instance=model_instance,
prompt_messages=prompt_messages,
output_model=ExampleOutput,
stream=False,
)
assert isinstance(result, ExampleOutput)
@ -532,8 +531,7 @@ def test_structured_output_with_pydantic_model_streaming():
model_schema=model_schema,
model_instance=model_instance,
prompt_messages=[UserPromptMessage(content="Return a JSON object with name.")],
output_model=ExampleOutput,
stream=True,
output_model=ExampleOutput
)
assert isinstance(result, ExampleOutput)
@ -555,8 +553,7 @@ def test_structured_output_with_pydantic_model_validation_error():
model_schema=model_schema,
model_instance=model_instance,
prompt_messages=[UserPromptMessage(content="test")],
output_model=ExampleOutput,
stream=False,
output_model=ExampleOutput
)