ai-agent/internal/ai/llm_test.go

package ai

import (
	"encoding/json"
	"testing"

	openai "github.com/openai/openai-go/v3"
	"github.com/openai/openai-go/v3/shared"

	"code.tczkiot.com/wlw/ai-agent/internal/models"
)

func TestApplyProviderSpecificChatParamsIncludesDashScopeThinkingFlag(t *testing.T) {
	params := openai.ChatCompletionNewParams{
		Messages: []openai.ChatCompletionMessageParamUnion{
			{
				OfUser: &openai.ChatCompletionUserMessageParam{
					Content: openai.ChatCompletionUserMessageParamContentUnion{
						OfString: openai.String("hello"),
					},
				},
			},
		},
		Model: shared.ChatModel("qwen3.5-plus"),
	}

	applyProviderSpecificChatParams(&params, models.AIConfig{
		BaseURL:   "https://dashscope.aliyuncs.com/compatible-mode/v1",
		ModelName: "qwen3.5-plus",
	})

	raw, err := json.Marshal(params)
	if err != nil {
		t.Fatalf("marshal params: %v", err)
	}
	var body map[string]any
	if err := json.Unmarshal(raw, &body); err != nil {
		t.Fatalf("unmarshal params: %v", err)
	}
	if got, ok := body["enable_thinking"].(bool); !ok || got {
		t.Fatalf("expected enable_thinking=false in request body, got body=%s", raw)
	}
}

func TestApplyProviderSpecificChatParamsDisablesDeepSeekV4Thinking(t *testing.T) {
	params := openai.ChatCompletionNewParams{
		Messages: []openai.ChatCompletionMessageParamUnion{
			{
				OfUser: &openai.ChatCompletionUserMessageParam{
					Content: openai.ChatCompletionUserMessageParamContentUnion{
						OfString: openai.String("hello"),
					},
				},
			},
		},
		Model: shared.ChatModel("deepseek-v4-flash"),
	}

	applyProviderSpecificChatParams(&params, models.AIConfig{
		BaseURL:   "https://api.deepseek.com",
		ModelName: "deepseek-v4-flash",
	})

	raw, err := json.Marshal(params)
	if err != nil {
		t.Fatalf("marshal params: %v", err)
	}
	var body map[string]any
	if err := json.Unmarshal(raw, &body); err != nil {
		t.Fatalf("unmarshal params: %v", err)
	}
	thinking, ok := body["thinking"].(map[string]any)
	if !ok || thinking["type"] != "disabled" {
		t.Fatalf("expected thinking.type=disabled in request body, got body=%s", raw)
	}
}