Class SearchFeature

ai/ai.ts:4541–4692

class SearchFeature extends BaseFeatureHandler {
  readonly name = "联网搜索";
  readonly command = "search";
  readonly description = "使用联网能力搜索并回答";

  private aiService: AIService;
  private messageUtils: MessageUtils;

  constructor(
    aiService: AIService,
    configManagerPromise: Promise<ConfigManager>,
    httpClient: HttpClient,
  ) {
    super(configManagerPromise);
    this.aiService = aiService;
    this.messageUtils = new MessageUtils(configManagerPromise, httpClient);
  }

  async execute(
    msg: Api.Message,
    args: string[],
    _prefixes: string[],
  ): Promise<void> {
    const prefixes = getPrefixes();
    const config = await this.getConfig();

    const promptInput = args.slice(1).join(" ").trim();

    const replyMsg = await safeGetReplyMessage(msg);
    requireUser(!!promptInput || !!replyMsg, "至少需要一条提示");

    if (
      !config.currentSearchTag ||
      !config.currentSearchModel ||
      !config.configs[config.currentSearchTag]
    ) {
      throw new UserError(
        `请先配置 API 并设置模型\n使用 ${prefixes[0]}ai config add <tag> <url> <key> [type] 和 ${prefixes[0]}ai model search <tag> <model-path>`,
      );
    }

    await sendProcessing(msg, "search");

    const replyToId = replyMsg?.id;

    let context = getMessageText(replyMsg);
    const imageParts = [
      ...(await getMessageImageParts(replyMsg)),
      ...(await getMessageImageParts(msg)),
    ];

    const normalizedPrompt = promptInput.trim();
    const normalizedContext = (context || "").trim();

    if (
      normalizedPrompt &&
      normalizedContext &&
      normalizedPrompt === normalizedContext
    ) {