persistence

type Checkpoint struct {
	ID        string                 `json:"id"`
	ThreadID  string                 `json:"thread_id"`
	State     *core.BaseState        `json:"state"`
	Metadata  map[string]interface{} `json:"metadata"`
	CreatedAt time.Time              `json:"created_at"`
	NodeID    string                 `json:"node_id"`
	StepID    int                    `json:"step_id"`
}

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

type CheckpointMetadata struct {
	ID        string                 `json:"id"`
	ThreadID  string                 `json:"thread_id"`
	Metadata  map[string]interface{} `json:"metadata"`
	CreatedAt time.Time              `json:"created_at"`
	NodeID    string                 `json:"node_id"`
	StepID    int                    `json:"step_id"`
}

type Checkpointer interface {
	// Save saves a state checkpoint
	Save(ctx context.Context, checkpoint *Checkpoint) error

	// Load loads a state checkpoint
	Load(ctx context.Context, threadID, checkpointID string) (*Checkpoint, error)

	// List lists checkpoints for a thread
	List(ctx context.Context, threadID string) ([]*CheckpointMetadata, error)

	// Delete deletes a checkpoint
	Delete(ctx context.Context, threadID, checkpointID string) error

	// Close closes the checkpointer
	Close() error
}

type DatabaseConfig struct {
	Type         DatabaseType `json:"type"` // "postgres", "pgvector", "redis", "opensearch", etc.
	Host         string       `json:"host"`
	Port         int          `json:"port"`
	Database     string       `json:"database"`
	Username     string       `json:"username"`
	Password     string       `json:"password"`
	SSLMode      string       `json:"ssl_mode"`
	MaxOpenConns int          `json:"max_open_conns"`
	MaxIdleConns int          `json:"max_idle_conns"`
	MaxLifetime  string       `json:"max_lifetime"`

	// Vector-specific configuration
	VectorDimension int    `json:"vector_dimension"`
	VectorMetric    string `json:"vector_metric"` // "cosine", "euclidean", "dot_product"

	// OpenSearch/Elasticsearch specific
	Index   string `json:"index"`
	APIKey  string `json:"api_key"`
	CloudID string `json:"cloud_id"`
	CACert  string `json:"ca_cert"`

	// Additional connection parameters
	ConnectionParams map[string]string `json:"connection_params"`

	// RAG-specific settings
	EnableRAG           bool    `json:"enable_rag"`
	EmbeddingModel      string  `json:"embedding_model"`
	EmbeddingDimension  int     `json:"embedding_dimension"`
	SimilarityThreshold float64 `json:"similarity_threshold"`
}

type DatabaseConnection interface {
	Connect() error
	Close() error
	Ping() error
	GetType() DatabaseType
	GetConfig() *DatabaseConfig
	ExecuteQuery(ctx context.Context, query string, args ...interface{}) error
	QueryRow(ctx context.Context, query string, args ...interface{}) interface{}
	QueryRows(ctx context.Context, query string, args ...interface{}) (interface{}, error)
}

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

type DatabaseType string

type Document struct {
	ID        string                 `json:"id"`
	ThreadID  string                 `json:"thread_id"`
	Content   string                 `json:"content"`
	Metadata  map[string]interface{} `json:"metadata"`
	Embedding []float64              `json:"embedding,omitempty"`
	CreatedAt time.Time              `json:"created_at"`
	UpdatedAt time.Time              `json:"updated_at"`
}

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

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

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

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

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

type Session struct {
	ID        string                 `json:"id"`
	ThreadID  string                 `json:"thread_id"`
	UserID    string                 `json:"user_id"`
	Metadata  map[string]interface{} `json:"metadata"`
	CreatedAt time.Time              `json:"created_at"`
	ExpiresAt *time.Time             `json:"expires_at,omitempty"`
}

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

type Thread struct {
	ID        string                 `json:"id"`
	Name      string                 `json:"name"`
	Metadata  map[string]interface{} `json:"metadata"`
	CreatedAt time.Time              `json:"created_at"`
	UpdatedAt time.Time              `json:"updated_at"`
}

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