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

fix: call get_text_content instead of accessing content directly

Browse Source
This commit is contained in:
Stream
2026-01-27 18:54:36 +08:00
parent adf104becd
commit ae23d30da2
1 changed files with 2 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
api/core/llm_generator/output_parser/structured_output.py
View File

@ -208,13 +208,7 @@ def invoke_llm_with_structured_output(
system_fingerprint = event.system_fingerprint
# Collect text content
if isinstance(event.delta.message.content, str):
result_text += event.delta.message.content
elif isinstance(event.delta.message.content, list):
for item in event.delta.message.content:
if isinstance(item, TextPromptMessageContent):
result_text += item.data
result_text += event.delta.message.get_text_content()
# Collect tool call arguments
if event.delta.message.tool_calls:
for tool_call in event.delta.message.tool_calls:
@ -350,9 +344,7 @@ def _extract_structured_output(llm_result: LLMResult) -> Mapping[str, Any]:
return _parse_tool_call_arguments(tool_call.function.arguments)
# Fallback to text content parsing
content = llm_result.message.content
if not isinstance(content, str):
raise OutputParserError(f"Failed to parse structured output, LLM result is not a string: {content}")
content = llm_result.message.get_text_content()
return _parse_structured_output(content)