mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-09-06 08:16:00 +00:00
d3aee1adf5
* feat(evolve): report auto-dream content changes * perf(evolve): use lightweight dream snapshots
100 lines
3.7 KiB
Python
100 lines
3.7 KiB
Python
"""Auto-dream schemas."""
|
|
|
|
from typing import Literal
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
from ..enumeration import DreamBucketEnum
|
|
|
|
|
|
class DreamUnit(BaseModel):
|
|
"""One cross-file memory unit emitted by global extract."""
|
|
|
|
name: str = Field(description="Short kebab-case handle for the abstraction.")
|
|
bucket: DreamBucketEnum = Field(description="Digest bucket; unknown raw values route to wiki before validation.")
|
|
summary: str = Field(description="Grounded abstraction summary with evidence pointers.")
|
|
paths: list[str] = Field(default_factory=list, description="Workspace-relative source paths.")
|
|
|
|
|
|
class DreamTopic(BaseModel):
|
|
"""One topic candidate emitted by global extract."""
|
|
|
|
title: str = Field(description="Specific user-interest topic title.")
|
|
reason: str = Field(description="Why this topic may interest the user.")
|
|
evidence: str = Field(description="Grounded evidence pointer.")
|
|
keywords: list[str] = Field(default_factory=list, description="Keywords for de-duplication.")
|
|
paths: list[str] = Field(default_factory=list, description="Workspace-relative source paths.")
|
|
|
|
|
|
class DreamExtractOutput(BaseModel):
|
|
"""Structured output for ``dream_extract_step``."""
|
|
|
|
units: list[DreamUnit] = Field(default_factory=list)
|
|
topics: list[DreamTopic] = Field(default_factory=list)
|
|
|
|
|
|
class IntegrateOutcome(BaseModel):
|
|
"""Structured output for one unit integration."""
|
|
|
|
action: Literal["CREATE", "CORROBORATE", "REFINE", "CORRECT"] = Field(description="Write decision.")
|
|
target_path: str = Field(description="Digest path written or edited.")
|
|
note: str = Field(default="", description="Short summary of what landed.")
|
|
|
|
|
|
class TopicSelectionOutput(BaseModel):
|
|
"""Structured output for daily topic selection."""
|
|
|
|
topics: list[DreamTopic] = Field(default_factory=list)
|
|
|
|
|
|
class ProactiveResult(BaseModel):
|
|
"""Result of reading daily interest topics."""
|
|
|
|
date: str = ""
|
|
path: str = ""
|
|
topics: list[dict] = Field(default_factory=list)
|
|
content: str = ""
|
|
skipped: bool = False
|
|
error: str = ""
|
|
summary: str = ""
|
|
|
|
|
|
class DreamState(BaseModel):
|
|
"""Shared state passed across the dream steps."""
|
|
|
|
date: str = ""
|
|
dates: list[str] = Field(default_factory=list)
|
|
scan_days: int = 2
|
|
hint: str = ""
|
|
daily_dir: str = ""
|
|
workspace: str = ""
|
|
files_scanned: int = 0
|
|
files_unchanged: int = 0
|
|
files_changed: int = 0
|
|
files_deleted: int = 0
|
|
changed_paths: list[str] = Field(default_factory=list)
|
|
unchanged_paths: list[str] = Field(default_factory=list)
|
|
deleted_paths: list[str] = Field(default_factory=list)
|
|
existing: dict[str, float] = Field(default_factory=dict)
|
|
indexed: dict[str, float] = Field(default_factory=dict)
|
|
units: list[dict] = Field(default_factory=list)
|
|
topics: list[dict] = Field(default_factory=list)
|
|
extract_summary: str = ""
|
|
integrate_results: list[dict] = Field(default_factory=list)
|
|
skipped_units: list[dict] = Field(default_factory=list)
|
|
nodes_created: list[str] = Field(default_factory=list)
|
|
nodes_updated: list[str] = Field(default_factory=list)
|
|
modified_paths: list[str] = Field(
|
|
default_factory=list,
|
|
description="Durable digest or interests files detected as created or changed during this run.",
|
|
)
|
|
failed_units: list[dict] = Field(default_factory=list)
|
|
failed_paths: list[str] = Field(default_factory=list)
|
|
interests_path: str = ""
|
|
interests_paths: list[str] = Field(default_factory=list)
|
|
topics_written: int = 0
|
|
topic_error: str = ""
|
|
checkpoint_paths: list[str] = Field(default_factory=list)
|
|
warnings: list[str] = Field(default_factory=list)
|
|
errors: list[str] = Field(default_factory=list)
|
|
summary: str = ""
|