pull/18852/head
hjlarry 1 year ago
parent 6dffefeddd
commit 1d4a8bcb73

@ -24,4 +24,4 @@ def convert_llm_result_chunk_to_str(content: None | str | list[PromptMessageCont
) )
else: else:
message_text = str(content) message_text = str(content)
return message_text return message_text

@ -268,7 +268,6 @@ class LLMNode(BaseNode[LLMNodeData]):
return self._handle_invoke_result(invoke_result=invoke_result) return self._handle_invoke_result(invoke_result=invoke_result)
def _handle_invoke_result(self, invoke_result: LLMResult | Generator) -> Generator[NodeEvent, None, None]: def _handle_invoke_result(self, invoke_result: LLMResult | Generator) -> Generator[NodeEvent, None, None]:
if isinstance(invoke_result, LLMResult): if isinstance(invoke_result, LLMResult):
message_text = convert_llm_result_chunk_to_str(invoke_result.message.content) message_text = convert_llm_result_chunk_to_str(invoke_result.message.content)

Loading…
Cancel
Save