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

refactor: remove streaming structured output from invoke_llm_with_structured_output

Browse Source 
Signed-off-by: Stream <Stream_2@qq.com>
This commit is contained in:
Stream
2026-01-29 23:41:08 +08:00
parent 749cebe60d
commit edce6d4152
4 changed files with 43 additions and 176 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -312,7 +312,6 @@ def test_structured_output_parser():
model_instance=model_instance,
prompt_messages=prompt_messages,
json_schema=case["json_schema"],
stream=case["stream"],
)
# Consume the generator to trigger the error
list(result_generator)
@ -323,7 +322,6 @@ def test_structured_output_parser():
model_instance=model_instance,
prompt_messages=prompt_messages,
json_schema=case["json_schema"],
stream=case["stream"],
)
else:
# Test successful cases
@ -338,7 +336,6 @@ def test_structured_output_parser():
model_instance=model_instance,
prompt_messages=prompt_messages,
json_schema=case["json_schema"],
stream=case["stream"],
model_parameters={"temperature": 0.7, "max_tokens": 100},
user="test_user",
)
@ -418,7 +415,6 @@ def test_parse_structured_output_edge_cases():
model_instance=model_instance,
prompt_messages=prompt_messages,
json_schema=testcase_list_with_dict["json_schema"],
stream=testcase_list_with_dict["stream"],
)
assert isinstance(result, LLMResultWithStructuredOutput)
@ -456,7 +452,6 @@ def test_model_specific_schema_preparation():
model_instance=model_instance,
prompt_messages=prompt_messages,
json_schema=gemini_case["json_schema"],
stream=gemini_case["stream"],
)
assert isinstance(result, LLMResultWithStructuredOutput)