graph

type BatchQuery struct {
	Action   string `` /* 235-byte string literal not displayed */
	EntityID string `json:"entity_id,omitempty" jsonschema:"description=Entity ID to query (required for explore and neighbors and get_entity)"`
	Limit    int    `json:"limit,omitempty" jsonschema:"description=Max neighbors to return (default 20)"`
	Depth    int    `json:"depth,omitempty" jsonschema:"description=Traversal depth for explore: 1 (default) or 2. Max 2"`
	SourceID string `json:"source_id,omitempty" jsonschema:"description=Source entity ID for shortest_path"`
	TargetID string `json:"target_id,omitempty" jsonschema:"description=Target entity ID for shortest_path"`
}

type BatchResult struct {
	Index    int                 `json:"index"`
	Action   string              `json:"action"`
	Response *GraphQueryResponse `json:"response,omitempty"`
	Error    string              `json:"error,omitempty"`
}

type BatchStoreItem struct {
	Action string `` /* 144-byte string literal not displayed */

	// Fields for action=entity
	ID    string            `json:"id,omitempty" jsonschema:"description=Entity ID (required when action=entity)"`
	Type  string            `json:"type,omitempty" jsonschema:"description=Entity type (required when action=entity)"`
	Attrs map[string]string `json:"attrs,omitempty" jsonschema:"description=Optional key-value attributes"`

	// Fields for action=relation
	SubjectID string `json:"subject_id,omitempty" jsonschema:"description=Subject entity ID (required when action=relation)"`
	Predicate string `json:"predicate,omitempty" jsonschema:"description=Relation type (required when action=relation)"`
	ObjectID  string `json:"object_id,omitempty" jsonschema:"description=Object entity ID (required when action=relation)"`
}

type BatchStoreResult struct {
	Index   int    `json:"index"`
	Action  string `json:"action"`
	Message string `json:"message,omitempty"`
	Error   string `json:"error,omitempty"`
}

type Config struct {
	// Disabled turns off the knowledge graph and graph_* tools. When true, no
	// graph store is created and no graph tools are registered.
	Disabled bool `yaml:"disabled,omitempty" toml:"disabled,omitempty"`
	// Backend selects the knowledge graph storage backend. Options:
	//   "inmemory" (default) — in-process dominikbraun/graph with gob+zstd snapshot persistence.
	//   "vectorstore"        — reuses the configured vector store (Qdrant); entities and
	//                          relations are stored as vector documents with metadata discriminators.
	// When Backend is "vectorstore", DataDir is ignored (persistence is handled by the vector store).
	Backend string `yaml:"backend,omitempty" toml:"backend,omitempty"`
	// DataDir is the directory for persistence (e.g. ~/.genie/<agent>). The
	// in-memory implementation writes memory.bin.zst (gob+zstd) here. Empty means no persistence.
	// Ignored when Backend is "vectorstore".
	DataDir string `yaml:"data_dir,omitempty" toml:"data_dir,omitempty"`
}

type Entity struct {
	ID    string            `json:"id"`
	Type  string            `json:"type"`
	Attrs map[string]string `json:"attrs,omitempty"`
}

type GraphQueryRequest struct {
	Action string `` /* 496-byte string literal not displayed */

	// Fields for action=neighbors, get_entity, and explore
	EntityID string `` /* 165-byte string literal not displayed */
	Limit    int    `` /* 132-byte string literal not displayed */

	// Fields for action=explore
	Depth int `` /* 178-byte string literal not displayed */

	// Fields for action=shortest_path
	SourceID string `json:"source_id,omitempty" jsonschema:"description=Starting entity ID for shortest_path"`
	TargetID string `json:"target_id,omitempty" jsonschema:"description=Destination entity ID for shortest_path"`

	// Fields for action=batch
	Queries []BatchQuery `` /* 219-byte string literal not displayed */
}

type GraphQueryResponse struct {
	// For action=neighbors
	Neighbors []Neighbor `json:"neighbors,omitempty"`
	Count     int        `json:"count,omitempty"`

	// For action=get_entity
	Entity *Entity `json:"entity,omitempty"`

	// Shared: true when the query matched data, false when entity not found
	Found bool `json:"found"`

	// For action=shortest_path
	Path []string `json:"path,omitempty"`

	// For action=explore — contains root entity, all connected entities,
	// their relations, and neighbor details in a single response.
	Subgraph *Subgraph `json:"subgraph,omitempty"`

	// For action=batch — one result per sub-query, in the same order.
	BatchResults []BatchResult `json:"batch_results,omitempty"`
}

type GraphStoreRequest struct {
	Action string `` /* 170-byte string literal not displayed */

	// Fields for action=entity
	ID    string            `json:"id,omitempty" jsonschema:"description=Unique identifier for the entity (required when action=entity)"`
	Type  string            `json:"type,omitempty" jsonschema:"description=Entity type e.g. person repo issue document (required when action=entity)"`
	Attrs map[string]string `json:"attrs,omitempty" jsonschema:"description=Optional key-value attributes for the entity"`

	// Fields for action=relation
	SubjectID string `json:"subject_id,omitempty" jsonschema:"description=ID of the subject entity (required when action=relation)"`
	Predicate string `` /* 126-byte string literal not displayed */
	ObjectID  string `json:"object_id,omitempty" jsonschema:"description=ID of the object entity (required when action=relation)"`

	// Fields for action=batch
	Items []BatchStoreItem `` /* 213-byte string literal not displayed */
}

type GraphStoreResponse struct {
	Message      string             `json:"message"`
	BatchResults []BatchStoreResult `json:"batch_results,omitempty"`
}

type IStore interface {
	// AddEntity stores an entity by id; overwrites if id exists.
	AddEntity(ctx context.Context, e Entity) error
	// AddRelation stores a directed relation; idempotent (same triple is a no-op).
	AddRelation(ctx context.Context, r Relation) error
	// GetEntity returns the entity by id, or nil if not found.
	GetEntity(ctx context.Context, id string) (*Entity, error)
	// RelationsOut returns relations where subject_id equals id (outgoing edges).
	RelationsOut(ctx context.Context, id string) ([]Relation, error)
	// RelationsIn returns relations where object_id equals id (incoming edges).
	RelationsIn(ctx context.Context, id string) ([]Relation, error)
	// Neighbors returns entities reachable in one hop from id (outgoing and
	// incoming), with the connecting predicate and direction. Limit caps the
	// total number of neighbors returned.
	Neighbors(ctx context.Context, id string, limit int) ([]Neighbor, error)
	// ShortestPath returns the shortest path of entity IDs from source to target.
	// Returns nil path and nil error if no path exists. Returns an error if
	// source or target vertex does not exist. Not all implementations may support
	// this; they may return an error.
	ShortestPath(ctx context.Context, sourceID, targetID string) ([]string, error)
	// Close releases resources (e.g. flush snapshot, close connection).
	Close(ctx context.Context) error
}

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

type InMemoryStoreOption func(*InMemoryStore)

type Neighbor struct {
	Entity    Entity `json:"entity"`
	Predicate string `json:"predicate"`
	Outgoing  bool   `json:"outgoing"` // true = from subject to this entity, false = from this entity to object
}

type Relation struct {
	SubjectID string `json:"subject_id"`
	Predicate string `json:"predicate"`
	ObjectID  string `json:"object_id"`
}

type Subgraph struct {
	// Root is the central entity that was explored.
	Root Entity `json:"root"`
	// Entities are all other entities discovered during traversal (excludes root).
	Entities []Entity `json:"entities"`
	// Relations are all directed edges (subject→object) involving the root.
	Relations []Relation `json:"relations"`
	// Neighbors provides the same data as action=neighbors would, with
	// predicate and direction info for each connected entity.
	Neighbors []Neighbor `json:"neighbors"`
}

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

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