raystack / compass / 23716887782

push

github

feat: entity system and temporal graph with Postgres-native search (#249)

* feat: add v2 entity system with Postgres-native search

New entity system built as an independent package alongside the existing
asset system. Zero changes to existing asset code. All search is
Postgres-native — no Elasticsearch dependency for v2.

Core domain (core/entity/):
- Entity: open type system, temporal (valid_from/valid_to), properties JSONB
- Edge: typed, directed, temporal relationships
- Chunk: pgvector embeddings for semantic search
- Service: context assembly, impact analysis, search orchestration
- HybridSearch: keyword + semantic fusion via Reciprocal Rank Fusion
- SearchRepository: Postgres tsvector + pg_trgm (replaces ES)

Storage (store/postgres/):
- EntityRepository: CRUD with temporal support
- EdgeRepository: recursive CTE graph traversal
- ChunkRepository: pgvector cosine similarity search
- EntitySearchRepository: tsvector full-text + pg_trgm fuzzy matching
- Migration 000019: entities, edges, chunks tables with search_vector
  generated column, GIN indexes, trigram indexes

Search architecture (all Postgres):
- Keyword: tsvector with weighted fields (URN/name=A, desc=B, source=C)
- Fuzzy: pg_trgm similarity with automatic fallback
- Semantic: pgvector cosine distance on chunk embeddings
- Hybrid: RRF fusion of keyword + semantic results
- Suggest: pg_trgm similarity-based autocomplete

MCP tools (internal/mcp/):
- search_entities: hybrid search across the entity graph
- get_context: assembled context subgraph
- impact: downstream blast radius analysis

* feat: add entity v2 handler with gRPC/REST endpoints

Generated Go code from proton proto definitions for entity v2 RPCs.
Created handler/entity.go implementing all 12 new RPC methods:

Entity CRUD: GetAllEntities, GetEntityByID, UpsertEntity, DeleteEntity
Search: SearchEntities, SuggestEntities, GetEntityTypes
Context: GetEntityContext, GetEntityImpact
Edges: UpsertEdge, GetEdges, DeleteEdge

Handle... (continued)

62 of 2499 new or added lines in 26 files covered. (2.48%)

189 of 8326 relevant lines covered (2.27%)

0.07 hits per line