Skip to content

Commit 05fc2ff

Browse files
committed
Add liveModel static method to construct LiveGenerativeModel
1 parent 4c930b4 commit 05fc2ff

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

FirebaseAI/Sources/FirebaseAI.swift

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -130,6 +130,18 @@ public final class FirebaseAI: Sendable {
130130
)
131131
}
132132

133+
public func liveModel(modelName: String,
134+
generationConfig: LiveGenerationConfig? = nil,
135+
requestOptions: RequestOptions = RequestOptions()) -> LiveGenerativeModel {
136+
return LiveGenerativeModel(
137+
modelResourceName: modelResourceName(modelName: modelName),
138+
firebaseInfo: firebaseInfo,
139+
apiConfig: apiConfig,
140+
generationConfig: generationConfig,
141+
requestOptions: requestOptions
142+
)
143+
}
144+
133145
/// Class to enable FirebaseAI to register via the Objective-C based Firebase component system
134146
/// to include FirebaseAI in the userAgent.
135147
@objc(FIRVertexAIComponent) class FirebaseVertexAIComponent: NSObject {}

0 commit comments

Comments
 (0)