Skip to content

Commit f10e56b

Browse files
committed
refactor(test): improve type assertions in ops endpoint context tests
1 parent 2f8e10d commit f10e56b

2 files changed

Lines changed: 9 additions & 8 deletions

File tree

backend/internal/handler/ops_error_logger_test.go

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -285,11 +285,15 @@ func TestSetOpsEndpointContext_SetsContextKeys(t *testing.T) {
285285

286286
v, ok := c.Get(opsUpstreamModelKey)
287287
require.True(t, ok)
288-
require.Equal(t, "claude-3-5-sonnet-20241022", v.(string))
288+
vStr, ok := v.(string)
289+
require.True(t, ok)
290+
require.Equal(t, "claude-3-5-sonnet-20241022", vStr)
289291

290292
rt, ok := c.Get(opsRequestTypeKey)
291293
require.True(t, ok)
292-
require.Equal(t, int16(2), rt.(int16))
294+
rtVal, ok := rt.(int16)
295+
require.True(t, ok)
296+
require.Equal(t, int16(2), rtVal)
293297
}
294298

295299
func TestSetOpsEndpointContext_EmptyModelNotStored(t *testing.T) {
@@ -305,7 +309,9 @@ func TestSetOpsEndpointContext_EmptyModelNotStored(t *testing.T) {
305309

306310
rt, ok := c.Get(opsRequestTypeKey)
307311
require.True(t, ok)
308-
require.Equal(t, int16(1), rt.(int16))
312+
rtVal, ok := rt.(int16)
313+
require.True(t, ok)
314+
require.Equal(t, int16(1), rtVal)
309315
}
310316

311317
func TestSetOpsEndpointContext_NilContext(t *testing.T) {

backend/internal/pkg/apicompat/anthropic_to_responses_response.go

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -397,11 +397,6 @@ func anthToResHandleMessageDelta(evt *AnthropicStreamEvent, state *AnthropicEven
397397
}
398398
}
399399

400-
// Extract stop_reason from delta
401-
if evt.Delta != nil && evt.Delta.StopReason != "" {
402-
// We'll use this in message_stop
403-
}
404-
405400
return nil
406401
}
407402

0 commit comments

Comments
 (0)