18 lines
379 B
Go
18 lines
379 B
Go
package ezollama
|
|
|
|
import (
|
|
"context"
|
|
"github.com/ollama/ollama/api"
|
|
)
|
|
|
|
func GenerateSync(ctx context.Context, client api.Client, req *api.GenerateRequest) (api.GenerateResponse, error) {
|
|
var lastResp api.GenerateResponse
|
|
err := client.Generate(ctx, req, func(resp api.GenerateResponse) error {
|
|
if resp.Done {
|
|
lastResp = resp
|
|
}
|
|
return nil
|
|
})
|
|
return lastResp, err
|
|
}
|