Add the Rakestrawhome Gemma profile

2026-08-23 17:12:17 +00:00
parent 93af155254
commit a1805fe550
4 changed files with 92 additions and 39 deletions
--- a/engine_test.go
+++ b/engine_test.go
@@ -1492,43 +1492,63 @@ func TestSelectedProfileRepositoryReadFailureMapsToProfileLoad(t *testing.T) {
 }

 func TestPrepareUsesBuiltInProfileWithoutProfileDir(t *testing.T) {
-	t.Setenv("OPENROUTER_API_KEY", "test-key")
-	engine, err := promptkit.NewEngine(promptkit.Config{
-		PromptDir: frameworkPromptDir,
-		SchemaDir: frameworkSchemaDir,
-	})
-	if err != nil {
-		t.Fatalf("expected engine construction to succeed, got %v", err)
+	tests := []struct {
+		name      string
+		profileID string
+		backendID string
+		endpoint  string
+		apiKeyEnv string
+		model     string
+	}{
+		{
+			name:      "OpenRouter",
+			profileID: "mistral-small-3",
+			backendID: promptkit.BackendOpenRouter,
+			endpoint:  "https://openrouter.ai/api/v1",
+			apiKeyEnv: "OPENROUTER_API_KEY",
+			model:     "mistralai/mistral-small-3.2-24b-instruct",
+		},
+		{
+			name:      "Rakestrawhome",
+			profileID: "rakestrawhome-gemma-4-31b",
+			backendID: promptkit.BackendRakestrawHome,
+			endpoint:  "https://inference.ai.rakestrawhome.com/v1",
+			apiKeyEnv: "RAKESTRAWHOME_INFERENCE_API_KEY",
+			model:     "google/gemma-4-31b-it",
+		},
 	}

-	prepared, err := engine.Prepare(context.Background(), promptkit.RunRequest{
-		PromptID:  frameworkMarkdownSummaryPromptID,
-		ProfileID: "mistral-small-3",
-		Inputs: map[string]promptkit.ArtifactRef{
-			"transcript": promptkit.Inline("Rin opens the gate."),
-			"glossary":   promptkit.Inline("gate: A guarded passage."),
-		},
-	})
-	if err != nil {
-		t.Fatalf("expected built-in profile prepare to succeed, got %v", err)
-	}
-	if prepared.SelectedProfileID != "mistral-small-3" {
-		t.Fatalf("unexpected selected profile: %q", prepared.SelectedProfileID)
-	}
-	if prepared.SelectedBackendID != promptkit.BackendOpenRouter {
-		t.Fatalf("unexpected selected backend: %q", prepared.SelectedBackendID)
-	}
-	if prepared.EffectiveModelParams.BackendID != promptkit.BackendOpenRouter {
-		t.Fatalf("unexpected effective backend: %q", prepared.EffectiveModelParams.BackendID)
-	}
-	if prepared.EffectiveModelParams.Endpoint != "https://openrouter.ai/api/v1" {
-		t.Fatalf("unexpected built-in endpoint: %q", prepared.EffectiveModelParams.Endpoint)
-	}
-	if prepared.EffectiveModelParams.APIKeyEnv != "OPENROUTER_API_KEY" {
-		t.Fatalf("unexpected built-in api key environment name: %q", prepared.EffectiveModelParams.APIKeyEnv)
-	}
-	if prepared.EffectiveModelParams.Model != "mistralai/mistral-small-3.2-24b-instruct" {
-		t.Fatalf("unexpected built-in model: %q", prepared.EffectiveModelParams.Model)
+	for _, tc := range tests {
+		t.Run(tc.name, func(t *testing.T) {
+			t.Setenv(tc.apiKeyEnv, "test-key")
+			engine, err := promptkit.NewEngine(promptkit.Config{
+				PromptDir: frameworkPromptDir,
+				SchemaDir: frameworkSchemaDir,
+			})
+			if err != nil {
+				t.Fatalf("expected engine construction to succeed, got %v", err)
+			}
+
+			prepared, err := engine.Prepare(context.Background(), promptkit.RunRequest{
+				PromptID:  frameworkMarkdownSummaryPromptID,
+				ProfileID: tc.profileID,
+				Inputs: map[string]promptkit.ArtifactRef{
+					"transcript": promptkit.Inline("Rin opens the gate."),
+					"glossary":   promptkit.Inline("gate: A guarded passage."),
+				},
+			})
+			if err != nil {
+				t.Fatalf("expected built-in profile prepare to succeed, got %v", err)
+			}
+			if prepared.SelectedProfileID != tc.profileID ||
+				prepared.SelectedBackendID != tc.backendID ||
+				prepared.EffectiveModelParams.BackendID != tc.backendID ||
+				prepared.EffectiveModelParams.Endpoint != tc.endpoint ||
+				prepared.EffectiveModelParams.APIKeyEnv != tc.apiKeyEnv ||
+				prepared.EffectiveModelParams.Model != tc.model {
+				t.Fatalf("unexpected built-in preparation: %#v", prepared)
+			}
+		})
 	}
 }