2025-06-17 18:24:09 +08:00
|
|
|
from collections.abc import Mapping
|
|
|
|
from enum import Enum
|
|
|
|
from typing import Any, Optional
|
|
|
|
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
|
|
|
|
|
|
class DatasourceStreamEvent(Enum):
|
|
|
|
"""
|
|
|
|
Datasource Stream event
|
|
|
|
"""
|
2025-06-17 18:29:02 +08:00
|
|
|
PROCESSING = "datasource_processing"
|
|
|
|
COMPLETED = "datasource_completed"
|
2025-06-17 18:24:09 +08:00
|
|
|
|
|
|
|
|
|
|
|
class BaseDatasourceEvent(BaseModel):
|
|
|
|
pass
|
|
|
|
|
|
|
|
class DatasourceCompletedEvent(BaseDatasourceEvent):
|
|
|
|
event: str = DatasourceStreamEvent.COMPLETED.value
|
|
|
|
data: Mapping[str,Any] | list = Field(..., description="result")
|
|
|
|
total: Optional[int] = Field(..., description="total")
|
|
|
|
completed: Optional[int] = Field(..., description="completed")
|
|
|
|
time_consuming: Optional[float] = Field(..., description="time consuming")
|
|
|
|
|
|
|
|
class DatasourceProcessingEvent(BaseDatasourceEvent):
|
|
|
|
event: str = DatasourceStreamEvent.PROCESSING.value
|
|
|
|
total: Optional[int] = Field(..., description="total")
|
|
|
|
completed: Optional[int] = Field(..., description="completed")
|
|
|
|
|