* feat: entity labels * feat: entity labels — optional, free_values, multi_value, UI polish Completes the entity labels system: **Schema & extraction** - Dynamic Pydantic Labels model per fact: each group becomes a typed field (Literal | None, list[Literal], str | None, or list[str]) - `optional: bool` flag per group — non-optional enum fields appear in JSON schema required array so structured-output providers enforce them - `free_values: bool` flag per group — accepts any LLM-generated string instead of a predefined enum; example values shown as hints in prompt - New `is_label_entity()` helper for labels-only mode filtering that handles both enum lookup and free_values key-prefix matching - Sentinel rejection: "None"/"null"/"n/a" strings dropped in post-processing **BM25 / dense retrieval** - `text_signals` column on memory_units: entity names + date tokens for enriched BM25 indexing without polluting stored fact text - Dense embedding includes occurred_end when it differs from occurred_start - Alembic migration z1u2v3w4x5y6 (merge revision fixing two heads) **UI (bank-config-view)** - Shadcn Switch replaces custom Toggle for both entity-labels and observations - Shadcn Checkbox for multi/optional/free_values per group - Input heights bumped to h-8 throughout the editor - "Label Groups" → "Entity Labels", "Free-form entities" → "Entities" - Free-text groups show "Example hints" banner in values section **Tests (45 unit + 3 LLM integration)** - build_labels_model: single, multi, mixed, free_values optional/required/multi - is_label_entity: enum match, free_values prefix match, no false positives - Post-processing: null/absent/string-None/free_values/sentinels/multi-value - Schema: labels in required, structured object, no labels when unconfigured - LLM integration: single-value enum, multi-value enum, free_values retain **Docs** - retain.md: new Entity Labels section covering groups, flags, examples - configuration.md: retain_free_form_entities env var + entity_labels note * fix(tests): update hierarchical fields count for entity_labels additions entity_labels and retain_free_form_entities are hierarchical fields, bumping the expected count from 11 to 13. * fix(migration): rename text_signals revision to avoid collision with main Main branch claimed z1u2v3w4x5y6 for observation_scopes. Rename our text_signals migration to a2b3c4d5e6f7, chaining after z1u2v3w4x5y6. * refactor(entity-labels): simplify free_values — always str|None, no multi - free_values groups always produce str | None (multi_value and optional flags are ignored for free text groups — always optional, never multi) - Prompt section for free_values groups shows only key + description, no values list (users put examples in the description instead) - UI: section title "Entities", toggle "Free Form Entities", replace per-group checkboxes with a type dropdown (Enum / Free text); only show multi checkbox and values list when type is Enum - Update tests to reflect new behaviour * refactor(entity-labels): replace free_values/multi_value booleans with type field - LabelGroup now uses type: "value" | "multi-values" | "text" instead of free_values/multi_value boolean pair - Backward-compat migration converts legacy dicts automatically - Rename retain_free_form_entities → entities_allow_free_form throughout - Update UI dropdown to show Single value / Multi-values / Free text - Remove separate multi checkbox (captured by type selection) - Update docs examples and configuration.md - Update all tests to use new field names * fix(migration): backfill observation_scopes column for DBs with swapped z1u2v3w4x5y6 Local DBs that had z1u2v3w4x5y6 applied when it referred to the old text_signals migration (before it was renamed to a2b3c4d5e6f7) won't have observation_scopes in their memory_units table. This migration adds the column with IF NOT EXISTS so it's a no-op on clean installs. * feat(entity-labels): add tag field to auto-populate memory unit tags from labels When a LabelGroup has tag=True, extracted key:value entities for that group are automatically written to the memory unit's tags array. This lets entity labels double as tags, enabling immediate filtering via the existing tags/tags_match API params with no extra infrastructure. - Add tag: bool = False to LabelGroup - _inject_label_tags() helper called in both sync and batch extraction paths - UI: add Tag checkbox per label group row - Docs: document the new tag field - Tests: 4 new unit tests covering all tag injection paths * style: ruff format migration file * fix(migration): fix multiple alembic heads after rebase — point text_signals after nullable_event_date * fix(clients): update timestamp field to use Timestamp wrapper type after timestamp=unset feature * style: ruff format agent.py * fix(docs): update Go quickstart example to use NullableTimestamp for timestamp field
12 KiB
| sidebar_position |
|---|
| 2 |
Retain: How Hindsight Stores Memories
When you call retain(), Hindsight transforms conversations and documents into structured, searchable memories that preserve meaning and context.
What Retain Does
graph LR
A[Your Content] --> B[Extract Facts]
B --> C[Identify Entities]
C --> D[Build Connections]
D --> E[Memory Bank]
Rich Fact Extraction
Hindsight doesn't just store what was said — it captures why, how, and what it means.
What Gets Captured
When you retain "Alice joined Google last spring and was thrilled about the research opportunities", Hindsight extracts:
The core facts:
- Alice joined Google
- This happened last spring
The emotions and meaning:
- She was thrilled
- It represented an important opportunity
The reasoning:
- She chose it for the research opportunities
This rich extraction means you can later ask "Why did Alice join Google?" and get a meaningful answer, not just "she joined Google."
Preserving Context
Traditional systems fragment information:
- "Bob suggested Summer Vibes"
- "Alice wanted something unique"
- "They chose Beach Beats"
Hindsight preserves the full narrative:
- "Alice and Bob discussed naming their summer party playlist. Bob suggested 'Summer Vibes' because it's catchy, but Alice wanted something unique. They ultimately decided on 'Beach Beats' for its playful tone."
This means search results include the full context, not disconnected fragments.
Two Types of Facts
Hindsight distinguishes between world facts (about others) and experience (conversations and events):
| Type | Description | Example |
|---|---|---|
| world | Facts about people, places, things | "Alice works at Google" |
| experience | Conversations and events | "I recommended Python to Alice" |
Note: Observations are consolidated automatically in the background after retain() operations complete. This consolidation process synthesizes patterns from new facts into the bank's knowledge base.
Entity Recognition
Hindsight automatically identifies and tracks entities — the people, organizations, and concepts that matter.
What Gets Recognized
- People: "Alice", "Dr. Smith", "Bob Chen"
- Organizations: "Google", "MIT", "OpenAI"
- Places: "Paris", "Central Park", "California"
- Products & Concepts: "Python", "TensorFlow", "machine learning"
Entity Resolution
The same entity mentioned different ways gets unified:
- "Alice" + "Alice Chen" + "Alice C." → one person
- "Bob" + "Robert Chen" → one person (nickname resolution)
Why it matters: You can ask "What do I know about Alice?" and get everything, even if she was mentioned as "Alice Chen" in some conversations.
Context-Aware Disambiguation
If "Alice" appears with "Google" and "Stanford" multiple times, a new "Alice" mentioning those is likely the same person. Hindsight uses co-occurrence patterns to disambiguate common names.
Building Connections
Memories aren't isolated — Hindsight creates a knowledge graph with four types of connections:
Entity Connections
All facts mentioning the same entity are linked together.
Enables: "Tell me everything about Alice" → retrieves all Alice-related facts
Time-Based Connections
Facts close in time are connected, with stronger links for closer dates.
Enables: "What else happened around then?" → finds contextually related events
Meaning-Based Connections
Semantically similar facts are linked, even if they use different words.
Enables: "Tell me about similar topics" → finds thematically related information
Causal Connections
Cause-effect relationships are explicitly tracked.
Enables: "Why did this happen?" → trace reasoning chains Example: "Alice felt burned out" ← caused by ← "She worked 80-hour weeks"
Understanding Time
Hindsight tracks two temporal dimensions:
When It Happened
For events (meetings, trips, milestones), Hindsight records when they occurred.
- "Alice got married in June 2024" → occurred in June 2024
For general facts (preferences, characteristics), there's no specific occurrence time.
- "Alice prefers Python" → ongoing preference
When You Learned It
Hindsight also tracks when you told it each fact.
Why both?
Imagine in January 2025, someone tells you "Alice got married in June 2024":
- Historical queries work: "What did Alice do in 2024?" → finds the marriage
- Recency ranking works: Recent mentions get priority in search
- Temporal reasoning works: "What happened before her marriage?" → finds earlier events
Without this distinction, old information would either be unsearchable by date or treated as irrelevant.
Tagging Memories
Tags enable visibility scoping—useful when one memory bank serves multiple users but each should only see relevant memories.
- Item tags: Tag individual memories with specific scopes
- Document tags: Apply tags to all items in a batch
- Tag filtering: Filter during recall/reflect by tags
See Retain API for code examples and Recall API for filtering options.
What You Get
After retain() completes:
- Structured facts that preserve meaning, emotions, and reasoning
- Unified entities that resolve different name variations
- Knowledge graph with entity, temporal, semantic, and causal links
- Temporal grounding for both historical and recency-based queries
- Optional tags for filtering during recall
All stored in your isolated memory bank, ready for recall() and reflect().
Steering Extraction with a Mission
By default, retain() extracts all significant facts from the content. You can narrow this focus with a retain mission (retain_mission) — a plain-language description of what this bank should pay attention to.
e.g. Always include technical decisions, API design choices, and architectural trade-offs.
Ignore meeting logistics, greetings, and social exchanges.
The mission is injected into the extraction prompt alongside the built-in rules — it steers the LLM without replacing the extraction logic. It works with any extraction mode (concise, verbose, custom).
For finer control, you can also change the extraction mode:
| Mode | When to use |
|---|---|
concise (default) |
General-purpose — selective, fast |
verbose |
When you need richer facts with full context and relationships |
custom |
When you want to write your own extraction rules entirely |
Set retain_mission and retain_extraction_mode via the bank config API or the HINDSIGHT_API_RETAIN_MISSION environment variable.
Entity Labels
Entity labels let you define a controlled vocabulary of classification labels that are extracted at retain time and stored as entities alongside regular named entities. Each label takes the form key:value (e.g. pedagogy:scaffolding, engagement:active).
Because labels become entities, they automatically:
- Appear in the knowledge graph — two memories with
pedagogy:scaffoldingare linked - Improve semantic and BM25 retrieval — label strings are included in both the dense embedding and the sparse
text_signalsfield - Support labels-only mode — optionally disable free-form entity extraction so only labels are stored
Labels are configured per bank via entity_labels in the bank config.
Defining Label Groups
Each label group defines one classification dimension:
{
"entity_labels": [
{
"key": "engagement",
"description": "Student engagement level during the session",
"type": "value",
"optional": true,
"values": [
{ "value": "active", "description": "Student is actively participating" },
{ "value": "passive", "description": "Student is listening but not participating" }
]
},
{
"key": "pedagogy",
"description": "Teaching strategies used",
"type": "multi-values",
"values": [
{ "value": "scaffolding", "description": "Breaking complex tasks into smaller steps" },
{ "value": "direct_instruction", "description": "Explicit explanation by the teacher" },
{ "value": "socratic_questioning", "description": "Guiding through questions rather than answers" }
]
}
]
}
| Field | Default | Description |
|---|---|---|
key |
— | Label group identifier. Becomes the prefix in key:value entities. |
description |
"" |
Shown to the LLM to help it assign the right label. |
type |
"value" |
"value" → single enum value; "multi-values" → multiple enum values; "text" → free-form string. |
values |
[] |
Allowed values for "value" and "multi-values" types. Ignored for "text" type. |
optional |
true |
true → the LLM may skip this label if not applicable (default). false → LLM must always assign a value. Has no effect on "multi-values" groups (always optional). |
tag |
false |
true → also write extracted key:value entities as tags on the memory unit, enabling filtering via the standard tags/tags_match API parameters. |
Enum vs Free-text Labels
Enum groups (type: "value" or type: "multi-values"): the LLM must pick from the predefined values list. Values not in the list are silently dropped. This is the most reliable option — the vocabulary is stable and graph clustering is tight. Use "multi-values" when a single fact can match multiple values.
Free-text groups (type: "text"): the LLM can write any string value. The values field is ignored — use the description to provide examples and guidance instead.
{
"key": "topic",
"description": "The specific subject being discussed. Examples: algebra, geometry, quadratic equations.",
"type": "text",
"optional": true,
"values": []
}
The trade-off with free-text: the LLM may use different phrasings for the same concept across sessions (topic:fractions vs topic:fraction arithmetic), so graph linking is less reliable than with enum groups.
Labels-only Mode
By default, entity labels are extracted alongside regular named entities (people, places, concepts). Set entities_allow_free_form: false to disable free-form extraction and store only label entities:
{
"entity_labels": [...],
"entities_allow_free_form": false
}
Configure both via the bank config API.
Observation Consolidation
After retain() completes, Hindsight automatically triggers observation consolidation in the background. This process:
- Analyzes new facts against existing observations
- Creates new observations when patterns emerge
- Refines existing observations with new evidence
- Tracks which facts support each observation
This happens asynchronously — your retain() call returns immediately while consolidation runs in the background.
See Observations for details on how consolidation works.
Next Steps
- Observations — How knowledge is consolidated after retain
- Recall — How multi-strategy search retrieves relevant memories
- Reflect — How the agentic loop uses observations
- Retain API — Code examples and parameters