integration

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)

Functions

func SkipIfNoKey

func SkipIfNoKey(t *testing.T, envVar string)

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

func TestGenerateText(t *testing.T, m stream.Model)

TestGenerateText tests basic text generation.

func TestGenerateTextWithSystemPrompt

func TestGenerateTextWithSystemPrompt(t *testing.T, m stream.Model)

TestGenerateTextWithSystemPrompt tests text generation with a system prompt.

func TestImageInput

func TestImageInput(t *testing.T, m stream.Model, imageURL string)

TestImageInput tests image input for vision models.

func TestStreamText

func TestStreamText(t *testing.T, m stream.Model)

TestStreamText tests streaming text generation.

func TestToolCalls

func TestToolCalls(t *testing.T, m stream.Model)

TestToolCalls tests tool calling functionality.

func TestTranscribe

func TestTranscribe(t *testing.T, m model.TranscriptionModel, audioData []byte)

TestTranscribe tests speech-to-text transcription.