package main

import (
	"testing"
	"time"

	"github.com/kami/maven/internal/llm"
)

func TestPickLLMRouterOff(t *testing.T) {
	if r := pickLLMRouter(false, llm.New("http://127.0.0.1:1", time.Second)); r != nil {
		t.Error("flag off should give no LLM router")
	}
}

// The operator can turn the flag on without an LLM phraser configured. That must
// leave the classifier running, not panic.
func TestPickLLMRouterOnWithoutClient(t *testing.T) {
	if r := pickLLMRouter(true, nil); r != nil {
		t.Error("no llama-server should give no LLM router")
	}
}

func TestPickLLMRouterOn(t *testing.T) {
	if r := pickLLMRouter(true, llm.New("http://127.0.0.1:1", time.Second)); r == nil {
		t.Error("flag on with a client should give an LLM router")
	}
}
