Templates do not have an implicit .prompt suffix and only global location supported

Author: paulb777

FirebaseAI/Sources/GenerateImagesRequest.swift

@@ -1,4 +1,3 @@
-
 // Copyright 2025 Google LLC
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
@@ -25,8 +24,7 @@ public class GenerateImagesRequest: @unchecked Sendable, GenerativeAIRequest {
     if case let .vertexAI(_, location) = apiConfig.service {
       urlString += "/locations/\(location)"
     }
-    let templateName = template.hasSuffix(".prompt") ? template : "\(template).prompt"
-    urlString += "/templates/\(templateName):\(ImageAPIMethod.generateImages.rawValue)"
+    urlString += "/templates/\(template):\(ImageAPIMethod.generateImages.rawValue)"
     return URL(string: urlString)!
   }
 

FirebaseAI/Sources/TemplateGenerateContentRequest.swift

@@ -49,8 +49,7 @@ extension TemplateGenerateContentRequest: GenerativeAIRequest {
     if case let .vertexAI(_, location) = apiConfig.service {
       urlString += "/locations/\(location)"
     }
-    let templateName = template.hasSuffix(".prompt") ? template : "\(template).prompt"
-    urlString += "/templates/\(templateName):templateGenerateContent"
+    urlString += "/templates/\(template):templateGenerateContent"
     if stream {
       urlString += "?alt=sse"
     }

FirebaseAI/Tests/TestApp/Tests/Integration/ServerPromptTemplateIntegrationTests.swift

@@ -23,10 +23,10 @@ final class ServerPromptTemplateIntegrationTests: XCTestCase {
   }
 
   func testGenerateContentWithText() async throws {
-    let model = FirebaseAI.firebaseAI(backend: .vertexAI()).templateGenerativeModel()
+    let model = FirebaseAI.firebaseAI(backend: .vertexAI(location: "global")).templateGenerativeModel()
     let userName = "paul"
     let response = try await model.generateContent(
-      template: "greeting",
+      template: "greeting2",
       variables: [
         "name": userName,
         "language": "Spanish",
@@ -40,7 +40,7 @@ final class ServerPromptTemplateIntegrationTests: XCTestCase {
     let model = FirebaseAI.firebaseAI(backend: .vertexAI()).templateGenerativeModel()
     let userName = "paul"
     let stream = try model.generateContentStream(
-      template: "greeting",
+      template: "greeting.prompt",
       variables: [
         "name": userName,
         "language": "English",
@@ -59,7 +59,7 @@ final class ServerPromptTemplateIntegrationTests: XCTestCase {
     let imagenModel = FirebaseAI.firebaseAI(backend: .vertexAI()).templateImagenModel()
     let imagenPrompt = "A cat picture"
     let response = try await imagenModel.generateImages(
-      template: "generate_images",
+      template: "generate_images.prompt",
       variables: [
         "prompt": imagenPrompt,
       ]
@@ -74,7 +74,7 @@ final class ServerPromptTemplateIntegrationTests: XCTestCase {
       let base64Image = imageBytes.base64EncodedString()
 
       let response = try await model.generateContent(
-        template: "media",
+        template: "media.prompt",
         variables: [
           "imageData": [
             "isInline": true,
@@ -96,7 +96,7 @@ final class ServerPromptTemplateIntegrationTests: XCTestCase {
       let base64Image = imageBytes.base64EncodedString()
 
       let stream = try model.generateContentStream(
-        template: "media",
+        template: "media.prompt",
         variables: [
           "imageData": [
             "isInline": true,
@@ -123,7 +123,7 @@ final class ServerPromptTemplateIntegrationTests: XCTestCase {
       ModelContent(role: "user", parts: "Hello!"),
       ModelContent(role: "model", parts: "Hi there! How can I help?"),
     ]
-    let chatSession = model.startChat(template: "chat_history", history: initialHistory)
+    let chatSession = model.startChat(template: "chat_history.prompt", history: initialHistory)
 
     let userMessage = "What's the weather like?"
 
@@ -142,7 +142,7 @@ final class ServerPromptTemplateIntegrationTests: XCTestCase {
       ModelContent(role: "user", parts: "Hello!"),
       ModelContent(role: "model", parts: "Hi there! How can I help?"),
     ]
-    let chatSession = model.startChat(template: "chat_history", history: initialHistory)
+    let chatSession = model.startChat(template: "chat_history.prompt", history: initialHistory)
 
     let userMessage = "What's the weather like?"