storage

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

type Journey struct {
	SessionID string
	StepNo    int
	Kind      string
	Duration  time.Duration
	Messages  []llm.Message
	Metadata  any
}

type NoneStorage struct {
}

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

type Session struct {
	UID        string         `json:"uid" gorm:"column:uid;type:varchar(255);"`                 // session id
	EngineID   string         `json:"engine_id" gorm:"column:engine_id;type:varchar(255);"`     // engine id
	EngineType string         `json:"engine_type" gorm:"column:engine_type;type:varchar(255);"` // engine type
	EngineData map[string]any `json:"engine_data" gorm:"column:engine_data;type:jsonb"`         // engine data

	Memory memory.Memory `json:"memory"` // 记忆

	Messages []llm.Message `json:"messages"`

	Data any `json:"data"`

	CreatedAt time.Time `json:"created_at" gorm:"column:created_at;autoCreateTime"`
	UpdatedAt time.Time `json:"updated_at" gorm:"column:updated_at;autoUpdateTime"`
}

type Storage interface {
	AddNewJourney(ctx context.Context, journey *Journey) error
	SaveSession(ctx context.Context, session *Session) error
	FetchSession(ctx context.Context, id string) (*Session, error)
	UpdateSession(ctx context.Context, session *Session) error
	DeleteSession(ctx context.Context, id string) error
}