mirror of https://gitee.com/namelin2022/ollama
2 changed files with 27 additions and 16 deletions
@ -0,0 +1,27 @@ |
|||||
|
//go:build darwin || windows
|
||||
|
|
||||
|
package cmd |
||||
|
|
||||
|
import ( |
||||
|
"context" |
||||
|
"errors" |
||||
|
"time" |
||||
|
|
||||
|
"github.com/ollama/ollama/api" |
||||
|
) |
||||
|
|
||||
|
func waitForServer(ctx context.Context, client *api.Client) error { |
||||
|
// wait for the server to start
|
||||
|
timeout := time.After(5 * time.Second) |
||||
|
tick := time.Tick(500 * time.Millisecond) |
||||
|
for { |
||||
|
select { |
||||
|
case <-timeout: |
||||
|
return errors.New("timed out waiting for server to start") |
||||
|
case <-tick: |
||||
|
if err := client.Heartbeat(ctx); err == nil { |
||||
|
return nil // server has started
|
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue