refactor(agent): remove memory agent, simplify runtime config system

2026-03-23 01:20:27 +08:00
parent 80ad5141a6
commit 3aacc756db
43 changed files with 1210 additions and 1312 deletions
@@ -1,20 +1,77 @@
-from schemas.automation.config import (
-    AutomationAgentType,
-    AutomationContextSource,
-    AutomationContextWindowMode,
-    AutomationJobConfig,
-    AutomationMemoryContextConfig,
-    default_memory_job_config,
-)
-from schemas.automation.scheduler import DueAutomationJob, SchedulerDispatchCommand
+from __future__ import annotations

-__all__ = [
-    "AutomationAgentType",
-    "AutomationContextSource",
-    "AutomationContextWindowMode",
-    "AutomationJobConfig",
-    "AutomationMemoryContextConfig",
-    "default_memory_job_config",
-    "DueAutomationJob",
-    "SchedulerDispatchCommand",
-]
+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 MemoryContextConfig(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 RuntimeConfig(BaseModel):
+    model_config = ConfigDict(extra="forbid")
+
+    enabled_tools: list[AgentTool] = Field(default_factory=list, max_length=32)
+    context: MemoryContextConfig = Field(default_factory=MemoryContextConfig)
+
+
+class AutomationJobConfig(RuntimeConfig):
+    model_config = ConfigDict(extra="forbid")
+
+    input_template: str = Field(..., min_length=1, max_length=4000)
+
+
+class AutomationJob(BaseModel):
+    model_config = ConfigDict(extra="forbid")
+
+    id: UUID
+    owner_id: UUID
+    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,
+            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,
+        )