jiazhizhong/higress
1
0
Fork 0
mirror of https://github.com/alibaba/higress.git synced 2026-06-26 10:45:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(ai-proxy): include tool call type in Claude stream deltas (#3985)

Browse Source 
Signed-off-by: DENG <33118163+XinhhD@users.noreply.github.com>
Co-authored-by: woody <yaodiwu618@gmail.com>
This commit is contained in:
DENG
2026-06-17 12:18:18 +08:00
committed by GitHub
parent 7e11e2f320
commit 3c3a067f07
3 changed files with 45 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
plugins/wasm-go/extensions/ai-proxy/provider/claude_test.go
View File

@@ -443,6 +443,28 @@ func TestClaudeProvider_StreamPreservesNativeSignatureAndStopsForClaudeConversio
assert.Equal(t, "content_block_stop", events[0].Name)
}
func TestClaudeProvider_StreamToolCallArgumentDeltaIncludesFunctionType(t *testing.T) {
provider := &claudeProvider{}
ctx := newMockMultipartHttpContext()
index := 0
response := provider.streamResponseClaude2OpenAI(ctx, &claudeTextGenStreamResponse{
Type: "content_block_delta",
Index: &index,
Delta: &claudeTextGenDelta{
Type: "input_json_delta",
PartialJson: `{"path":"/tmp/example"}`,
},
})
require.NotNil(t, response)
require.Len(t, response.Choices, 1)
require.NotNil(t, response.Choices[0].Delta)
require.Len(t, response.Choices[0].Delta.ToolCalls, 1)
assert.Equal(t, "function", response.Choices[0].Delta.ToolCalls[0].Type)
assert.Equal(t, `{"path":"/tmp/example"}`, response.Choices[0].Delta.ToolCalls[0].Function.Arguments)
}
func TestClaudeProvider_BuildClaudeTextGenRequest_ClaudeCodeMode(t *testing.T) {
provider := &claudeProvider{
config: ProviderConfig{