mirror of
https://github.com/ollama/ollama.git
synced 2026-03-27 02:58:43 +07:00
22 lines
523 B
Go
22 lines
523 B
Go
package launch
|
|
|
|
import "strings"
|
|
|
|
// OverrideIntegration replaces one registry entry's runner for tests and returns a restore function.
|
|
func OverrideIntegration(name string, runner Runner) func() {
|
|
spec, err := LookupIntegrationSpec(name)
|
|
if err != nil {
|
|
key := strings.ToLower(name)
|
|
integrationSpecsByName[key] = &IntegrationSpec{Name: key, Runner: runner}
|
|
return func() {
|
|
delete(integrationSpecsByName, key)
|
|
}
|
|
}
|
|
|
|
original := spec.Runner
|
|
spec.Runner = runner
|
|
return func() {
|
|
spec.Runner = original
|
|
}
|
|
}
|