qzl
101 lines
2.9 KiB
Python
101 lines
2.9 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
from enum import Enum
|
|
from uuid import UUID
|
|
|
|
from core.agentscope.tools.tool_config import AgentTool
|
|
from pydantic import BaseModel, ConfigDict, Field
|
|
|
|
from models.automation_jobs import AutomationJob as OrmAutomationJob
|
|
from models.automation_jobs import AutomationJobStatus, ScheduleType
|
|
|
|
|
|
class ContextSource(str, Enum):
|
|
LATEST_CHAT = "latest_chat"
|
|
|
|
|
|
class ContextWindowMode(str, Enum):
|
|
DAY = "day"
|
|
NUMBER = "number"
|
|
|
|
|
|
class MessageContextConfig(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
source: ContextSource = ContextSource.LATEST_CHAT
|
|
window_mode: ContextWindowMode = ContextWindowMode.DAY
|
|
window_count: int = Field(default=2, ge=1, le=200)
|
|
|
|
|
|
class ScheduleRunAt(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
hour: int = Field(default=8, ge=0, le=23)
|
|
minute: int = Field(default=0, ge=0, le=59)
|
|
|
|
|
|
class ScheduleConfig(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
type: ScheduleType
|
|
run_at: ScheduleRunAt
|
|
|
|
|
|
class RuntimeConfig(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
enabled_tools: list[AgentTool] = Field(default_factory=list, max_length=32)
|
|
context: MessageContextConfig = Field(default_factory=MessageContextConfig)
|
|
|
|
|
|
class AutomationJobConfig(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
enabled_tools: list[AgentTool] | None = Field(default=None, max_length=32)
|
|
context: MessageContextConfig | None = None
|
|
input_template: str | None = Field(default=None, min_length=1, max_length=4000)
|
|
schedule: ScheduleConfig | None = None
|
|
|
|
|
|
class AutomationJob(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
id: UUID
|
|
owner_id: UUID
|
|
bootstrap_key: str | None = Field(default=None, min_length=1, max_length=64)
|
|
title: str = Field(..., min_length=1, max_length=255)
|
|
config: AutomationJobConfig
|
|
schedule_type: ScheduleType
|
|
run_at: datetime
|
|
next_run_at: datetime
|
|
timezone: str = Field(default="UTC", min_length=1, max_length=50)
|
|
last_run_at: datetime | None = None
|
|
status: AutomationJobStatus
|
|
created_by: UUID | None = None
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
@classmethod
|
|
def from_orm(cls, obj: OrmAutomationJob) -> "AutomationJob":
|
|
return cls(
|
|
id=obj.id,
|
|
owner_id=obj.owner_id,
|
|
bootstrap_key=obj.bootstrap_key,
|
|
title=obj.title,
|
|
config=AutomationJobConfig.model_validate(obj.config or {}),
|
|
schedule_type=obj.schedule_type,
|
|
run_at=obj.run_at,
|
|
next_run_at=obj.next_run_at,
|
|
timezone=obj.timezone,
|
|
last_run_at=obj.last_run_at,
|
|
status=obj.status,
|
|
created_by=obj.created_by,
|
|
created_at=obj.created_at,
|
|
updated_at=obj.updated_at,
|
|
)
|
|
|
|
@property
|
|
def is_system(self) -> bool:
|
|
return self.bootstrap_key is not None
|