Documentation
¶
Overview ¶
Package integration provides utilities for integration testing of AI providers. These tests make actual API calls and require valid API keys.
Index ¶
- func SkipIfNoKey(t *testing.T, envVar string)
- func TestEmbed(t *testing.T, m model.EmbeddingModel)
- func TestEmbedMany(t *testing.T, m model.EmbeddingModel)
- func TestGenerateImage(t *testing.T, m model.ImageModel)
- func TestGenerateSpeech(t *testing.T, m model.SpeechModel)
- func TestGenerateText(t *testing.T, m stream.Model)
- func TestGenerateTextWithSystemPrompt(t *testing.T, m stream.Model)
- func TestImageInput(t *testing.T, m stream.Model, imageURL string)
- func TestStreamText(t *testing.T, m stream.Model)
- func TestToolCalls(t *testing.T, m stream.Model)
- func TestTranscribe(t *testing.T, m model.TranscriptionModel, audioData []byte)
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
func SkipIfNoKey ¶
SkipIfNoKey skips the test if the specified environment variable is not set.
func TestEmbed ¶
func TestEmbed(t *testing.T, m model.EmbeddingModel)
TestEmbed tests single embedding generation.
func TestEmbedMany ¶
func TestEmbedMany(t *testing.T, m model.EmbeddingModel)
TestEmbedMany tests multiple embedding generation.
func TestGenerateImage ¶
func TestGenerateImage(t *testing.T, m model.ImageModel)
TestGenerateImage tests image generation.
func TestGenerateSpeech ¶
func TestGenerateSpeech(t *testing.T, m model.SpeechModel)
TestGenerateSpeech tests text-to-speech generation.
func TestGenerateText ¶
TestGenerateText tests basic text generation.
func TestGenerateTextWithSystemPrompt ¶
TestGenerateTextWithSystemPrompt tests text generation with a system prompt.
func TestImageInput ¶
TestImageInput tests image input for vision models.
func TestStreamText ¶
TestStreamText tests streaming text generation.
func TestToolCalls ¶
TestToolCalls tests tool calling functionality.
func TestTranscribe ¶
func TestTranscribe(t *testing.T, m model.TranscriptionModel, audioData []byte)
TestTranscribe tests speech-to-text transcription.
Types ¶
This section is empty.
Source Files