package xai_test

import (
	"context"
	"errors"
	"testing"

	"gitea.stevedudenhoeffer.com/steve/go-llm/v2/openaicompat"
	"gitea.stevedudenhoeffer.com/steve/go-llm/v2/provider"
	"gitea.stevedudenhoeffer.com/steve/go-llm/v2/xai"
)

func TestNew_Basic(t *testing.T) {
	if p := xai.New("key", ""); p == nil {
		t.Fatal("New returned nil")
	}
}

func TestRules_Grok2RejectsImages(t *testing.T) {
	p := xai.New("key", "")
	req := provider.Request{
		Model: "grok-2",
		Messages: []provider.Message{{
			Role:   "user",
			Images: []provider.Image{{URL: "a"}},
		}},
	}
	_, err := p.Complete(context.Background(), req)
	var fue *openaicompat.FeatureUnsupportedError
	if !errors.As(err, &fue) || fue.Feature != "vision" {
		t.Fatalf("want FeatureUnsupportedError(vision), got %v", err)
	}
}