wpmind/includes/SDK/SDKAdapter.php

<?php
/**
 * WP AI Client SDK 适配器
 *
 * 封装 SDK 调用，提供与 PublicAPI 兼容的接口。
 * SDK 使用异常而非 WP_Error，响应是 GenerativeAiResult 对象而非数组。
 * 本适配器负责两者之间的转换。
 *
 * @package WPMind\SDK
 * @since 3.6.0
 */

declare(strict_types=1);

namespace WPMind\SDK;

use WP_Error;

/**
 * SDK 适配器
 *
 * 将 WP AI Client SDK 的调用方式适配为 PublicAPI 兼容的数组格式。
 *
 * @since 3.6.0
 */
class SDKAdapter {

	/**
	 * SDK 内置 Provider 映射
	 *
	 * @var array<string, string>
	 */
	private const BUILTIN_PROVIDERS = [
		'openai'    => 'WordPress\\AiClient\\Providers\\ProviderImplementations\\OpenAi\\OpenAiProvider',
		'anthropic' => 'WordPress\\AiClient\\Providers\\ProviderImplementations\\Anthropic\\AnthropicProvider',
		'google'    => 'WordPress\\AiClient\\Providers\\ProviderImplementations\\Google\\GoogleProvider',
	];

	/**
	 * AI 对话
	 *
	 * @param array  $args     请求参数（messages, max_tokens, temperature, json_mode, tools, tool_choice）
	 * @param string $provider 服务商标识
	 * @param string $model    模型标识
	 * @return array|WP_Error
	 */
	public function chat(array $args, string $provider, string $model): array|WP_Error {
		// 检查 SDK 可用性
		if (!class_exists('WordPress\\AiClient\\AiClient')) {
			return new WP_Error('wpmind_sdk_unavailable', __('WP AI Client SDK 不可用', 'wpmind'));
		}

		// 解析 Provider 类名
		$provider_class = $this->resolve_provider_class($provider);

		try {
			$registry = \WordPress\AiClient\AiClient::defaultRegistry();

			// 获取模型实例
			$model_instance = null;
			if ($provider_class && $model !== 'auto' && $model !== 'default') {
				try {
					$model_instance = $registry->getProviderModel($provider_class, $model);
				} catch (\Exception $e) {
					// 模型不存在，尝试不指定模型
				}
			}

			// 构建 PromptBuilder
			$builder = \WordPress\AiClient\AiClient::prompt($args['messages']);

			if ($model_instance) {
				$builder->usingModel($model_instance);
			} elseif ($provider_class) {
				$builder->usingProvider($provider_class);
			}

			$builder->usingTemperature($args['temperature'] ?? 0.7);
			$builder->usingMaxTokens($args['max_tokens'] ?? 2000);

			// 提取 System instruction
			$system_msg = null;
			foreach ($args['messages'] as $msg) {
				if (($msg['role'] ?? '') === 'system') {
					$system_msg = $msg['content'] ?? '';
					break;
				}
			}
			if ($system_msg) {
				$builder->usingSystemInstruction($system_msg);
			}

			// JSON mode
			if (!empty($args['json_mode'])) {
				$builder->asJsonResponse();
			}

			// 执行请求
			$result = $builder->generateTextResult();

			// 提取 finish_reason
			$finish_reason = '';
			$candidates = $result->getCandidates();
			if (!empty($candidates)) {
				$fr = $candidates[0]->getFinishReason();
				if ($fr !== null) {
					$finish_reason = is_object($fr) && property_exists($fr, 'value') ? $fr->value : (string) $fr;
				}
			}

			return [
				'content'       => $result->toText(),
				'provider'      => $provider,
				'model'         => $model,
				'usage'         => $this->extract_token_usage($result),
				'finish_reason' => $finish_reason,
			];
		} catch (\InvalidArgumentException $e) {
			return new WP_Error('wpmind_sdk_invalid_args', $e->getMessage());
		} catch (\RuntimeException $e) {
			return $this->convert_exception_to_wp_error($e);
		} catch (\Exception $e) {
			return $this->convert_exception_to_wp_error($e);
		}
	}

	/**
	 * 安全提取 token 用量
	 *
	 * @param object $result SDK 结果对象
	 * @return array
	 */
	private function extract_token_usage(object $result): array {
		try {
			$usage = $result->getTokenUsage();
			if ($usage === null) {
				return ['prompt_tokens' => 0, 'completion_tokens' => 0, 'total_tokens' => 0];
			}
			return [
				'prompt_tokens'     => $usage->getPromptTokens() ?? 0,
				'completion_tokens' => $usage->getCompletionTokens() ?? 0,
				'total_tokens'      => $usage->getTotalTokens() ?? 0,
			];
		} catch (\Throwable $e) {
			return ['prompt_tokens' => 0, 'completion_tokens' => 0, 'total_tokens' => 0];
		}
	}

	/**
	 * 解析 Provider 类名
	 *
	 * 先检查 WPMind 注册的 Provider，再检查 SDK 内置 Provider。
	 *
	 * @param string $provider 服务商标识
	 * @return string|null Provider 完整类名，未找到返回 null
	 */
	private function resolve_provider_class(string $provider): ?string {
		// 先检查 WPMind 注册的 Provider
		if (class_exists('WPMind\\Providers\\ProviderRegistrar')) {
			$class = \WPMind\Providers\ProviderRegistrar::getProviderClass($provider);
			if ($class) {
				return $class;
			}
		}

		// 再检查 SDK 内置 Provider
		return self::BUILTIN_PROVIDERS[$provider] ?? null;
	}

	/**
	 * 将异常转换为 WP_Error
	 *
	 * 尝试从异常消息中提取 HTTP 状态码。
	 *
	 * @param \Exception $e 异常
	 * @return WP_Error
	 */
	private function convert_exception_to_wp_error(\Exception $e): WP_Error {
		$message = $e->getMessage();
		$status = 0;

		// 尝试从异常消息中提取 HTTP 状态码
		if (preg_match('/\b(4\d{2}|5\d{2})\b/', $message, $matches)) {
			$status = (int) $matches[1];
		}

		$error_data = [];
		if ($status > 0) {
			$error_data['status'] = $status;
		}

		// 仅在 debug 模式下记录完整异常信息
		if (defined('WP_DEBUG') && WP_DEBUG) {
			error_log(sprintf('[WPMind SDK] Exception: %s', $message));
		}

		// 对外返回通用描述，不暴露内部细节
		$user_message = $status > 0
			? sprintf(__('SDK 请求失败 (HTTP %d)', 'wpmind'), $status)
			: __('SDK 请求失败', 'wpmind');

		return new WP_Error(
			'wpmind_sdk_error',
			$user_message,
			$error_data
		);
	}
}