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) { ...@@ -285,11 +285,15 @@ func TestSetOpsEndpointContext_SetsContextKeys(t *testing.T) {
v, ok := c.Get(opsUpstreamModelKey) v, ok := c.Get(opsUpstreamModelKey)
require.True(t, ok) 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) rt, ok := c.Get(opsRequestTypeKey)
require.True(t, ok) 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) { func TestSetOpsEndpointContext_EmptyModelNotStored(t *testing.T) {
...@@ -305,7 +309,9 @@ func TestSetOpsEndpointContext_EmptyModelNotStored(t *testing.T) { ...@@ -305,7 +309,9 @@ func TestSetOpsEndpointContext_EmptyModelNotStored(t *testing.T) {
rt, ok := c.Get(opsRequestTypeKey) rt, ok := c.Get(opsRequestTypeKey)
require.True(t, ok) 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) { func TestSetOpsEndpointContext_NilContext(t *testing.T) {
......
...@@ -397,11 +397,6 @@ func anthToResHandleMessageDelta(evt *AnthropicStreamEvent, state *AnthropicEven ...@@ -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 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