Add test for completion with Google client

.github/workflows/ci.yml

@@ -41,6 +41,7 @@ jobs:
         env:
           OPENAI_TOKEN: ${{ secrets.OPENAI_TOKEN }}
           ANTHROPIC_TOKEN: ${{ secrets.ANTHROPIC_TOKEN }}
+          GOOGLE_TOKEN: ${{ secrets.GOOGLE_TOKEN }}
 
   evaluate:
     name: Evaluate

anthropic_test.go

@@ -20,7 +20,7 @@ func TestNewAnthropicClient(t *testing.T) {
 	})
 }
 
-func TestAnthropicClientPrompt(t *testing.T) {
+func TestAnthropicClientCompletion(t *testing.T) {
 	_ = env.Load(".env.test.local")
 
 	t.Run("can do a basic chat completion", func(t *testing.T) {

google_test.go

@@ -1,8 +1,13 @@
 package llm_test
 
 import (
+	"context"
+	"fmt"
+	"strings"
 	"testing"
 
+	"github.com/google/generative-ai-go/genai"
+	"maragu.dev/env"
 	"maragu.dev/is"
 
 	"maragu.dev/llm"
@@ -14,3 +19,19 @@ func TestNewGoogleClient(t *testing.T) {
 		is.NotNil(t, client)
 	})
 }
+
+func TestGoogleClientCompletion(t *testing.T) {
+	_ = env.Load(".env.test.local")
+
+	t.Run("can do a basic chat completion", func(t *testing.T) {
+		client := llm.NewGoogleClient(llm.NewGoogleClientOptions{Token: env.GetStringOrDefault("GOOGLE_TOKEN", "")})
+		is.NotNil(t, client)
+
+		model := client.Client.GenerativeModel("models/gemini-1.5-flash-latest")
+		model.SystemInstruction = genai.NewUserContent(genai.Text(`Only say the word "Hi", nothing more.`))
+		res, err := model.GenerateContent(context.Background(), genai.Text("Hi."))
+		is.NotError(t, err)
+		is.True(t, len(res.Candidates) > 0)
+		is.True(t, strings.Contains(fmt.Sprint(res.Candidates[0].Content.Parts), "Hi"))
+	})
+}

openai_test.go

@@ -19,7 +19,7 @@ func TestNewOpenAIClient(t *testing.T) {
 	})
 }
 
-func TestOpenAIClientPrompt(t *testing.T) {
+func TestOpenAIClientCompletion(t *testing.T) {
 	_ = env.Load(".env.test.local")
 
 	t.Run("can do a basic chat completion", func(t *testing.T) {