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)
}