fix: update remaining core.file imports missed in previous commit

Made-with: Cursor
This commit is contained in:
Novice
2026-03-23 08:39:48 +08:00
parent a883b74272
commit 51c418f497
13 changed files with 13 additions and 13 deletions

View File

@ -2,13 +2,13 @@
from unittest.mock import patch
from core.model_runtime.entities.message_entities import ImagePromptMessageContent
from core.workflow.file import File, FileTransferMethod, FileType
from core.workflow.file.file_manager import (
_encode_file_ref,
restore_multimodal_content,
to_prompt_message_content,
)
from core.model_runtime.entities.message_entities import ImagePromptMessageContent
class TestEncodeFileRef:

View File

@ -4,7 +4,6 @@ Unit tests for sandbox file path detection and conversion.
import pytest
from core.workflow.file import File, FileTransferMethod, FileType
from core.llm_generator.output_parser.file_ref import (
FILE_PATH_DESCRIPTION_SUFFIX,
FILE_PATH_FORMAT,
@ -14,6 +13,7 @@ from core.llm_generator.output_parser.file_ref import (
is_file_path_property,
)
from core.variables.segments import ArrayFileSegment, FileSegment
from core.workflow.file import File, FileTransferMethod, FileType
def _build_file(file_id: str) -> File: