Merge pull request #425 from dgageot/text-examples

Fix gemini model name (And test all examples)

Author: krissetto

examples/bio.yaml

@@ -3,7 +3,7 @@ version: "2"
 
 agents:
   root:
-    model: gemini/gemini-2.5-flash
+    model: google/gemini-2.5-flash
     description: Create a biography of the user based on internet searches.
     instruction: |
       Your goal is to present the user with a concise and informative biography based on their online presence.

examples/pirate.yaml

@@ -7,5 +7,5 @@ agents:
     instruction: Always answer by talking like a pirate.
     model: openai/gpt-4o
     # model: anthropic/claude-3-5-sonnet-latest
-    # model: gemini/gemini-2.5-flash
+    # model: google/gemini-2.5-flash
     # model: dmr/ai/llama3.2

pkg/teamloader/teamloader_test.go

@@ -2,7 +2,9 @@ package teamloader
 
 import (
 	"context"
+	"io/fs"
 	"os"
+	"path/filepath"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
@@ -94,6 +96,40 @@ func TestCheckRequiredEnvVarsWithModelGateway(t *testing.T) {
 	require.NoError(t, err)
 }
 
+func TestLoadExamples(t *testing.T) {
+	// Collect the missing env vars.
+	missingEnvs := map[string]bool{}
+
+	for _, file := range collectExamples(t) {
+		t.Run(file, func(t *testing.T) {
+			_, err := Load(t.Context(), file, config.RuntimeConfig{})
+			if err != nil {
+				envErr := &environment.RequiredEnvError{}
+				require.ErrorAs(t, err, &envErr)
+
+				for _, env := range envErr.Missing {
+					missingEnvs[env] = true
+				}
+			}
+		})
+	}
+
+	for name := range missingEnvs {
+		t.Setenv(name, "dummy")
+	}
+
+	// Load all the examples.
+	for _, file := range collectExamples(t) {
+		t.Run(file, func(t *testing.T) {
+			t.Parallel()
+
+			teams, err := Load(t.Context(), file, config.RuntimeConfig{})
+			require.NoError(t, err)
+			require.NotEmpty(t, teams)
+		})
+	}
+}
+
 func openRoot(t *testing.T, dir string) *os.Root {
 	t.Helper()
 
@@ -103,3 +139,22 @@ func openRoot(t *testing.T, dir string) *os.Root {
 
 	return root
 }
+
+func collectExamples(t *testing.T) []string {
+	t.Helper()
+
+	var files []string
+	err := filepath.WalkDir(filepath.Join("..", "..", "examples"), func(path string, d fs.DirEntry, err error) error {
+		if err != nil {
+			return err
+		}
+		if !d.IsDir() && filepath.Ext(path) == ".yaml" {
+			files = append(files, path)
+		}
+		return nil
+	})
+	require.NoError(t, err)
+	assert.NotEmpty(t, files)
+
+	return files
+}