Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,7 @@ public enum A2AParserError: Error {
/// Declared as a case-less enum to prevent instantiation.
public enum A2AResponseParser {
private static let a2uiJsonMimeType = "application/json+a2ui"
private static let a2uiV1MimeType = "application/a2ui+json"
private static let a2uiJsonTagOpen = "<a2ui-json>"
private static let a2uiJsonTagClose = "</a2ui-json>"

Expand Down Expand Up @@ -94,7 +95,7 @@ public enum A2AResponseParser {
mimeType
?? (isA2UIPayload(dataPayload) ? a2uiJsonMimeType : nil)

if resolvedMimeType == a2uiJsonMimeType {
if resolvedMimeType == a2uiJsonMimeType || resolvedMimeType == a2uiV1MimeType {
a2uiPayloads.append(dataPayload)
} else {
flushA2UI()
Expand Down
26 changes: 26 additions & 0 deletions client/ios/GoogleMapsA2UI/Tests/A2AResponseParserTests.swift
Original file line number Diff line number Diff line change
Expand Up @@ -187,6 +187,32 @@ final class A2AResponseParserTests: XCTestCase {
XCTAssertEqual(a2uiArray.count, 1)
}

/// Tests that a data part with the standard `application/a2ui+json` mime type is parsed and batched into an array.
func testParse_dataPartWithStandardA2UIMimeType() throws {
let payload: [String: Any] = [
"parts": [
[
"kind": "data",
"data": [
"version": "v0.9",
"updateComponents": [
"surfaceId": "sushi-seattle",
"components": [],
],
],
"metadata": ["mimeType": "application/a2ui+json"],
]
]
]

let events = try A2AResponseParser.parse(payload)
XCTAssertEqual(events.count, 1)
XCTAssertEqual(events.first?.metadataMimeType, "application/json+a2ui")

let a2uiArray = try XCTUnwrap(events.first?.dataArray)
XCTAssertEqual(a2uiArray.count, 1)
}

/// Tests that a data part is inferred as A2UI if it contains recognized keys, even without a mime type.
func testParse_dataPartWithImplicitA2UIKey() throws {
let payload: [String: Any] = [
Expand Down
Loading