Skip to content

Commit 131d1ca

Browse files
committed
fixes
1 parent 97fdb52 commit 131d1ca

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

FirebaseAI/Tests/TestApp/Tests/Integration/ServerPromptTemplateIntegrationTests.swift

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,8 @@ final class ServerPromptTemplateIntegrationTests: XCTestCase {
2323
}
2424

2525
func testGenerateContentWithText() async throws {
26-
let model = FirebaseAI.firebaseAI(backend: .vertexAI(location: "global")).templateGenerativeModel()
26+
let model = FirebaseAI.firebaseAI(backend: .vertexAI(location: "global"))
27+
.templateGenerativeModel()
2728
let userName = "paul"
2829
let response = try await model.generateContent(
2930
template: "greeting2",

FirebaseAI/Tests/Unit/TestUtilities/GenerativeModelTestUtil.swift

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ enum GenerativeModelTestUtil {
4949
let requestURL = try XCTUnwrap(request.url)
5050
if isTemplateRequest {
5151
XCTAssertEqual(
52-
requestURL.path.occurrenceCount(of: "templates/test-template.prompt:template"),
52+
requestURL.path.occurrenceCount(of: "templates/test-template:template"),
5353
1
5454
)
5555
} else {

0 commit comments

Comments
 (0)