search

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

type HybridConfig struct {
	Strategy string
	RRFK     int
	Weights  HybridWeights
}

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

type HybridWeights struct {
	Keyword float64
	Vector  float64
}

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

type KeywordStore interface {
	MessageSource
	SearchMessagesByKeywords(ctx context.Context, q *SearchQuery) ([]*Message, error)
}

type Message struct {
	ID        string
	SessionID string
	UserID    string
	Role      string
	Content   string
	Parts     []schema.MessageInputPart
	CreatedAt time.Time
}

type MessageSource interface {
	ListMessages(ctx context.Context, sessionID, userID string) ([]*Message, error)
}

type SearchHit struct {
	Message *Message
	Score   float64
	Snippet string
}

type SearchMode string

type SearchQuery struct {
	SessionID string
	UserID    string

	Since *time.Time
	Until *time.Time
	Role  string
	Limit int

	Keywords []string
	Match    string

	Query string
}

type Searcher interface {
	Search(ctx context.Context, q *SearchQuery) ([]*SearchHit, error)
	Index(ctx context.Context, msg *Message) error
	Reindex(ctx context.Context, sessionID, userID string) error
}

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

type VectorStore interface {
	Upsert(ctx context.Context, msg *Message, vector []float64) error
	Search(ctx context.Context, q *SearchQuery, vector []float64, limit int) ([]*SearchHit, error)
}