Просмотр исходного кода

Add `liveModel` static method to construct `LiveGenerativeModel`

Andrew Heard 8 месяцев назад
Родитель
Сommit
05fc2ff2a4
1 измененных файлов с 12 добавлено и 0 удалено
  1. 12 0
      FirebaseAI/Sources/FirebaseAI.swift

+ 12 - 0
FirebaseAI/Sources/FirebaseAI.swift

@@ -130,6 +130,18 @@ public final class FirebaseAI: Sendable {
     )
   }
 
+  public func liveModel(modelName: String,
+                        generationConfig: LiveGenerationConfig? = nil,
+                        requestOptions: RequestOptions = RequestOptions()) -> LiveGenerativeModel {
+    return LiveGenerativeModel(
+      modelResourceName: modelResourceName(modelName: modelName),
+      firebaseInfo: firebaseInfo,
+      apiConfig: apiConfig,
+      generationConfig: generationConfig,
+      requestOptions: requestOptions
+    )
+  }
+
   /// Class to enable FirebaseAI to register via the Objective-C based Firebase component system
   /// to include FirebaseAI in the userAgent.
   @objc(FIRVertexAIComponent) class FirebaseVertexAIComponent: NSObject {}