qzl
112 lines
3.4 KiB
Python
112 lines
3.4 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, model_validator
|
|
|
|
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
|
|
weekdays: list[int] | None = None
|
|
|
|
@model_validator(mode="after")
|
|
def validate_weekdays(self) -> "ScheduleConfig":
|
|
if self.type == ScheduleType.WEEKLY:
|
|
if not self.weekdays:
|
|
raise ValueError("weekdays is required when schedule type is weekly")
|
|
invalid = [day for day in self.weekdays if day < 1 or day > 7]
|
|
if invalid:
|
|
raise ValueError("weekdays must be within 1-7")
|
|
deduped = sorted(set(self.weekdays))
|
|
self.weekdays = deduped
|
|
else:
|
|
self.weekdays = None
|
|
return self
|
|
|
|
|
|
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
|
|
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 {}),
|
|
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
|