youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-22 19:57:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: circular import

Browse Source
This commit is contained in:
Stream
2026-01-30 07:11:20 +08:00
parent 7926024569
commit ec9ade62f5
2 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
api/core/plugin/utils/chunk_merger.py
View File

@ -1,11 +1,13 @@
from collections.abc import Generator
from dataclasses import dataclass, field
from typing import TypeVar, Union
from typing import TYPE_CHECKING, TypeVar
from core.agent.entities import AgentInvokeMessage
from core.tools.entities.tool_entities import ToolInvokeMessage
MessageType = TypeVar("MessageType", bound=Union[ToolInvokeMessage, AgentInvokeMessage])
if TYPE_CHECKING:
from core.agent.entities import AgentInvokeMessage
MessageType = TypeVar("MessageType", bound=ToolInvokeMessage)
@dataclass
@ -87,7 +89,7 @@ def merge_blob_chunks(
),
meta=resp.meta,
)
assert isinstance(merged_message, (ToolInvokeMessage, AgentInvokeMessage))
assert isinstance(merged_message, ToolInvokeMessage)
yield merged_message # type: ignore
# Clean up the buffer
del files[chunk_id]