enne2
20766de540
- gateway/rerank.py: catena da RERANK_CHAIN (JSON, per-nodo key+timeout),
cooldown 60s sui nodi falliti, score sigmoide [0,1], degrada con grazia
all'ordine di fusione se tutti i nodi sono giù
- routes: /v1/memories:search applica il rerank post-fusione (fetch esteso a
RERANK_CANDIDATES), risposta con rerank{used,backend,took_ms}, flag
per-query rerank=false; /v1/version espone lo stato rerank
- store: search() accetta limit esteso; models: SearchIn.rerank
- metrics: qmem_rerank_calls_total + durata per backend
- test: 10 nuovi (fallback, cooldown, degradazione, integrazione) — 46 pass
61 lines
3.1 KiB
Python
61 lines
3.1 KiB
Python
"""Schemi Pydantic del Memory Gateway."""
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
from typing import Literal, Optional
|
|
|
|
from pydantic import BaseModel, Field, field_validator
|
|
|
|
from config import MAX_TEXT_LEN
|
|
|
|
|
|
class MemoryLink(BaseModel):
|
|
target_id: str = Field(..., description="UUID del record target collegato")
|
|
predicate: str = Field(default="part_of", max_length=64, description="Tipo di relazione: parent_of, part_of, relates_to, supersedes...")
|
|
weight: float = Field(default=1.0, ge=0.0, le=1.0)
|
|
|
|
|
|
class MemoryIn(BaseModel):
|
|
text: str = Field(min_length=1, max_length=MAX_TEXT_LEN)
|
|
kind: Literal["decision", "fact", "episode", "preference"] = "fact"
|
|
agent_id: Optional[str] = Field(default=None, max_length=64, description="Solo provenienza, nessun isolamento")
|
|
project_id: str = Field(min_length=1, max_length=64, description="OBBLIGATORIO: progetto/dominio di appartenenza (kebab-case)")
|
|
scope: Literal["agent", "project", "org"] = "agent"
|
|
source: Optional[str] = Field(default=None, max_length=256)
|
|
confidence: Literal["high", "medium", "low"] = Field(default="medium", description="Affidabilità del record")
|
|
expires_at: Optional[str] = None
|
|
supersedes_id: Optional[str] = None
|
|
supersede_reason: Optional[str] = Field(default=None, max_length=512)
|
|
parent_id: Optional[str] = Field(default=None, description="UUID del record genitore per gerarchia/subtopic")
|
|
level: Optional[Literal["L1_ROOT", "L2_SUBTOPIC", "L3_DETAIL"]] = Field(default=None, description="Livello gerarchico")
|
|
topic: Optional[str] = Field(default=None, max_length=128, description="Topic gerarchico")
|
|
links: Optional[list[MemoryLink]] = Field(default=None, description="Collegamenti semantici e relazionali")
|
|
private: bool = Field(default=False, description="Riservato: escluso dalle ricerche standard, visibile solo con include_private o topic esplicito")
|
|
|
|
@field_validator("expires_at")
|
|
@classmethod
|
|
def _validate_expires_at(cls, v: Optional[str]) -> Optional[str]:
|
|
if v is None:
|
|
return v
|
|
try:
|
|
datetime.fromisoformat(v.replace("Z", "+00:00"))
|
|
except ValueError:
|
|
raise ValueError("expires_at deve essere una data ISO 8601 valida (es. 2026-09-01T00:00:00Z)")
|
|
return v
|
|
|
|
|
|
class SearchIn(BaseModel):
|
|
query: str = Field(min_length=1, max_length=512)
|
|
kind: Optional[Literal["decision", "fact", "episode", "preference"]] = None
|
|
project_id: Optional[str] = None
|
|
scope: Optional[Literal["agent", "project", "org"]] = None
|
|
include_superseded: bool = False
|
|
min_score: Optional[float] = Field(default=None, ge=0.0, le=1.0)
|
|
top_k: int = Field(default=5, ge=1, le=20)
|
|
hybrid: bool = False
|
|
parent_id: Optional[str] = None
|
|
level: Optional[Literal["L1_ROOT", "L2_SUBTOPIC", "L3_DETAIL"]] = None
|
|
topic: Optional[str] = None
|
|
include_private: bool = Field(default=False, description="Includi i record privati (solo ricerche esplicite)")
|
|
rerank: Optional[bool] = Field(default=None, description="Override per-query dello stadio rerank (None = default server)")
|