mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-05-27 19:25:15 +02:00
24 lines
736 B
Python
24 lines
736 B
Python
|
"""Structured output schemas for memory extraction."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from typing import Literal
|
||
|
|
|
||
|
|
from pydantic import BaseModel, Field
|
||
|
|
|
||
|
|
|
||
|
|
class MemoryExtractionDecision(BaseModel):
|
||
|
|
"""Structured extraction result; avoids string sentinel parsing."""
|
||
|
|
|
||
|
|
action: Literal["no_update", "save"] = Field(
|
||
|
|
description="Choose no_update when nothing durable should be saved; choose save otherwise."
|
||
|
|
)
|
||
|
|
reason: str | None = Field(
|
||
|
|
default=None,
|
||
|
|
description="Short reason for no_update, or brief summary of the memory update.",
|
||
|
|
)
|
||
|
|
updated_memory: str | None = Field(
|
||
|
|
default=None,
|
||
|
|
description="The full updated markdown memory document when action is save.",
|
||
|
|
)
|