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)
|
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) {
|
|||||||
|
|
||||||
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
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Extract stop_reason from delta
|
|
||||||
if evt.Delta != nil && evt.Delta.StopReason != "" {
|
|
||||||
// We'll use this in message_stop
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user