@@ -81,7 +81,8 @@ func TestStreamReaderRecvRaw(t *testing.T) {
81
81
func TestStreamReaderParsesErrorEvents (t * testing.T ) {
82
82
// Test case simulating Groq's error event format
83
83
errorEvent := `event: error
84
- data: {"error":{"message":"Invalid tool_call: tool \"name_unknown\" does not exist.","type":"invalid_request_error","code":"invalid_tool_call"}}
84
+ data: {"error":{"message":"Invalid tool_call: tool \"name_unknown\" does not exist.",` +
85
+ `"type":"invalid_request_error","code":"invalid_tool_call"}}
85
86
86
87
`
87
88
stream := & streamReader [ChatCompletionStreamResponse ]{
@@ -98,8 +99,8 @@ data: {"error":{"message":"Invalid tool_call: tool \"name_unknown\" does not exi
98
99
}
99
100
100
101
// Verify it's an APIError
101
- apiErr , ok := err .( * APIError )
102
- if ! ok {
102
+ var apiErr * APIError
103
+ if ! errors . As ( err , & apiErr ) {
103
104
t .Fatalf ("Expected APIError type but got %T: %v" , err , err )
104
105
}
105
106
@@ -145,8 +146,8 @@ data: [DONE]
145
146
}
146
147
147
148
// Verify it's an APIError
148
- apiErr , ok := err .( * APIError )
149
- if ! ok {
149
+ var apiErr * APIError
150
+ if ! errors . As ( err , & apiErr ) {
150
151
t .Fatalf ("Expected APIError type but got %T: %v" , err , err )
151
152
}
152
153
@@ -175,8 +176,8 @@ data: {"error":{"message":"Second error","type":"error_type_2"}}
175
176
if err1 == nil {
176
177
t .Fatal ("Expected first error but got nil" )
177
178
}
178
- apiErr1 , ok := err1 .( * APIError )
179
- if ! ok {
179
+ var apiErr1 * APIError
180
+ if ! errors . As ( err1 , & apiErr1 ) {
180
181
t .Fatalf ("Expected APIError type but got %T: %v" , err1 , err1 )
181
182
}
182
183
if apiErr1 .Message != "First error" {
@@ -188,8 +189,8 @@ data: {"error":{"message":"Second error","type":"error_type_2"}}
188
189
if err2 == nil {
189
190
t .Fatal ("Expected second error but got nil" )
190
191
}
191
- apiErr2 , ok := err2 .( * APIError )
192
- if ! ok {
192
+ var apiErr2 * APIError
193
+ if ! errors . As ( err2 , & apiErr2 ) {
193
194
t .Fatalf ("Expected APIError type but got %T: %v" , err2 , err2 )
194
195
}
195
196
if apiErr2 .Message != "Second error" {
0 commit comments