feat: mypy for all type check (#10921)

This commit is contained in:
yihong
2024-12-24 18:38:51 +08:00
committed by GitHub
parent c91e8b1737
commit 56e15d09a9
584 changed files with 3975 additions and 2826 deletions

View File

@ -5,7 +5,7 @@ from typing import Any, Optional
from pydantic import BaseModel, ConfigDict, Field, ValidationInfo, field_validator
from constants import UUID_NIL
from core.app.app_config.entities import AppConfig, EasyUIBasedAppConfig, WorkflowUIBasedAppConfig
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
@ -79,7 +79,7 @@ class AppGenerateEntity(BaseModel):
task_id: str
# app config
app_config: AppConfig
app_config: Any
file_upload_config: Optional[FileUploadConfig] = None
inputs: Mapping[str, Any]

View File

@ -308,7 +308,7 @@ class QueueNodeSucceededEvent(AppQueueEvent):
inputs: Optional[Mapping[str, Any]] = None
process_data: Optional[Mapping[str, Any]] = None
outputs: Optional[Mapping[str, Any]] = None
execution_metadata: Optional[dict[NodeRunMetadataKey, Any]] = None
execution_metadata: Optional[Mapping[NodeRunMetadataKey, Any]] = None
error: Optional[str] = None
"""single iteration duration map"""

View File

@ -70,7 +70,7 @@ class StreamResponse(BaseModel):
event: StreamEvent
task_id: str
def to_dict(self) -> dict:
def to_dict(self):
return jsonable_encoder(self)
@ -474,8 +474,8 @@ class IterationNodeStartStreamResponse(StreamResponse):
title: str
created_at: int
extras: dict = {}
metadata: dict = {}
inputs: dict = {}
metadata: Mapping = {}
inputs: Mapping = {}
parallel_id: Optional[str] = None
parallel_start_node_id: Optional[str] = None
@ -526,15 +526,15 @@ class IterationNodeCompletedStreamResponse(StreamResponse):
node_id: str
node_type: str
title: str
outputs: Optional[dict] = None
outputs: Optional[Mapping] = None
created_at: int
extras: Optional[dict] = None
inputs: Optional[dict] = None
inputs: Optional[Mapping] = None
status: WorkflowNodeExecutionStatus
error: Optional[str] = None
elapsed_time: float
total_tokens: int
execution_metadata: Optional[dict] = None
execution_metadata: Optional[Mapping] = None
finished_at: int
steps: int
parallel_id: Optional[str] = None
@ -628,7 +628,7 @@ class AppBlockingResponse(BaseModel):
task_id: str
def to_dict(self) -> dict:
def to_dict(self):
return jsonable_encoder(self)