Radu Macocian (admac)
88 lines
2.1 KiB
Go
88 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"log/slog"
|
|
"os"
|
|
"time"
|
|
|
|
tea "charm.land/bubbletea/v2"
|
|
"git.estatecloud.org/radumaco/souvenir/config"
|
|
"git.estatecloud.org/radumaco/souvenir/db/embed"
|
|
"git.estatecloud.org/radumaco/souvenir/db/history"
|
|
"git.estatecloud.org/radumaco/souvenir/llm"
|
|
ui "git.estatecloud.org/radumaco/souvenir/ui"
|
|
)
|
|
|
|
func main() {
|
|
ctx := context.Background()
|
|
config, err := config.Load(".config.json")
|
|
if err != nil {
|
|
slog.Error("Config error:", "message", err.Error())
|
|
os.Exit(1)
|
|
}
|
|
var logger = slog.Default().With("Component", "Main")
|
|
logger.Debug("Config initialized")
|
|
|
|
db, err := history.Init(ctx, config.Db)
|
|
if err != nil {
|
|
logger.Error("Error while initializing database", "message", err)
|
|
os.Exit(1)
|
|
}
|
|
defer db.Close()
|
|
es, _ := setupEmbedding(config, logger)
|
|
if es != nil {
|
|
defer es.Close()
|
|
}
|
|
|
|
p := tea.NewProgram(ui.InitialModel(ctx, *config, *db))
|
|
if _, err := p.Run(); err != nil {
|
|
logger.Error("Alas, there's been an error:", "message", err)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func setupEmbedding(cfg *config.Config, logger *slog.Logger) (*embed.EmbedStore, error) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
embedder := llm.NewEmbedder(cfg.Embedding)
|
|
embedStore, err := embed.NewEmbedStore(ctx, cfg.Db, *embedder)
|
|
if err != nil {
|
|
logger.Error("Could not initialize embedder", "error", err)
|
|
return nil, err
|
|
}
|
|
ticker := time.NewTicker(time.Duration(cfg.Embedding.Interval) * time.Minute)
|
|
go func() {
|
|
defer ticker.Stop()
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
return
|
|
case <-ticker.C:
|
|
runEmbedding(ctx, embedStore, logger)
|
|
}
|
|
}
|
|
}()
|
|
return embedStore, nil
|
|
}
|
|
|
|
func runEmbedding(ctx context.Context, em *embed.EmbedStore, logger *slog.Logger) error {
|
|
convs, err := em.GetConversationsToEmbed(ctx)
|
|
if err != nil {
|
|
logger.Error("Could not get conversations to embed", "error", err)
|
|
return err
|
|
}
|
|
var errs []error
|
|
for _, conv := range convs {
|
|
if err := em.EmbedConversation(ctx, conv); err != nil {
|
|
logger.Error("Could not embed conversation", "conversation_id", conv, "error", err)
|
|
errs = append(errs, err)
|
|
}
|
|
}
|
|
if errs != nil {
|
|
return errors.Join(errs...)
|
|
}
|
|
return nil
|
|
}
|