fix: resolve Gemini compat test lint

This commit is contained in:
wucm667
2026-07-24 03:00:26 +08:00
parent 4dc0aedbfe
commit 8276fdf605
@@ -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": "![image](data:image/png;base64,aW1hZ2U=)"}, 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])
}