tyler.durden/compose
1
0
Fork 0
mirror of https://github.com/docker/compose.git synced 2025-07-26 07:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

add a Done event to model progress display

Browse Source 
Signed-off-by: Guillaume Lours <705411+glours@users.noreply.github.com>
This commit is contained in:
Guillaume Lours 2025-07-02 11:12:20 +02:00 committed by Nicolas De loof
parent b046a5ef72
commit f42374bb18
1 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
pkg/compose/model.go
View File

@ -83,20 +83,25 @@ func (s *composeService) ensureModels(ctx context.Context, project *types.Projec
config.Name = name config.Name = name
} }
eg.Go(func() error { eg.Go(func() error {
w := progress.ContextWriter(gctx)
if !slices.Contains(availableModels, config.Model) { if !slices.Contains(availableModels, config.Model) {
err = s.pullModel(gctx, dockerModel, config, quietPull) err = s.pullModel(gctx, dockerModel, config, quietPull, w)
if err != nil { if err != nil {
return err return err
} }
} }
return s.configureModel(gctx, dockerModel, config) err = s.configureModel(gctx, dockerModel, config, w)
if err != nil {
return err
}
w.Event(progress.CreatedEvent(config.Name))
return nil
}) })
} }
return eg.Wait() return eg.Wait()
} }
func (s *composeService) pullModel(ctx context.Context, dockerModel *manager.Plugin, model types.ModelConfig, quietPull bool) error { func (s *composeService) pullModel(ctx context.Context, dockerModel *manager.Plugin, model types.ModelConfig, quietPull bool, w progress.Writer) error {
w := progress.ContextWriter(ctx)
w.Event(progress.Event{ w.Event(progress.Event{
ID: model.Name, ID: model.Name,
Status: progress.Working, Status: progress.Working,
@ -145,7 +150,12 @@ func (s *composeService) pullModel(ctx context.Context, dockerModel *manager.Plu
return err return err
} }
func (s *composeService) configureModel(ctx context.Context, dockerModel *manager.Plugin, config types.ModelConfig) error { func (s *composeService) configureModel(ctx context.Context, dockerModel *manager.Plugin, config types.ModelConfig, w progress.Writer) error {
w.Event(progress.Event{
ID: config.Name,
Status: progress.Working,
Text: "Configuring",
})
// configure [--context-size=<n>] MODEL [-- <runtime-flags...>] // configure [--context-size=<n>] MODEL [-- <runtime-flags...>]
args := []string{"configure"} args := []string{"configure"}
if config.ContextSize > 0 { if config.ContextSize > 0 {