refactor(test): improve type assertions in ops endpoint context tests
This commit is contained in:
@@ -285,11 +285,15 @@ func TestSetOpsEndpointContext_SetsContextKeys(t *testing.T) {
|
||||
|
||||
v, ok := c.Get(opsUpstreamModelKey)
|
||||
require.True(t, ok)
|
||||
require.Equal(t, "claude-3-5-sonnet-20241022", v.(string))
|
||||
vStr, ok := v.(string)
|
||||
require.True(t, ok)
|
||||
require.Equal(t, "claude-3-5-sonnet-20241022", vStr)
|
||||
|
||||
rt, ok := c.Get(opsRequestTypeKey)
|
||||
require.True(t, ok)
|
||||
require.Equal(t, int16(2), rt.(int16))
|
||||
rtVal, ok := rt.(int16)
|
||||
require.True(t, ok)
|
||||
require.Equal(t, int16(2), rtVal)
|
||||
}
|
||||
|
||||
func TestSetOpsEndpointContext_EmptyModelNotStored(t *testing.T) {
|
||||
@@ -305,7 +309,9 @@ func TestSetOpsEndpointContext_EmptyModelNotStored(t *testing.T) {
|
||||
|
||||
rt, ok := c.Get(opsRequestTypeKey)
|
||||
require.True(t, ok)
|
||||
require.Equal(t, int16(1), rt.(int16))
|
||||
rtVal, ok := rt.(int16)
|
||||
require.True(t, ok)
|
||||
require.Equal(t, int16(1), rtVal)
|
||||
}
|
||||
|
||||
func TestSetOpsEndpointContext_NilContext(t *testing.T) {
|
||||
|
||||
@@ -397,11 +397,6 @@ func anthToResHandleMessageDelta(evt *AnthropicStreamEvent, state *AnthropicEven
|
||||
}
|
||||
}
|
||||
|
||||
// Extract stop_reason from delta
|
||||
if evt.Delta != nil && evt.Delta.StopReason != "" {
|
||||
// We'll use this in message_stop
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user