feat(cli): add verbose logging option for CLI commands
Some checks failed
CI / build-and-test (push) Failing after -32m43s
Some checks failed
CI / build-and-test (push) Failing after -32m43s
* Introduced a new flag `--verbose` to enable detailed logging. * Implemented logging for connection events in SSE and stdio commands. * Added a utility function to handle verbose logging.
This commit is contained in:
@@ -55,24 +55,41 @@ func NewBackfillTool(db *store.DB, embeddings *ai.EmbeddingRunner, sessions *ses
|
||||
// It is used by capture when the embedding provider is temporarily unavailable.
|
||||
func (t *BackfillTool) QueueThought(ctx context.Context, id uuid.UUID, content string) {
|
||||
go func() {
|
||||
started := time.Now()
|
||||
t.logger.Info("background embedding started",
|
||||
slog.String("thought_id", id.String()),
|
||||
slog.String("provider", t.embeddings.PrimaryProvider()),
|
||||
slog.String("model", t.embeddings.PrimaryModel()),
|
||||
)
|
||||
|
||||
result, err := t.embeddings.Embed(ctx, content)
|
||||
if err != nil {
|
||||
t.logger.Warn("background embedding retry failed",
|
||||
t.logger.Warn("background embedding error",
|
||||
slog.String("thought_id", id.String()),
|
||||
slog.String("provider", t.embeddings.PrimaryProvider()),
|
||||
slog.String("model", t.embeddings.PrimaryModel()),
|
||||
slog.String("stage", "embed"),
|
||||
slog.Duration("duration", time.Since(started)),
|
||||
slog.String("error", err.Error()),
|
||||
)
|
||||
return
|
||||
}
|
||||
if err := t.store.UpsertEmbedding(ctx, id, result.Model, result.Vector); err != nil {
|
||||
t.logger.Warn("background embedding upsert failed",
|
||||
t.logger.Warn("background embedding error",
|
||||
slog.String("thought_id", id.String()),
|
||||
slog.String("provider", t.embeddings.PrimaryProvider()),
|
||||
slog.String("model", result.Model),
|
||||
slog.String("stage", "upsert"),
|
||||
slog.Duration("duration", time.Since(started)),
|
||||
slog.String("error", err.Error()),
|
||||
)
|
||||
return
|
||||
}
|
||||
t.logger.Info("background embedding retry succeeded",
|
||||
t.logger.Info("background embedding complete",
|
||||
slog.String("thought_id", id.String()),
|
||||
slog.String("provider", t.embeddings.PrimaryProvider()),
|
||||
slog.String("model", result.Model),
|
||||
slog.Duration("duration", time.Since(started)),
|
||||
)
|
||||
}()
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user