Skip to content

Commit b0660f9

Browse files
Add x-ios-bundle-identifier header to GenerativeAIService requests
This change adds the `x-ios-bundle-identifier` header to requests made by `GenerativeAIService`. The value is set to `Bundle.main.bundleIdentifier` if available. This header is sent unconditionally. A test assertion has been added to `GenerativeModelTestUtil.swift` to verify the presence and correctness of this header in unit tests.
1 parent 61aae4c commit b0660f9

File tree

2 files changed

+7
-0
lines changed

2 files changed

+7
-0
lines changed

FirebaseAI/Sources/GenerativeAIService.swift

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -211,6 +211,10 @@ struct GenerativeAIService {
211211
}
212212
}
213213

214+
if let bundleID = Bundle.main.bundleIdentifier {
215+
urlRequest.setValue(bundleID, forHTTPHeaderField: "x-ios-bundle-identifier")
216+
}
217+
214218
let encoder = JSONEncoder()
215219
urlRequest.httpBody = try encoder.encode(request)
216220
urlRequest.timeoutInterval = request.options.timeout

FirebaseAI/Tests/Unit/TestUtilities/GenerativeModelTestUtil.swift

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,9 @@ enum GenerativeModelTestUtil {
6969
XCTAssertEqual(firebaseAppID, dataCollection ? "My app ID" : nil)
7070
XCTAssertEqual(appVersion, dataCollection ? expectedAppVersion : nil)
7171

72+
let bundleID = request.value(forHTTPHeaderField: "x-ios-bundle-identifier")
73+
XCTAssertEqual(bundleID, Bundle.main.bundleIdentifier)
74+
7275
if let authToken {
7376
XCTAssertEqual(
7477
request.value(forHTTPHeaderField: "Authorization"),

0 commit comments

Comments
 (0)