mirror of
https://github.com/andrewyng/openworker.git
synced 2026-09-03 23:03:22 +00:00
Rohit C PrasadandDevika
2b45018ffa
Imported from andrewyng/aisuite@1b4bbf303e (contents of its platform/ directory, hoisted to the repo root). Development history prior to this commit lives in that repository. Co-authored-by: Devika <devikaverma11@gmail.com>
71 lines
1.9 KiB
Python
71 lines
1.9 KiB
Python
"""Persistent memory — adapter interface + scopes.
|
|
|
|
Memory is the long-lived layer above transient conversation state: durable facts,
|
|
preferences, task notes, summaries. Scopes: global (user-wide), workspace (per project),
|
|
session. Backends are adapters (`SQLiteMemoryStore` now, `PostgresMemoryStore` later).
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from abc import ABC, abstractmethod
|
|
from dataclasses import dataclass
|
|
from enum import Enum
|
|
from typing import Optional
|
|
|
|
|
|
class Scope(str, Enum):
|
|
GLOBAL = "global"
|
|
WORKSPACE = "workspace"
|
|
SESSION = "session"
|
|
|
|
|
|
@dataclass
|
|
class MemoryItem:
|
|
id: int
|
|
scope: Scope
|
|
content: str
|
|
key: Optional[str] = None
|
|
workspace: Optional[str] = None
|
|
session_id: Optional[str] = None
|
|
created_at: Optional[str] = None
|
|
|
|
|
|
class MemoryStore(ABC):
|
|
@abstractmethod
|
|
def add(
|
|
self,
|
|
content: str,
|
|
*,
|
|
scope: Scope = Scope.WORKSPACE,
|
|
key: Optional[str] = None,
|
|
workspace: Optional[str] = None,
|
|
session_id: Optional[str] = None,
|
|
) -> MemoryItem: ...
|
|
|
|
@abstractmethod
|
|
def get(self, item_id: int) -> Optional[MemoryItem]: ...
|
|
|
|
@abstractmethod
|
|
def list(
|
|
self,
|
|
*,
|
|
scope: Optional[Scope] = None,
|
|
workspace: Optional[str] = None,
|
|
session_id: Optional[str] = None,
|
|
) -> list[MemoryItem]: ...
|
|
|
|
@abstractmethod
|
|
def update(self, item_id: int, content: str) -> Optional[MemoryItem]: ...
|
|
|
|
@abstractmethod
|
|
def delete(self, item_id: int) -> bool: ...
|
|
|
|
|
|
def format_memories(items: list[MemoryItem]) -> str:
|
|
"""Render memories for injection into the system prompt. Ids are shown so the agent
|
|
can revise a memory (`memory_update`) or retire it (`memory_forget`)."""
|
|
if not items:
|
|
return ""
|
|
lines = [f"- [#{item.id}] {item.content}" for item in items]
|
|
return "Known memories (from earlier sessions):\n" + "\n".join(lines)
|