Function TestLoadCaps_VisionModel

pkg/modelinfo/modelinfo_test.go:459–482

(t *testing.T)

}

func TestLoadCaps_VisionModel(t *testing.T) {
	t.Parallel()

	store := modelsdev.NewDatabaseStore(&modelsdev.Database{Providers: map[string]modelsdev.Provider{
		"anthropic": {
			Models: map[string]modelsdev.Model{
				"claude-3-5-sonnet": {
					Name: "Claude 3.5 Sonnet",
					Modalities: modelsdev.Modalities{
						Input:  []string{"text", "image", "pdf"},
						Output: []string{"text"},
					},
				},
			},
		},
	}})

	mc := LoadCaps(t.Context(), store, modelsdev.NewID("anthropic", "claude-3-5-sonnet"))

	assert.True(t, mc.Supports("image/jpeg"))
	assert.True(t, mc.Supports("image/png"))
	assert.True(t, mc.Supports("application/pdf"))
	assert.True(t, mc.Supports("text/plain"))
}

func TestLoadCaps_TextOnlyModel(t *testing.T) {
	t.Parallel()