AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-15 22:05:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix bug

Browse Source
This commit is contained in:
takatost 2024-08-23 00:38:42 +08:00
parent 5b22d8f8b2
commit 42899fb3be
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
api/core/workflow/nodes/llm/llm_node.py
View File

@ -52,6 +52,7 @@ class ModelInvokeCompleted(BaseModel):
""" """
text: str text: str
usage: LLMUsage usage: LLMUsage
finish_reason: Optional[str] = None
class LLMNode(BaseNode): class LLMNode(BaseNode):

4
api/core/workflow/nodes/question_classifier/question_classifier_node.py
View File

@ -74,12 +74,12 @@ class QuestionClassifierNode(LLMNode):
result_text = '' result_text = ''
usage = LLMUsage.empty_usage() usage = LLMUsage.empty_usage()
finished_reason = None finish_reason = None
for event in generator: for event in generator:
if isinstance(event, ModelInvokeCompleted): if isinstance(event, ModelInvokeCompleted):
result_text = event.text result_text = event.text
usage = event.usage usage = event.usage
finished_reason = event.finished_reason finish_reason = event.finish_reason
break break
category_name = node_data.classes[0].name category_name = node_data.classes[0].name