fixes

Author: paulb777

FirebaseAI/Sources/TemplateVariable.swift

@@ -30,6 +30,8 @@ enum TemplateVariable: Encodable, Sendable {
       self = .int(value)
     case let value as Double:
       self = .double(value)
+    case let value as Float:
+      self = .double(Double(value))
     case let value as Bool:
       self = .bool(value)
     case let value as [Any]:

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

@@ -33,7 +33,6 @@ final class ServerPromptTemplateIntegrationTests: XCTestCase {
       ]
     )
     let text = try XCTUnwrap(response.text)
-    print(text)
     XCTAssert(text.contains("Paul"))
   }
 
@@ -161,4 +160,4 @@ final class ServerPromptTemplateIntegrationTests: XCTestCase {
     XCTAssertEqual(chatSession.history.count, 4)
     XCTAssertEqual((chatSession.history[2].parts.first as? TextPart)?.text, userMessage)
   }
-}
+}

FirebaseAI/Tests/Unit/TemplateChatSessionTests.swift

@@ -65,12 +65,7 @@ final class TemplateChatSessionTests: XCTestCase {
     let chat = model.startChat(template: "test-template")
     let stream = try chat.sendMessageStream("Hello", variables: ["name": "test"])
 
-    var content = ""
-    for try await response in stream {
-      if let text = response.text {
-        content += text
-      }
-    }
+    let content = try await GenerativeModelTestUtil.collectTextFromStream(stream)
 
     XCTAssertEqual(content, "The capital of Wyoming is **Cheyenne**.\n")
     XCTAssertEqual(chat.history.count, 2)

FirebaseAI/Tests/Unit/TemplateGenerativeModelTests.swift

@@ -66,12 +66,7 @@ final class TemplateGenerativeModelTests: XCTestCase {
       variables: ["name": "test"]
     )
 
-    var content = ""
-    for try await response in stream {
-      if let text = response.text {
-        content += text
-      }
-    }
+    let content = try await GenerativeModelTestUtil.collectTextFromStream(stream)
     XCTAssertEqual(content, "The capital of Wyoming is **Cheyenne**.\n")
   }
 }

FirebaseAI/Tests/Unit/TestUtilities/GenerativeModelTestUtil.swift

@@ -88,6 +88,19 @@ enum GenerativeModelTestUtil {
     #endif // os(watchOS)
   }
 
+  static func collectTextFromStream(_ stream: AsyncThrowingStream<
+    GenerateContentResponse,
+    Error
+  >) async throws -> String {
+    var content = ""
+    for try await response in stream {
+      if let text = response.text {
+        content += text
+      }
+    }
+    return content
+  }
+
   static func nonHTTPRequestHandler() throws -> ((URLRequest) -> (
     URLResponse,
     AsyncLineSequence<URL.AsyncBytes>?