Commit f10e56be authored by Ethan0x0000's avatar Ethan0x0000
Browse files

refactor(test): improve type assertions in ops endpoint context tests

parent 2f8e10db
......@@ -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
}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment