af0d4edd74
- Introduced `image_generate` tool for generating images from text prompts, with options for specifying output file paths and image dimensions. - Updated `image_read` functionality to allow optional provider specification for enhanced image analysis. - Implemented new `GenerateImage` method in the LLM API for seamless integration of image generation capabilities. - Enhanced documentation to include detailed usage examples for both image reading and generation tools. - Updated tests to validate new image generation features and ensure robust functionality across image tools.
54 lines
1.3 KiB
Go
54 lines
1.3 KiB
Go
package image
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/yaoapp/gou/process"
|
|
)
|
|
|
|
func TestModelHasCapability_Found(t *testing.T) {
|
|
caps := []string{"chat", "image_generation", "vision"}
|
|
if !modelHasCapability(caps, "image_generation") {
|
|
t.Error("expected true for image_generation")
|
|
}
|
|
if !modelHasCapability(caps, "vision") {
|
|
t.Error("expected true for vision")
|
|
}
|
|
}
|
|
|
|
func TestModelHasCapability_NotFound(t *testing.T) {
|
|
caps := []string{"chat", "embedding"}
|
|
if modelHasCapability(caps, "image_generation") {
|
|
t.Error("expected false for image_generation")
|
|
}
|
|
}
|
|
|
|
func TestModelHasCapability_Empty(t *testing.T) {
|
|
if modelHasCapability(nil, "image_generation") {
|
|
t.Error("expected false for nil caps")
|
|
}
|
|
if modelHasCapability([]string{}, "image_generation") {
|
|
t.Error("expected false for empty caps")
|
|
}
|
|
}
|
|
|
|
func TestProvidersHandler_NoAuth(t *testing.T) {
|
|
proc := &process.Process{
|
|
Args: []interface{}{"image_generation"},
|
|
}
|
|
result := ProvidersHandler(proc)
|
|
m, ok := result.(map[string]interface{})
|
|
if !ok {
|
|
t.Fatal("expected map result")
|
|
}
|
|
if _, hasErr := m["error"]; !hasErr {
|
|
t.Error("expected error when no auth info")
|
|
}
|
|
}
|
|
|
|
func TestFindFirstImageGenConnector_NoGlobal(t *testing.T) {
|
|
result := findFirstImageGenConnector(nil)
|
|
if result != "" {
|
|
t.Errorf("expected empty string, got %q", result)
|
|
}
|
|
}
|