Make template chat APIs internal for now

Author: paulb777

FirebaseAI/Sources/TemplateChatSession.swift

@@ -14,11 +14,13 @@
 
 import Foundation
 
+// TODO: Restore `public` to class and methods when determined to be releaseable.
+
 /// A chat session that allows for conversation with a model.
 ///
 /// **Public Preview**: This API is a public preview and may be subject to change.
 @available(iOS 15.0, macOS 12.0, macCatalyst 15.0, tvOS 15.0, watchOS 8.0, *)
-public final class TemplateChatSession: Sendable {
+final class TemplateChatSession: Sendable {
   private let model: TemplateGenerativeModel
   private let templateID: String
   private let _history: History
@@ -49,7 +51,7 @@ public final class TemplateChatSession: Sendable {
   /// request timeout.
   /// - Returns: The content generated by the model.
   /// - Throws: A ``GenerateContentError`` if the request failed.
-  public func sendMessage(_ content: [ModelContent],
+  func sendMessage(_ content: [ModelContent],
                           inputs: [String: Any],
                           options: RequestOptions = RequestOptions()) async throws
     -> GenerateContentResponse {
@@ -79,7 +81,7 @@ public final class TemplateChatSession: Sendable {
   /// request timeout.
   /// - Returns: The content generated by the model.
   /// - Throws: A ``GenerateContentError`` if the request failed.
-  public func sendMessage(_ message: any PartsRepresentable,
+  func sendMessage(_ message: any PartsRepresentable,
                           inputs: [String: Any],
                           options: RequestOptions = RequestOptions()) async throws
     -> GenerateContentResponse {
@@ -100,7 +102,7 @@ public final class TemplateChatSession: Sendable {
   /// request timeout.
   /// - Returns: An `AsyncThrowingStream` that yields `GenerateContentResponse` objects.
   /// - Throws: A ``GenerateContentError`` if the request failed.
-  public func sendMessageStream(_ content: [ModelContent],
+  func sendMessageStream(_ content: [ModelContent],
                                 inputs: [String: Any],
                                 options: RequestOptions = RequestOptions()) throws
     -> AsyncThrowingStream<GenerateContentResponse, Error> {
@@ -155,7 +157,7 @@ public final class TemplateChatSession: Sendable {
   /// request timeout.
   /// - Returns: An `AsyncThrowingStream` that yields `GenerateContentResponse` objects.
   /// - Throws: A ``GenerateContentError`` if the request failed.
-  public func sendMessageStream(_ message: any PartsRepresentable,
+  func sendMessageStream(_ message: any PartsRepresentable,
                                 inputs: [String: Any],
                                 options: RequestOptions = RequestOptions()) throws
     -> AsyncThrowingStream<GenerateContentResponse, Error> {

FirebaseAI/Sources/TemplateGenerativeModel.swift

@@ -122,13 +122,15 @@ public final class TemplateGenerativeModel: Sendable {
     return generativeAIService.loadRequestStream(request: request)
   }
 
+  // TODO: Restore `public` determined to be releaseable along with the contents of TemplateChatSession.
+
   /// Creates a new chat conversation using this model with the provided history and template.
   ///
   /// - Parameters:
   ///   - templateID: The ID of the prompt template to use.
   ///   - history: The conversation history to use.
   /// - Returns: A new ``TemplateChatSession`` instance.
-  public func startChat(templateID: String,
+  func startChat(templateID: String,
                         history: [ModelContent] = []) -> TemplateChatSession {
     return TemplateChatSession(
       model: self,

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

@@ -12,7 +12,8 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-import FirebaseAI
+// TODO: remove @testable when Template Chat is restored to the public API.
+@testable import FirebaseAILogic
 import Testing
 #if canImport(UIKit)
   import UIKit
@@ -118,7 +119,7 @@ struct ServerPromptTemplateIntegrationTests {
     let base64Image = imageBytes.base64EncodedString()
 
     let stream = try model.generateContentStream(
-      templateID: "media.prompt",
+      templateID: "media",
       inputs: [
         "imageData": [
           "isInline": true,
@@ -165,7 +166,7 @@ struct ServerPromptTemplateIntegrationTests {
       ModelContent(role: "user", parts: "Hello!"),
       ModelContent(role: "model", parts: "Hi there! How can I help?"),
     ]
-    let chatSession = model.startChat(templateID: "chat_history.prompt", history: initialHistory)
+    let chatSession = model.startChat(templateID: "chat-history", history: initialHistory)
 
     let userMessage = "What's the weather like?"