refactor(memory): simplify Dream config naming and rename gitstore module
This commit is contained in:
@@ -3,10 +3,12 @@
|
||||
from pathlib import Path
|
||||
from typing import Literal
|
||||
|
||||
from pydantic import BaseModel, ConfigDict, Field
|
||||
from pydantic import AliasChoices, BaseModel, ConfigDict, Field
|
||||
from pydantic.alias_generators import to_camel
|
||||
from pydantic_settings import BaseSettings
|
||||
|
||||
from nanobot.cron.types import CronSchedule
|
||||
|
||||
|
||||
class Base(BaseModel):
|
||||
"""Base model that accepts both camelCase and snake_case keys."""
|
||||
@@ -31,11 +33,30 @@ class ChannelsConfig(Base):
|
||||
class DreamConfig(Base):
|
||||
"""Dream memory consolidation configuration."""
|
||||
|
||||
cron: str = "0 */2 * * *" # Every 2 hours
|
||||
model: str | None = None # Override model for Dream
|
||||
_HOUR_MS = 3_600_000
|
||||
|
||||
interval_h: int = Field(default=2, ge=1) # Every 2 hours by default
|
||||
cron: str | None = Field(default=None, exclude=True) # Legacy compatibility override
|
||||
model_override: str | None = Field(
|
||||
default=None,
|
||||
validation_alias=AliasChoices("modelOverride", "model", "model_override"),
|
||||
) # Optional Dream-specific model override
|
||||
max_batch_size: int = Field(default=20, ge=1) # Max history entries per run
|
||||
max_iterations: int = Field(default=10, ge=1) # Max tool calls per Phase 2
|
||||
|
||||
def build_schedule(self, timezone: str) -> CronSchedule:
|
||||
"""Build the runtime schedule, preferring the legacy cron override if present."""
|
||||
if self.cron:
|
||||
return CronSchedule(kind="cron", expr=self.cron, tz=timezone)
|
||||
return CronSchedule(kind="every", every_ms=self.interval_h * self._HOUR_MS)
|
||||
|
||||
def describe_schedule(self) -> str:
|
||||
"""Return a human-readable summary for logs and startup output."""
|
||||
if self.cron:
|
||||
return f"cron {self.cron} (legacy)"
|
||||
hours = self.interval_h
|
||||
return f"every {hours}h"
|
||||
|
||||
|
||||
class AgentDefaults(Base):
|
||||
"""Default agent configuration."""
|
||||
|
||||
Reference in New Issue
Block a user