mirror of
https://github.com/langgenius/dify.git
synced 2025-12-25 01:00:42 -05:00
Fix validation (#28985)
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
This commit is contained in:
@@ -4,15 +4,15 @@ from typing import TYPE_CHECKING, Any, Optional
|
||||
|
||||
from pydantic import BaseModel, ConfigDict, Field, ValidationInfo, field_validator
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from core.ops.ops_trace_manager import TraceQueueManager
|
||||
|
||||
from constants import UUID_NIL
|
||||
from core.app.app_config.entities import EasyUIBasedAppConfig, WorkflowUIBasedAppConfig
|
||||
from core.entities.provider_configuration import ProviderModelBundle
|
||||
from core.file import File, FileUploadConfig
|
||||
from core.model_runtime.entities.model_entities import AIModelEntity
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from core.ops.ops_trace_manager import TraceQueueManager
|
||||
|
||||
|
||||
class InvokeFrom(StrEnum):
|
||||
"""
|
||||
@@ -130,7 +130,7 @@ class AppGenerateEntity(BaseModel):
|
||||
# extra parameters, like: auto_generate_conversation_name
|
||||
extras: dict[str, Any] = Field(default_factory=dict)
|
||||
|
||||
# tracing instance; use forward ref to avoid circular import at import time
|
||||
# tracing instance
|
||||
trace_manager: Optional["TraceQueueManager"] = None
|
||||
|
||||
|
||||
@@ -275,23 +275,14 @@ class RagPipelineGenerateEntity(WorkflowAppGenerateEntity):
|
||||
start_node_id: str | None = None
|
||||
|
||||
|
||||
# NOTE: Avoid importing heavy tracing modules at import time to prevent circular imports.
|
||||
# Forward reference to TraceQueueManager is kept as a string; we rebuild with a stub now to
|
||||
# avoid Pydantic forward-ref errors in test contexts, and with the real class at app startup.
|
||||
from core.ops.ops_trace_manager import TraceQueueManager
|
||||
|
||||
|
||||
# Minimal stub to satisfy Pydantic model_rebuild in environments where the real type is not importable yet.
|
||||
class _TraceQueueManagerStub:
|
||||
pass
|
||||
|
||||
|
||||
_ns = {"TraceQueueManager": _TraceQueueManagerStub}
|
||||
AppGenerateEntity.model_rebuild(_types_namespace=_ns)
|
||||
EasyUIBasedAppGenerateEntity.model_rebuild(_types_namespace=_ns)
|
||||
ConversationAppGenerateEntity.model_rebuild(_types_namespace=_ns)
|
||||
ChatAppGenerateEntity.model_rebuild(_types_namespace=_ns)
|
||||
CompletionAppGenerateEntity.model_rebuild(_types_namespace=_ns)
|
||||
AgentChatAppGenerateEntity.model_rebuild(_types_namespace=_ns)
|
||||
AdvancedChatAppGenerateEntity.model_rebuild(_types_namespace=_ns)
|
||||
WorkflowAppGenerateEntity.model_rebuild(_types_namespace=_ns)
|
||||
RagPipelineGenerateEntity.model_rebuild(_types_namespace=_ns)
|
||||
AppGenerateEntity.model_rebuild()
|
||||
EasyUIBasedAppGenerateEntity.model_rebuild()
|
||||
ConversationAppGenerateEntity.model_rebuild()
|
||||
ChatAppGenerateEntity.model_rebuild()
|
||||
CompletionAppGenerateEntity.model_rebuild()
|
||||
AgentChatAppGenerateEntity.model_rebuild()
|
||||
AdvancedChatAppGenerateEntity.model_rebuild()
|
||||
WorkflowAppGenerateEntity.model_rebuild()
|
||||
RagPipelineGenerateEntity.model_rebuild()
|
||||
|
||||
Reference in New Issue
Block a user