41 lines
1.4 KiB
Go
41 lines
1.4 KiB
Go
package ai
|
||
|
||
import (
|
||
"testing"
|
||
|
||
"github.com/stretchr/testify/require"
|
||
)
|
||
|
||
func TestExtractChatContent_String(t *testing.T) {
|
||
raw := []byte(`{"model":"grok-3","choices":[{"finish_reason":"stop","message":{"content":"你好世界"}}]}`)
|
||
text, meta, err := extractChatContent(raw)
|
||
require.NoError(t, err)
|
||
require.Equal(t, "你好世界", text)
|
||
require.Equal(t, "stop", meta.FinishReason)
|
||
}
|
||
|
||
func TestExtractChatContent_ArrayParts(t *testing.T) {
|
||
raw := []byte(`{"choices":[{"message":{"content":[{"type":"text","text":"第一段"},{"type":"text","text":"第二段"}]}}]}`)
|
||
text, _, err := extractChatContent(raw)
|
||
require.NoError(t, err)
|
||
require.Contains(t, text, "第一段")
|
||
require.Contains(t, text, "第二段")
|
||
}
|
||
|
||
func TestExtractChatContent_Empty(t *testing.T) {
|
||
raw := []byte(`{"choices":[{"finish_reason":"length","message":{"content":""}}]}`)
|
||
text, meta, err := extractChatContent(raw)
|
||
require.NoError(t, err)
|
||
require.Equal(t, "", text)
|
||
require.Equal(t, "length", meta.FinishReason)
|
||
}
|
||
|
||
func TestExtractChatContent_IgnoresReasoningOnly(t *testing.T) {
|
||
// deepseek / kimi 思考模型常只回 reasoning_content；不可當聊天回覆
|
||
raw := []byte(`{"choices":[{"finish_reason":"length","message":{"content":null,"reasoning_content":"我們需要根據最近的對話來回應…"}}]}`)
|
||
text, meta, err := extractChatContent(raw)
|
||
require.NoError(t, err)
|
||
require.Equal(t, "", text)
|
||
require.Equal(t, "length", meta.FinishReason)
|
||
}
|