Fix existing tests and samples

andrewheard · andrewheard · commit 9e6fba231090 · 2024-07-29T16:37:32.000-04:00
diff --git a/Examples/GenerativeAISample/FunctionCallingSample/ViewModels/FunctionCallingViewModel.swift b/Examples/GenerativeAISample/FunctionCallingSample/ViewModels/FunctionCallingViewModel.swift
@@ -157,7 +157,7 @@ class FunctionCallingViewModel: ObservableObject {
       case let .functionCall(functionCall):
         messages.insert(functionCall.chatMessage(), at: messages.count - 1)
         functionCalls.append(functionCall)
-      case .data, .fileData, .functionResponse:
+      case .data, .fileData, .functionResponse, .executableCode, .codeExecutionResult:
         fatalError("Unsupported response content.")
       }
     }
diff --git a/Tests/GoogleAITests/GenerativeModelTests.swift b/Tests/GoogleAITests/GenerativeModelTests.swift
@@ -285,7 +285,7 @@ final class GenerativeModelTests: XCTestCase {
     let functionCalls = response.functionCalls
     XCTAssertEqual(functionCalls.count, 2)
     let text = try XCTUnwrap(response.text)
-    XCTAssertEqual(text, "The sum of [1, 2, 3] is")
+    XCTAssertEqual(text, "The sum of [1, 2,\n3] is")
   }
 
   func testGenerateContent_usageMetadata() async throws {

Original file line number	Diff line number	Diff line change
`@@ -157,7 +157,7 @@ class FunctionCallingViewModel: ObservableObject {`
`157`	`157`	`case let .functionCall(functionCall):`
`158`	`158`	`messages.insert(functionCall.chatMessage(), at: messages.count - 1)`
`159`	`159`	`functionCalls.append(functionCall)`
`160`		`- case .data, .fileData, .functionResponse:`
	`160`	`+ case .data, .fileData, .functionResponse, .executableCode, .codeExecutionResult:`
`161`	`161`	`fatalError("Unsupported response content.")`
`162`	`162`	`}`
`163`	`163`	`}`
Original file line number	Diff line number	Diff line change
`@@ -285,7 +285,7 @@ final class GenerativeModelTests: XCTestCase {`
`285`	`285`	`let functionCalls = response.functionCalls`
`286`	`286`	`XCTAssertEqual(functionCalls.count, 2)`
`287`	`287`	`let text = try XCTUnwrap(response.text)`
`288`		`- XCTAssertEqual(text, "The sum of [1, 2, 3] is")`
	`288`	`+ XCTAssertEqual(text, "The sum of [1, 2,\n3] is")`
`289`	`289`	`}`
`290`	`290`
`291`	`291`	`func testGenerateContent_usageMetadata() async throws {`