Skip to content

Commit c83919a

Browse files
committed
review/style
1 parent c94a89c commit c83919a

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

FirebaseAI/Tests/TestApp/Tests/Integration/GenerateContentIntegrationTests.swift

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -134,8 +134,7 @@ struct GenerateContentIntegrationTests {
134134
#expect(text == "Blue")
135135

136136
let usageMetadata = try #require(response.usageMetadata)
137-
if config == InstanceConfig.googleAI_v1beta ||
138-
config == InstanceConfig.googleAI_v1beta_appCheckLimitedUse {
137+
if case .googleAI = config.apiConfig.service {
139138
#expect(usageMetadata.promptTokenCount.isEqual(to: 11, accuracy: tokenCountAccuracy))
140139
#expect(usageMetadata.candidatesTokensDetails.count == 0)
141140
} else {

0 commit comments

Comments
 (0)