Skip to content

Commit 0840940

Browse files
committed
make helper func more flexible
1 parent 3021047 commit 0840940

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

Tests/OpenAITests/MockServerSentEvent.swift

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -21,15 +21,16 @@ struct MockServerSentEvent {
2121
"{\n \"error\": {\n \"message\": \"The model `o3-mini` does not exist or you do not have access to it.\",\n \"type\": \"invalid_request_error\",\n \"param\": null,\n \"code\": \"model_not_found\"\n }\n}\n".data(using: .utf8)!
2222
}
2323

24-
static func responseOutputTextDelta(
24+
static func responseStreamEvent(
2525
itemId: String = "msg_1",
26+
payloadType: String,
2627
outputIndex: Int = 0,
2728
contentIndex: Int = 0,
28-
delta: String = "Hi",
29+
delta: String = "",
2930
sequenceNumber: Int = 1
3031
) -> Data {
3132
let json = """
32-
{"type":"response.output_text.delta","output_index":\(outputIndex),"item_id":"\(itemId)","content_index":\(contentIndex),"delta":"\(delta)","sequence_number":\(sequenceNumber)}
33+
{"type":"\(payloadType)","output_index":\(outputIndex),"item_id":"\(itemId)","content_index":\(contentIndex),"delta":"\(delta)","sequence_number":\(sequenceNumber)}
3334
"""
3435
return "data: \(json)\n\n".data(using: .utf8)!
3536
}

Tests/OpenAITests/ModelResponseEventsStreamInterpreterTests.swift

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,8 +56,9 @@ final class ModelResponseEventsStreamInterpreterTests: XCTestCase {
5656
}
5757

5858
interpreter.processData(
59-
MockServerSentEvent.responseOutputTextDelta(
59+
MockServerSentEvent.responseStreamEvent(
6060
itemId: "msg_1",
61+
payloadType: "response.output_text.delta",
6162
outputIndex: 0,
6263
contentIndex: 0,
6364
delta: "Hi",

0 commit comments

Comments
 (0)