Skip to content

Commit 5a209d1

Browse files
committed
Expect thoughtSignature from Vertex AI
1 parent 28e531b commit 5a209d1

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

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

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -296,12 +296,8 @@ struct GenerateContentIntegrationTests {
296296
"country": .string("Canada"),
297297
])
298298
#expect(temperatureFunctionCall.isThought == false)
299-
if let _ = thinkingConfig.includeThoughts, case .googleAI = config.apiConfig.service {
300-
let thoughtSignature = try #require(temperatureFunctionCall.thoughtSignature)
301-
#expect(!thoughtSignature.isEmpty)
302-
} else {
303-
#expect(temperatureFunctionCall.thoughtSignature == nil)
304-
}
299+
let thoughtSignature = try #require(temperatureFunctionCall.thoughtSignature)
300+
#expect(!thoughtSignature.isEmpty)
305301

306302
let temperatureFunctionResponse = FunctionResponsePart(
307303
name: temperatureFunctionCall.name,

0 commit comments

Comments
 (0)