Temporarily add support for function behavior

Author: daymxn

FirebaseAI/Sources/Tool.swift

@@ -29,6 +29,9 @@ public struct FunctionDeclaration: Sendable {
   /// Describes the parameters to this function; must be of type `DataType.object`.
   let parameters: Schema?
 
+  // TODO: remove (added for testing)
+  let behavior: FunctionBehavior?
+
   /// Constructs a new `FunctionDeclaration`.
   ///
   /// - Parameters:
@@ -40,8 +43,21 @@ public struct FunctionDeclaration: Sendable {
   ///   calls; by default, all parameters are considered required.
   public init(name: String, description: String, parameters: [String: Schema],
               optionalParameters: [String] = []) {
+    self.init(
+      name: name,
+      description: description,
+      parameters: parameters,
+      optionalParameters: optionalParameters,
+      functionBehavior: nil
+    )
+  }
+
+  // TODO: remove (added for testing)
+  public init(name: String, description: String, parameters: [String: Schema],
+              optionalParameters: [String] = [], functionBehavior: FunctionBehavior? = nil) {
     self.name = name
     self.description = description
+    behavior = functionBehavior
     self.parameters = Schema.object(
       properties: parameters,
       optionalProperties: optionalParameters,
@@ -50,6 +66,12 @@ public struct FunctionDeclaration: Sendable {
   }
 }
 
+// TODO: remove (added for testing)
+public enum FunctionBehavior: String, Sendable, Encodable {
+  case blocking = "BLOCKING"
+  case nonBlocking = "NON_BLOCKING"
+}
+
 /// A tool that allows the generative model to connect to Google Search to access and incorporate
 /// up-to-date information from the web into its responses.
 ///
@@ -200,13 +222,15 @@ extension FunctionDeclaration: Encodable {
     case name
     case description
     case parameters
+    case behavior // TODO: remove (added for testing)
   }
 
   public func encode(to encoder: Encoder) throws {
     var container = encoder.container(keyedBy: CodingKeys.self)
     try container.encode(name, forKey: .name)
     try container.encode(description, forKey: .description)
     try container.encode(parameters, forKey: .parameters)
+    try container.encode(behavior, forKey: .behavior) // TODO: remove (added for testing)
   }
 }