graph

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"`
	// 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.
	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 GetEntityRequest struct {
	EntityID string `json:"entity_id" jsonschema:"description=ID of the entity to look up,required"`
}

type GetEntityResponse struct {
	Entity *Entity `json:"entity,omitempty" jsonschema:"description=The entity if found"`
	Found  bool    `json:"found" jsonschema:"description=Whether the entity exists"`
}

type GraphQueryRequest struct {
	EntityID string `json:"entity_id" jsonschema:"description=ID of the entity to query from,required"`
	Limit    int    `json:"limit,omitempty" jsonschema:"description=Max number of neighbors to return (default 20)"`
}

type GraphQueryResponse struct {
	Neighbors []Neighbor `json:"neighbors"`
	Count     int        `json:"count"`
}

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 ShortestPathRequest struct {
	SourceID string `json:"source_id" jsonschema:"description=ID of the starting entity,required"`
	TargetID string `json:"target_id" jsonschema:"description=ID of the destination entity,required"`
}

type ShortestPathResponse struct {
	Path  []string `json:"path" jsonschema:"description=Ordered list of entity IDs from source to target, empty if no path"`
	Found bool     `json:"found" jsonschema:"description=Whether a path exists"`
}

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

type StoreEntityResponse struct {
	Message string `json:"message"`
}

type StoreRelationRequest struct {
	SubjectID string `json:"subject_id" jsonschema:"description=ID of the subject entity,required"`
	Predicate string `json:"predicate" jsonschema:"description=Relation type (e.g. WORKED_ON, OWNS, MENTIONS),required"`
	ObjectID  string `json:"object_id" jsonschema:"description=ID of the object entity,required"`
}

type StoreRelationResponse struct {
	Message string `json:"message"`
}

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