import type { ProviderModelConfig } from "@earendil-works/pi-coding-agent";
import { OPENAI_COMPLETIONS_COMPAT } from "./_protocols.ts";
import { isAtLeastHigh, type PayloadWriter, type ThinkingApplier, type ThinkingLevel } from "../_tools.ts";

/**
 * 阿里云通义（Alibaba Qwen）系列。
 * 官方文档：https://help.aliyun.com/zh/model-studio/
 */
export function qwenModels(root: string): ProviderModelConfig[] {
	const v1 = `${root}/v1`;
	return [
		{
			id: "qwen3.7-plus",
			name: "Qwen3.7 Plus",
			api: "openai-completions",
			baseUrl: v1,
			reasoning: true,
			input: ["text", "image"],
			cost: {
				input: 0.295,
				output: 1.18,
				cacheRead: 0.059,
				cacheWrite: 0.36875,
			},
			contextWindow: 1000000,
			maxTokens: 64000,
			compat: OPENAI_COMPLETIONS_COMPAT,
		},
		{
			id: "qwen3.7-max",
			name: "Qwen3.7 Max",
			api: "openai-completions",
			baseUrl: v1,
			reasoning: true,
			input: ["text"],
			cost: { input: 1.77, output: 5.31, cacheRead: 0.354, cacheWrite: 2.2125 },
			contextWindow: 1000000,
			maxTokens: 64000,
			compat: OPENAI_COMPLETIONS_COMPAT,
		},
		{
			id: "qwen3.6-plus",
			name: "Qwen3.6 Plus",
			api: "openai-completions",
			baseUrl: v1,
			reasoning: true,
			input: ["text", "image"],
			// 阿里云百炼 CNY 定价按 0.1475 USD/CNY 换算（与 qwen3.7 系列一致）。
			cost: { input: 0.295, output: 1.77, cacheRead: 0.0295, cacheWrite: 0.36875 },
			contextWindow: 1000000,
			maxTokens: 65536,
			compat: OPENAI_COMPLETIONS_COMPAT,
		},
		{
			id: "qwen3.5-plus",
			name: "Qwen3.5 Plus",
			api: "openai-completions",
			baseUrl: v1,
			reasoning: true,
			input: ["text", "image"],
			// 阿里云百炼 CNY 定价按 0.1475 USD/CNY 换算（与 qwen3.7 系列一致）。
			cost: { input: 0.118, output: 0.708, cacheRead: 0.0118, cacheWrite: 0.1475 },
			contextWindow: 1000000,
			maxTokens: 65536,
			compat: OPENAI_COMPLETIONS_COMPAT,
		},
	];
}

// Qwen 思维链：`enable_thinking`/`preserve_thinking`，高级档位加预算与搜索/解释器。
function applyQwen(writer: PayloadWriter, level: ThinkingLevel): void {
	writer.remove("reasoning_effort");
	writer.set("enable_search", true);
	writer.set("parallel_tool_calls", true);
	if (level === "off") {
		writer.set("enable_thinking", false);
		writer.set("preserve_thinking", false);
		writer.remove("thinking_budget");
		writer.remove("search_options");
		writer.remove("enable_code_interpreter");
		return;
	}

	writer.set("enable_thinking", true);
	writer.set("preserve_thinking", true);
	if (isAtLeastHigh(level)) {
		writer.set("thinking_budget", 6000);
		writer.set("search_options", { search_strategy: "agent_max" });
		writer.set("enable_code_interpreter", true);
		return;
	}
	writer.remove("thinking_budget");
	writer.remove("search_options");
	writer.remove("enable_code_interpreter");
}

export const qwenThinking: Record<string, ThinkingApplier> = {
	"qwen3.7-plus": (w, c) => applyQwen(w, c.thinkingLevel),
	"qwen3.7-max": (w, c) => applyQwen(w, c.thinkingLevel),
	"qwen3.6-plus": (w, c) => applyQwen(w, c.thinkingLevel),
	"qwen3.5-plus": (w, c) => applyQwen(w, c.thinkingLevel),
};