Skip to content

Commit 9b1d73c

Browse files
committed
self review, including file/naming updates
1 parent 131d1ca commit 9b1d73c

File tree

4 files changed

+6
-6
lines changed

4 files changed

+6
-6
lines changed

FirebaseAI/Sources/GenerateImagesRequest.swift renamed to FirebaseAI/Sources/TemplateGenerateImagesRequest.swift

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -15,10 +15,10 @@
1515
import Foundation
1616

1717
@available(iOS 15.0, macOS 12.0, macCatalyst 15.0, tvOS 15.0, watchOS 8.0, *)
18-
public class GenerateImagesRequest: @unchecked Sendable, GenerativeAIRequest {
19-
public typealias Response = ImagenGenerationResponse<ImagenInlineImage>
18+
class TemplateGenerateImagesRequest: @unchecked Sendable, GenerativeAIRequest {
19+
typealias Response = ImagenGenerationResponse<ImagenInlineImage>
2020

21-
public var url: URL {
21+
var url: URL {
2222
var urlString =
2323
"\(apiConfig.service.endpoint.rawValue)/\(apiConfig.version.rawValue)/projects/\(projectID)"
2424
if case let .vertexAI(_, location) = apiConfig.service {
@@ -28,7 +28,7 @@ public class GenerateImagesRequest: @unchecked Sendable, GenerativeAIRequest {
2828
return URL(string: urlString)!
2929
}
3030

31-
public let options: RequestOptions
31+
let options: RequestOptions
3232

3333
let apiConfig: APIConfig
3434

@@ -49,7 +49,7 @@ public class GenerateImagesRequest: @unchecked Sendable, GenerativeAIRequest {
4949
case variables = "inputs"
5050
}
5151

52-
public func encode(to encoder: Encoder) throws {
52+
func encode(to encoder: Encoder) throws {
5353
var container = encoder.container(keyedBy: CodingKeys.self)
5454
try container.encode(variables, forKey: .variables)
5555
}

FirebaseAI/Sources/TemplateImagenModel.swift

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ public final class TemplateImagenModel: Sendable {
4040
-> ImagenGenerationResponse<ImagenInlineImage> {
4141
let templateVariables = try variables.mapValues { try TemplateVariable(value: $0) }
4242
let projectID = generativeAIService.firebaseInfo.projectID
43-
let request = GenerateImagesRequest(
43+
let request = TemplateGenerateImagesRequest(
4444
template: template,
4545
variables: templateVariables,
4646
projectID: projectID,

0 commit comments

Comments
 (0)