Browse Source

fix: aws text content blocks must be non-empty

Seefs 5 days ago
parent
commit
17d1224141
2 changed files with 76 additions and 75 deletions
  1. 4 1
      relay/channel/claude/relay-claude.go
  2. 72 74
      relay/channel/claude/relay_claude_test.go

+ 4 - 1
relay/channel/claude/relay-claude.go

@@ -250,6 +250,9 @@ func RequestOpenAI2ClaudeMessage(c *gin.Context, textRequest dto.GeneralOpenAIRe
 		}
 		if message.Role == "assistant" && message.ToolCalls != nil {
 			fmtMessage.ToolCalls = message.ToolCalls
+			if message.IsStringContent() && message.StringContent() == "" {
+				fmtMessage.SetNullContent()
+			}
 		}
 		if lastMessage.Role == message.Role && lastMessage.Role != "tool" {
 			if lastMessage.IsStringContent() && message.IsStringContent() {
@@ -258,7 +261,7 @@ func RequestOpenAI2ClaudeMessage(c *gin.Context, textRequest dto.GeneralOpenAIRe
 				formatMessages = formatMessages[:len(formatMessages)-1]
 			}
 		}
-		if fmtMessage.Content == nil {
+		if fmtMessage.Content == nil && !(message.Role == "assistant" && message.ToolCalls != nil) {
 			fmtMessage.SetStringContent("...")
 		}
 		formatMessages = append(formatMessages, fmtMessage)

+ 72 - 74
relay/channel/claude/relay_claude_test.go

@@ -5,6 +5,8 @@ import (
 	"testing"
 
 	"github.com/QuantumNous/new-api/dto"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 )
 
 func TestFormatClaudeResponseInfo_MessageStart(t *testing.T) {
@@ -26,28 +28,15 @@ func TestFormatClaudeResponseInfo_MessageStart(t *testing.T) {
 	}
 
 	ok := FormatClaudeResponseInfo(claudeResponse, nil, claudeInfo)
-	if !ok {
-		t.Fatal("expected true")
-	}
-	if claudeInfo.Usage.PromptTokens != 100 {
-		t.Errorf("PromptTokens = %d, want 100", claudeInfo.Usage.PromptTokens)
-	}
-	if claudeInfo.Usage.PromptTokensDetails.CachedTokens != 30 {
-		t.Errorf("CachedTokens = %d, want 30", claudeInfo.Usage.PromptTokensDetails.CachedTokens)
-	}
-	if claudeInfo.Usage.PromptTokensDetails.CachedCreationTokens != 50 {
-		t.Errorf("CachedCreationTokens = %d, want 50", claudeInfo.Usage.PromptTokensDetails.CachedCreationTokens)
-	}
-	if claudeInfo.ResponseId != "msg_123" {
-		t.Errorf("ResponseId = %s, want msg_123", claudeInfo.ResponseId)
-	}
-	if claudeInfo.Model != "claude-3-5-sonnet" {
-		t.Errorf("Model = %s, want claude-3-5-sonnet", claudeInfo.Model)
-	}
+	require.True(t, ok)
+	assert.Equal(t, 100, claudeInfo.Usage.PromptTokens)
+	assert.Equal(t, 30, claudeInfo.Usage.PromptTokensDetails.CachedTokens)
+	assert.Equal(t, 50, claudeInfo.Usage.PromptTokensDetails.CachedCreationTokens)
+	assert.Equal(t, "msg_123", claudeInfo.ResponseId)
+	assert.Equal(t, "claude-3-5-sonnet", claudeInfo.Model)
 }
 
 func TestFormatClaudeResponseInfo_MessageDelta_FullUsage(t *testing.T) {
-	// message_start 先积累 usage
 	claudeInfo := &ClaudeResponseInfo{
 		Usage: &dto.Usage{
 			PromptTokens: 100,
@@ -59,7 +48,6 @@ func TestFormatClaudeResponseInfo_MessageDelta_FullUsage(t *testing.T) {
 		},
 	}
 
-	// message_delta 带完整 usage(原生 Anthropic 场景)
 	claudeResponse := &dto.ClaudeResponse{
 		Type: "message_delta",
 		Usage: &dto.ClaudeUsage{
@@ -71,25 +59,14 @@ func TestFormatClaudeResponseInfo_MessageDelta_FullUsage(t *testing.T) {
 	}
 
 	ok := FormatClaudeResponseInfo(claudeResponse, nil, claudeInfo)
-	if !ok {
-		t.Fatal("expected true")
-	}
-	if claudeInfo.Usage.PromptTokens != 100 {
-		t.Errorf("PromptTokens = %d, want 100", claudeInfo.Usage.PromptTokens)
-	}
-	if claudeInfo.Usage.CompletionTokens != 200 {
-		t.Errorf("CompletionTokens = %d, want 200", claudeInfo.Usage.CompletionTokens)
-	}
-	if claudeInfo.Usage.TotalTokens != 300 {
-		t.Errorf("TotalTokens = %d, want 300", claudeInfo.Usage.TotalTokens)
-	}
-	if !claudeInfo.Done {
-		t.Error("expected Done = true")
-	}
+	require.True(t, ok)
+	assert.Equal(t, 100, claudeInfo.Usage.PromptTokens)
+	assert.Equal(t, 200, claudeInfo.Usage.CompletionTokens)
+	assert.Equal(t, 300, claudeInfo.Usage.TotalTokens)
+	assert.True(t, claudeInfo.Done)
 }
 
 func TestFormatClaudeResponseInfo_MessageDelta_OnlyOutputTokens(t *testing.T) {
-	// 模拟 Bedrock: message_start 已积累 usage
 	claudeInfo := &ClaudeResponseInfo{
 		Usage: &dto.Usage{
 			PromptTokens: 100,
@@ -103,53 +80,29 @@ func TestFormatClaudeResponseInfo_MessageDelta_OnlyOutputTokens(t *testing.T) {
 		},
 	}
 
-	// Bedrock 的 message_delta 只有 output_tokens,缺少 input_tokens 和 cache 字段
 	claudeResponse := &dto.ClaudeResponse{
 		Type: "message_delta",
 		Usage: &dto.ClaudeUsage{
 			OutputTokens: 200,
-			// InputTokens, CacheCreationInputTokens, CacheReadInputTokens 都是 0
 		},
 	}
 
 	ok := FormatClaudeResponseInfo(claudeResponse, nil, claudeInfo)
-	if !ok {
-		t.Fatal("expected true")
-	}
-	// PromptTokens 应保持 message_start 的值(因为 message_delta 的 InputTokens=0,不更新)
-	if claudeInfo.Usage.PromptTokens != 100 {
-		t.Errorf("PromptTokens = %d, want 100", claudeInfo.Usage.PromptTokens)
-	}
-	if claudeInfo.Usage.CompletionTokens != 200 {
-		t.Errorf("CompletionTokens = %d, want 200", claudeInfo.Usage.CompletionTokens)
-	}
-	if claudeInfo.Usage.TotalTokens != 300 {
-		t.Errorf("TotalTokens = %d, want 300", claudeInfo.Usage.TotalTokens)
-	}
-	// cache 字段应保持 message_start 的值
-	if claudeInfo.Usage.PromptTokensDetails.CachedTokens != 30 {
-		t.Errorf("CachedTokens = %d, want 30", claudeInfo.Usage.PromptTokensDetails.CachedTokens)
-	}
-	if claudeInfo.Usage.PromptTokensDetails.CachedCreationTokens != 50 {
-		t.Errorf("CachedCreationTokens = %d, want 50", claudeInfo.Usage.PromptTokensDetails.CachedCreationTokens)
-	}
-	if claudeInfo.Usage.ClaudeCacheCreation5mTokens != 10 {
-		t.Errorf("ClaudeCacheCreation5mTokens = %d, want 10", claudeInfo.Usage.ClaudeCacheCreation5mTokens)
-	}
-	if claudeInfo.Usage.ClaudeCacheCreation1hTokens != 20 {
-		t.Errorf("ClaudeCacheCreation1hTokens = %d, want 20", claudeInfo.Usage.ClaudeCacheCreation1hTokens)
-	}
-	if !claudeInfo.Done {
-		t.Error("expected Done = true")
-	}
+	require.True(t, ok)
+	assert.Equal(t, 100, claudeInfo.Usage.PromptTokens)
+	assert.Equal(t, 200, claudeInfo.Usage.CompletionTokens)
+	assert.Equal(t, 300, claudeInfo.Usage.TotalTokens)
+	assert.Equal(t, 30, claudeInfo.Usage.PromptTokensDetails.CachedTokens)
+	assert.Equal(t, 50, claudeInfo.Usage.PromptTokensDetails.CachedCreationTokens)
+	assert.Equal(t, 10, claudeInfo.Usage.ClaudeCacheCreation5mTokens)
+	assert.Equal(t, 20, claudeInfo.Usage.ClaudeCacheCreation1hTokens)
+	assert.True(t, claudeInfo.Done)
 }
 
 func TestFormatClaudeResponseInfo_NilClaudeInfo(t *testing.T) {
 	claudeResponse := &dto.ClaudeResponse{Type: "message_start"}
 	ok := FormatClaudeResponseInfo(claudeResponse, nil, nil)
-	if ok {
-		t.Error("expected false for nil claudeInfo")
-	}
+	assert.False(t, ok)
 }
 
 func TestFormatClaudeResponseInfo_ContentBlockDelta(t *testing.T) {
@@ -166,10 +119,55 @@ func TestFormatClaudeResponseInfo_ContentBlockDelta(t *testing.T) {
 	}
 
 	ok := FormatClaudeResponseInfo(claudeResponse, nil, claudeInfo)
-	if !ok {
-		t.Fatal("expected true")
+	require.True(t, ok)
+	assert.Equal(t, "hello", claudeInfo.ResponseText.String())
+}
+
+func TestRequestOpenAI2ClaudeMessage_AssistantToolCallWithEmptyContent(t *testing.T) {
+	request := dto.GeneralOpenAIRequest{
+		Model: "claude-opus-4-6",
+		Messages: []dto.Message{
+			{
+				Role:    "user",
+				Content: "what time is it",
+			},
+		},
 	}
-	if claudeInfo.ResponseText.String() != "hello" {
-		t.Errorf("ResponseText = %q, want %q", claudeInfo.ResponseText.String(), "hello")
+	assistantMessage := dto.Message{
+		Role:    "assistant",
+		Content: "",
+	}
+	assistantMessage.SetToolCalls([]dto.ToolCallRequest{
+		{
+			ID:   "call_1",
+			Type: "function",
+			Function: dto.FunctionRequest{
+				Name:      "get_current_time",
+				Arguments: "{}",
+			},
+		},
+	})
+	request.Messages = append(request.Messages, assistantMessage)
+
+	claudeRequest, err := RequestOpenAI2ClaudeMessage(nil, request)
+	require.NoError(t, err)
+	require.Len(t, claudeRequest.Messages, 2)
+
+	assistantClaudeMessage := claudeRequest.Messages[1]
+	assert.Equal(t, "assistant", assistantClaudeMessage.Role)
+
+	contentBlocks, ok := assistantClaudeMessage.Content.([]dto.ClaudeMediaMessage)
+	require.True(t, ok)
+	require.Len(t, contentBlocks, 1)
+
+	assert.Equal(t, "tool_use", contentBlocks[0].Type)
+	assert.Equal(t, "call_1", contentBlocks[0].Id)
+	assert.Equal(t, "get_current_time", contentBlocks[0].Name)
+	if assert.NotNil(t, contentBlocks[0].Input) {
+		_, isMap := contentBlocks[0].Input.(map[string]any)
+		assert.True(t, isMap)
+	}
+	if contentBlocks[0].Text != nil {
+		assert.NotEqual(t, "", *contentBlocks[0].Text)
 	}
 }