memory

package
v0.0.4 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Jul 28, 2025 License: AGPL-3.0 Imports: 19 Imported by: 0

Documentation

Overview

internal/memory/k8s_manager.go

internal/memory/mcp_server.go

internal/memory/mcp_tools.go

internal/memory/memory_store.go

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

This section is empty.

Types

type Entity 

type Entity struct {
	Name         string    `json:"name"`
	EntityType   string    `json:"entityType"`
	Observations []string  `json:"observations"`
	CreatedAt    time.Time `json:"createdAt"`
	UpdatedAt    time.Time `json:"updatedAt"`
}

Entity represents a named entity in the knowledge graph

type K8sManager 

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

K8sManager is a system manager for the memory service

func NewK8sManager 

func NewK8sManager(cfg *config.ComposeConfig, k8sClient client.Client, namespace string) *K8sManager

NewK8sManager creates a new system memory manager

func NewManagerFromConfig 

func NewManagerFromConfig(cfg *config.ComposeConfig, k8sClient client.Client, namespace string) *K8sManager

Helper function to convert the old Manager interface to the new K8sManager

func (*K8sManager) GetLogs 

func (m *K8sManager) GetLogs() (string, error)

GetLogs returns logs from the memory service (placeholder for future implementation)

func (*K8sManager) GetMemoryInfo 

func (m *K8sManager) GetMemoryInfo() (*MemoryInfo, error)

GetMemoryInfo returns detailed information about the memory service

func (*K8sManager) Restart 

func (m *K8sManager) Restart() error

Restart restarts the memory service

func (*K8sManager) SetConfigFile 

func (m *K8sManager) SetConfigFile(configFile string)

SetConfigFile sets the configuration file path

func (*K8sManager) Start 

func (m *K8sManager) Start() error

Start starts the memory service using Kubernetes resources

func (*K8sManager) Status 

func (m *K8sManager) Status() (string, error)

Status returns the status of the memory service

func (*K8sManager) Stop 

func (m *K8sManager) Stop() error

Stop stops the memory service

type KnowledgeGraph 

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

KnowledgeGraph represents the complete graph structure

type MCPContent 

type MCPContent struct {
	Type string `json:"type"`
	Text string `json:"text"`
}

MCPContent represents MCP content

type MCPError 

type MCPError struct {
	Code    int         `json:"code"`
	Message string      `json:"message"`
	Data    interface{} `json:"data,omitempty"`
}

MCPError represents an MCP protocol error

type MCPMemoryServer 

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

MCPMemoryServer provides HTTP/SSE/WebSocket endpoints for memory MCP tools

func NewMCPMemoryServer 

func NewMCPMemoryServer(store *MemoryStore, logger logr.Logger) *MCPMemoryServer

func (*MCPMemoryServer) BroadcastToClients 

func (s *MCPMemoryServer) BroadcastToClients(message interface{})

BroadcastToClients sends a message to all connected WebSocket clients

func (*MCPMemoryServer) GetClientIDs 

func (s *MCPMemoryServer) GetClientIDs() []string

GetClientIDs returns a list of connected client IDs

func (*MCPMemoryServer) GetServerInfo 

func (s *MCPMemoryServer) GetServerInfo() map[string]interface{}

GetServerInfo returns information about the MCP server

func (*MCPMemoryServer) Start 

func (s *MCPMemoryServer) Start(host string, port int) error

func (*MCPMemoryServer) Stop 

func (s *MCPMemoryServer) Stop() error

type MCPMemoryTools 

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

MCPMemoryTools provides MCP protocol tools for memory management

func NewMCPMemoryTools 

func NewMCPMemoryTools(store *MemoryStore, logger logr.Logger) *MCPMemoryTools

func (*MCPMemoryTools) ExecuteMCPTool 

func (mt *MCPMemoryTools) ExecuteMCPTool(ctx context.Context, toolName string, parameters map[string]interface{}) (map[string]interface{}, error)

ExecuteMCPTool executes an MCP tool with the given parameters

func (*MCPMemoryTools) GetMCPTools 

func (mt *MCPMemoryTools) GetMCPTools() []MCPToolDefinition

GetMCPTools returns all available memory MCP tools

type MCPRequest 

type MCPRequest struct {
	JSONRPC string                 `json:"jsonrpc"`
	ID      interface{}            `json:"id"`
	Method  string                 `json:"method"`
	Params  map[string]interface{} `json:"params,omitempty"`
}

MCPRequest represents an MCP protocol request

type MCPResponse 

type MCPResponse struct {
	JSONRPC string      `json:"jsonrpc"`
	ID      interface{} `json:"id"`
	Result  interface{} `json:"result,omitempty"`
	Error   *MCPError   `json:"error,omitempty"`
}

MCPResponse represents an MCP protocol response

type MCPToolCallRequest 

type MCPToolCallRequest struct {
	Name      string                 `json:"name"`
	Arguments map[string]interface{} `json:"arguments,omitempty"`
}

MCPToolCallRequest represents a tool call request

type MCPToolCallResponse 

type MCPToolCallResponse struct {
	Content []MCPContent `json:"content"`
}

MCPToolCallResponse represents a tool call response

type MCPToolDefinition 

type MCPToolDefinition struct {
	Name        string                 `json:"name"`
	Description string                 `json:"description"`
	InputSchema map[string]interface{} `json:"inputSchema"`
}

MCPToolDefinition represents an MCP tool definition for memory

type MCPToolsListResponse 

type MCPToolsListResponse struct {
	Tools []MCPToolDefinition `json:"tools"`
}

MCPToolsListResponse represents the response to tools/list

type MemoryInfo 

type MemoryInfo struct {
	Status         string   `json:"status"`
	ReadyReplicas  int32    `json:"readyReplicas"`
	TotalReplicas  int32    `json:"totalReplicas"`
	PostgresStatus string   `json:"postgresStatus"`
	Conditions     []string `json:"conditions"`
	Endpoint       string   `json:"endpoint,omitempty"`
}

MemoryInfo represents detailed information about the memory service

type MemoryStore 

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

MemoryStore provides PostgreSQL-backed knowledge graph storage

func NewMemoryStore 

func NewMemoryStore(databaseURL string, logger logr.Logger) (*MemoryStore, error)

func (*MemoryStore) AddObservations 

func (ms *MemoryStore) AddObservations(observations map[string][]string) error

AddObservations adds new observations to existing entities

func (*MemoryStore) Close 

func (ms *MemoryStore) Close() error

func (*MemoryStore) CreateEntities 

func (ms *MemoryStore) CreateEntities(entities []Entity) error

CreateEntities creates multiple entities in the knowledge graph

func (*MemoryStore) CreateRelations 

func (ms *MemoryStore) CreateRelations(relations []Relation) error

CreateRelations creates typed relationships between entities

func (*MemoryStore) DeleteEntities 

func (ms *MemoryStore) DeleteEntities(entityNames []string) error

DeleteEntities deletes entities and their associated data

func (*MemoryStore) DeleteObservations 

func (ms *MemoryStore) DeleteObservations(deletions map[string][]string) error

DeleteObservations removes specific observations from entities

func (*MemoryStore) DeleteRelations 

func (ms *MemoryStore) DeleteRelations(relations []Relation) error

DeleteRelations removes specific relationships

func (*MemoryStore) GetStats 

func (ms *MemoryStore) GetStats() (map[string]interface{}, error)

GetStats returns statistics about the knowledge graph

func (*MemoryStore) HealthCheck 

func (ms *MemoryStore) HealthCheck() error

HealthCheck verifies database connectivity

func (*MemoryStore) OpenNodes 

func (ms *MemoryStore) OpenNodes(names []string) ([]Entity, error)

OpenNodes retrieves specific nodes by their names

func (*MemoryStore) ReadGraph 

func (ms *MemoryStore) ReadGraph() (*KnowledgeGraph, error)

ReadGraph retrieves the entire knowledge graph

func (*MemoryStore) SearchNodes 

func (ms *MemoryStore) SearchNodes(query string) ([]SearchResult, error)

SearchNodes searches for nodes using full-text search

type Relation 

type Relation struct {
	From         string    `json:"from"`
	To           string    `json:"to"`
	RelationType string    `json:"relationType"`
	CreatedAt    time.Time `json:"createdAt"`
}

Relation represents a typed relationship between entities

type SearchResult 

type SearchResult struct {
	Entity    Entity   `json:"entity"`
	Relevance float64  `json:"relevance"`
	Matches   []string `json:"matches"`
}

SearchResult represents search results with relevance

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.