mirror of
https://github.com/langgenius/dify.git
synced 2026-03-25 23:01:33 -04:00
fix: resolve test failures after segment 2 merge
- Backend: fix deduct_llm_quota import path in llm node - Backend: update core.variables → core.workflow.variables imports - Frontend: update UpdateWorkflowNodesMapPayload in tests - Frontend: fix various test expectations to match merged code Made-with: Cursor
This commit is contained in:
@@ -6,14 +6,14 @@ from pathlib import PurePosixPath
|
||||
from typing import Any, cast
|
||||
|
||||
from core.sandbox.bash.session import SANDBOX_READY_TIMEOUT
|
||||
from core.workflow.variables import ArrayFileSegment
|
||||
from core.workflow.variables.segments import ArrayStringSegment, FileSegment
|
||||
from core.virtual_environment.__base.command_future import CommandCancelledError, CommandTimeoutError
|
||||
from core.virtual_environment.__base.helpers import pipeline
|
||||
from core.workflow.enums import NodeType, WorkflowNodeExecutionStatus
|
||||
from core.workflow.file import File, FileTransferMethod
|
||||
from core.workflow.node_events import NodeRunResult
|
||||
from core.workflow.nodes.base.node import Node
|
||||
from core.workflow.variables import ArrayFileSegment
|
||||
from core.workflow.variables.segments import ArrayStringSegment, FileSegment
|
||||
from core.zip_sandbox import SandboxDownloadItem
|
||||
|
||||
from .entities import FileUploadNodeData
|
||||
|
||||
@@ -1969,7 +1969,9 @@ class LLMNode(Node[LLMNodeData]):
|
||||
LLMStructuredOutput(structured_output=structured_raw) if structured_raw else None
|
||||
)
|
||||
|
||||
llm_utils.deduct_llm_quota(tenant_id=self.tenant_id, model_instance=model_instance, usage=usage)
|
||||
from core.app.llm.quota import deduct_llm_quota
|
||||
|
||||
deduct_llm_quota(tenant_id=self.tenant_id, model_instance=model_instance, usage=usage)
|
||||
completed = True
|
||||
|
||||
case LLMStructuredOutput():
|
||||
|
||||
Reference in New Issue
Block a user