go-micro.dev/v5@v5.12.0/genai/openai/openai_test.go (about)

     1  package openai
     2  
     3  import (
     4  	"go-micro.dev/v5/genai"
     5  	"os"
     6  	"testing"
     7  )
     8  
     9  func TestOpenAI_GenerateText(t *testing.T) {
    10  	apiKey := os.Getenv("OPENAI_API_KEY")
    11  	if apiKey == "" {
    12  		t.Skip("OPENAI_API_KEY not set")
    13  	}
    14  	client := New(genai.WithAPIKey(apiKey))
    15  	res, err := client.Generate("Say hello world", genai.Text)
    16  	if err != nil {
    17  		t.Fatalf("Generate error: %v", err)
    18  	}
    19  	if res == nil || res.Text == "" {
    20  		t.Error("Expected non-empty text response")
    21  	}
    22  }
    23  
    24  func TestOpenAI_GenerateImage(t *testing.T) {
    25  	apiKey := os.Getenv("OPENAI_API_KEY")
    26  	if apiKey == "" {
    27  		t.Skip("OPENAI_API_KEY not set")
    28  	}
    29  	client := New(genai.WithAPIKey(apiKey))
    30  	res, err := client.Generate("A cat wearing sunglasses", genai.Image)
    31  	if err != nil {
    32  		t.Fatalf("Generate error: %v", err)
    33  	}
    34  	if res == nil || res.Text == "" {
    35  		t.Error("Expected non-empty image URL")
    36  	}
    37  }