mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-10-07 12:57:57 +08:00
fix: resolve Gemini compat test lint
This commit is contained in:
@@ -109,20 +109,26 @@ func TestConvertGeminiToClaudeMessageOmitsInlineDataForAnthropicMessages(t *test
|
||||
require.NoError(t, err)
|
||||
|
||||
withInlineData, _ := convertGeminiToClaudeMessage(geminiResp, "gemini-test", rawData, true)
|
||||
contentWithInlineData := withInlineData["content"].([]any)
|
||||
contentWithInlineData, ok := withInlineData["content"].([]any)
|
||||
require.True(t, ok)
|
||||
require.Len(t, contentWithInlineData, 4)
|
||||
require.Equal(t, map[string]any{"type": "text", "text": "before"}, contentWithInlineData[0])
|
||||
require.Equal(t, map[string]any{"type": "text", "text": ""}, contentWithInlineData[1])
|
||||
require.Equal(t, "tool_use", contentWithInlineData[2].(map[string]any)["type"])
|
||||
require.Equal(t, "get_weather", contentWithInlineData[2].(map[string]any)["name"])
|
||||
toolUse, ok := contentWithInlineData[2].(map[string]any)
|
||||
require.True(t, ok)
|
||||
require.Equal(t, "tool_use", toolUse["type"])
|
||||
require.Equal(t, "get_weather", toolUse["name"])
|
||||
require.Equal(t, map[string]any{"type": "text", "text": "after"}, contentWithInlineData[3])
|
||||
|
||||
withoutInlineData, _ := convertGeminiToClaudeMessage(geminiResp, "gemini-test", rawData, false)
|
||||
contentWithoutInlineData := withoutInlineData["content"].([]any)
|
||||
contentWithoutInlineData, ok := withoutInlineData["content"].([]any)
|
||||
require.True(t, ok)
|
||||
require.Len(t, contentWithoutInlineData, 3)
|
||||
require.Equal(t, map[string]any{"type": "text", "text": "before"}, contentWithoutInlineData[0])
|
||||
require.Equal(t, "tool_use", contentWithoutInlineData[1].(map[string]any)["type"])
|
||||
require.Equal(t, "get_weather", contentWithoutInlineData[1].(map[string]any)["name"])
|
||||
toolUseWithoutInlineData, ok := contentWithoutInlineData[1].(map[string]any)
|
||||
require.True(t, ok)
|
||||
require.Equal(t, "tool_use", toolUseWithoutInlineData["type"])
|
||||
require.Equal(t, "get_weather", toolUseWithoutInlineData["name"])
|
||||
require.Equal(t, map[string]any{"type": "text", "text": "after"}, contentWithoutInlineData[2])
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user