graph

type Entity struct {
	Name         string   `json:"name"`
	EntityType   string   `json:"entityType"`
	Observations []string `json:"observations"`
}

type GraphRepository interface {
	CreateEntities([]Entity) ([]Entity, error)
	CreateRelations([]Relation) ([]Relation, error)
	AddObservations([]Observation) ([]Observation, error)
	DeleteEntities(entityNames []string) error
	DeleteObservations(deletions []Observation) error
	DeleteRelations(relations []Relation) error
	ReadGraph() (KnowledgeGraph, error)
	SearchNodes(query string) (KnowledgeGraph, error)
	SearchNodesFiltered(query string, includeArchived bool) (KnowledgeGraph, error)
	OpenNodes(names []string) (KnowledgeGraph, error)
	OpenNodesFiltered(names []string, includeArchived bool) (KnowledgeGraph, error)
	EntityCount() (int64, error)
	EntityTypeCounts() (map[string]int64, error)
	ListEntities(entityType string) (KnowledgeGraph, error)
	ListRelations(relationType, fromEntity string) ([]Relation, error)
	TraverseGraph(entity, relationType, direction string, maxDepth int) ([]TraverseNode, []TraverseEdge, error)
	GetIntegrationMap(ctx context.Context, service string, depth int) (IntegrationMap, error)
	FindPath(from, to string, maxDepth int) ([]PathEdge, error)
	UpdateEntity(oldName, newName, newType string) error
}

type GraphService interface {
	GraphRepository
}

type IntegrationEdge struct {
	Target     string `json:"target"`
	Schema     string `json:"schema,omitempty"`      // event-topic schema name
	Version    string `json:"version,omitempty"`     // api version
	Paths      int    `json:"paths,omitempty"`       // api path count
	Confidence string `json:"confidence"`            // "authoritative" | "inferred"
	SourceRepo string `json:"source_repo,omitempty"` // originating repo
}

type IntegrationMap struct {
	Service      string            `json:"service"`
	Publishes    []IntegrationEdge `json:"publishes"`
	Subscribes   []IntegrationEdge `json:"subscribes"`
	ExposesAPI   []IntegrationEdge `json:"exposes_api"`
	ProvidesGRPC []IntegrationEdge `json:"provides_grpc"`
	GRPCDeps     []IntegrationEdge `json:"grpc_deps"`
	Calls        []IntegrationEdge `json:"calls"`
	Coverage     string            `json:"graph_coverage"` // "full" | "partial" | "inferred" | "none"
}

type KnowledgeGraph struct {
	Entities  []Entity   `json:"entities"`
	Relations []Relation `json:"relations"`
}

type MemoryService struct {
	// contains filtered or unexported fields
}

type Observation struct {
	EntityName   string   `json:"entityName"`
	Contents     []string `json:"contents"`
	Observations []string `json:"observations,omitempty"` // For deletion
}

type PathEdge struct {
	From         string `json:"from"`
	RelationType string `json:"relationType"`
	To           string `json:"to"`
	Confidence   string `json:"confidence,omitempty"`
}

type Relation struct {
	From         string `json:"from"`
	To           string `json:"to"`
	RelationType string `json:"relationType"`
	// Confidence indicates how the relation was derived.
	// "authoritative" — from an explicit contract file.
	// "inferred"      — from a config heuristic.
	// "ambiguous"     — caller explicitly marked the edge as uncertain.
	Confidence string `json:"confidence,omitempty"`
}

type TraverseEdge struct {
	From         string `json:"from"`
	To           string `json:"to"`
	RelationType string `json:"relationType"`
	Confidence   string `json:"confidence,omitempty"`
}

type TraverseNode struct {
	Name         string   `json:"name"`
	EntityType   string   `json:"entityType"`
	Observations []string `json:"observations"`
	Distance     int      `json:"distance"`
	Path         []string `json:"path"` // entity names from start (exclusive) to this node (inclusive)
}