38 lines
1.1 KiB
Go
38 lines
1.1 KiB
Go
package cognitive_providers_test
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"git.cin.su/m/Assistant/adapters/models"
|
|
"git.cin.su/m/Assistant/contracts/capability"
|
|
)
|
|
|
|
func TestLiveReadOnlyDiscovery(t *testing.T) {
|
|
if os.Getenv("ASSISTANT_LIVE_LOCAL_DISCOVERY") != "1" {
|
|
t.Skip("set ASSISTANT_LIVE_LOCAL_DISCOVERY=1 for read-only local provider probes")
|
|
}
|
|
ctx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
|
|
defer cancel()
|
|
cpu, err := models.NewOpenAICompatibleLocalAdapter("http://docker.cin.su:18082/v1", nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
cpuModels, cpuProbe := cpu.Discover(ctx)
|
|
if cpuProbe.Health != capability.HealthAvailable || len(cpuModels) == 0 {
|
|
t.Fatalf("CPU probe=%+v models=%+v", cpuProbe, cpuModels)
|
|
}
|
|
t.Logf("CPU probe=%+v models=%+v", cpuProbe, cpuModels)
|
|
gpu, err := models.NewOllamaAdapter("http://192.168.220.91:11434", nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
gpuModels, gpuProbe := gpu.Discover(ctx)
|
|
if gpuProbe.Health != capability.HealthAvailable || len(gpuModels) == 0 {
|
|
t.Fatalf("GPU probe=%+v models=%+v", gpuProbe, gpuModels)
|
|
}
|
|
t.Logf("GPU probe=%+v models=%+v", gpuProbe, gpuModels)
|
|
}
|