From cf77aa8c22da236f87f1906764f3dbd788e33137 Mon Sep 17 00:00:00 2001 From: Tyler Rockwood Date: Mon, 9 Sep 2024 14:07:10 +0000 Subject: [PATCH] ollama: reduce log noise when pulling models --- internal/impl/ollama/base_processor.go | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/internal/impl/ollama/base_processor.go b/internal/impl/ollama/base_processor.go index 6b8fef85f..c245cbae4 100644 --- a/internal/impl/ollama/base_processor.go +++ b/internal/impl/ollama/base_processor.go @@ -334,9 +334,11 @@ func newBaseProcessor(conf *service.ParsedConfig, mgr *service.Resources) (p *ba if err = p.waitForServer(context.Background()); err != nil { return } + p.logger.Infof("Pulling %q", p.model) if err = p.pullModel(context.Background()); err != nil { return } + p.logger.Infof("Finished pulling %q", p.model) return } @@ -360,7 +362,7 @@ func (o *baseOllamaProcessor) pullModel(ctx context.Context) error { Model: o.model, } return o.client.Pull(ctx, &pr, func(resp api.ProgressResponse) error { - o.logger.Infof("Pulling %q: %s [%s/%s]", o.model, resp.Status, humanize.Bytes(uint64(resp.Completed)), humanize.Bytes(uint64(resp.Total))) + o.logger.Tracef("Pulling %q: %s [%s/%s]", o.model, resp.Status, humanize.Bytes(uint64(resp.Completed)), humanize.Bytes(uint64(resp.Total))) return nil }) }