Skip to content

Commit f63c4c0

Browse files
committed
misc review fixes
1 parent fe3ae25 commit f63c4c0

File tree

6 files changed

+18
-41
lines changed

6 files changed

+18
-41
lines changed

FirebaseAI/Sources/AILog.swift

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ import os.log
1717

1818
internal import FirebaseCoreExtension
1919

20+
@available(iOS 15.0, macOS 12.0, macCatalyst 15.0, tvOS 15.0, watchOS 8.0, *)
2021
enum AILog {
2122
/// Log message codes for the Firebase AI SDK
2223
///

FirebaseAI/Sources/GenerateContentRequest.swift

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,15 @@ extension GenerateContentRequest: Encodable {
6060
}
6161
}
6262

63+
@available(iOS 15.0, macOS 12.0, macCatalyst 15.0, tvOS 15.0, watchOS 8.0, *)
64+
extension GenerateContentRequest {
65+
enum APIMethod: String {
66+
case generateContent
67+
case streamGenerateContent
68+
case countTokens
69+
}
70+
}
71+
6372
@available(iOS 15.0, macOS 12.0, macCatalyst 15.0, tvOS 15.0, watchOS 8.0, *)
6473
extension GenerateContentRequest: GenerativeAIRequest {
6574
typealias Response = GenerateContentResponse

FirebaseAI/Sources/TemplateGenerateContentRequest.swift

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,9 +49,11 @@ extension TemplateGenerateContentRequest: GenerativeAIRequest {
4949
if case let .vertexAI(_, location) = apiConfig.service {
5050
urlString += "/locations/\(location)"
5151
}
52-
urlString += "/templates/\(template):templateGenerateContent"
52+
5353
if stream {
54-
urlString += "?alt=sse"
54+
urlString += "/templates/\(template):templateStreamGenerateContent?alt=sse"
55+
} else {
56+
urlString += "/templates/\(template):templateGenerateContent"
5557
}
5658
guard let url = URL(string: urlString) else {
5759
throw AILog.makeInternalError(message: "Malformed URL: \(urlString)", code: .malformedURL)

FirebaseAI/Sources/TemplateGenerateImagesRequest.swift

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,10 @@
1414

1515
import Foundation
1616

17+
enum ImageAPIMethod: String {
18+
case generateImages = "templatePredict"
19+
}
20+
1721
@available(iOS 15.0, macOS 12.0, macCatalyst 15.0, tvOS 15.0, watchOS 8.0, *)
1822
class TemplateGenerateImagesRequest: @unchecked Sendable, GenerativeAIRequest {
1923
typealias Response = ImagenGenerationResponse<ImagenInlineImage>

FirebaseAI/Sources/Types/Internal/APIMethod.swift

Lines changed: 0 additions & 21 deletions
This file was deleted.

FirebaseAI/Sources/Types/Internal/ImageAPIMethod.swift

Lines changed: 0 additions & 18 deletions
This file was deleted.

0 commit comments

Comments
 (0)