Method generateGeminiImageRest

ai/ai.ts:3262–3379

(
    providerConfig: ProviderConfig,
    model: string,
    prompt: string,
    modeConfig: ProviderModeConfig,
    image?: AIImage,
    token?: AbortToken,
  )

  private async generateGeminiImageRest(
    providerConfig: ProviderConfig,
    model: string,
    prompt: string,
    modeConfig: ProviderModeConfig,
    image?: AIImage,
    token?: AbortToken,
  ): Promise<AIImage[]> {
    const baseUrl = resolveBaseUrl(providerConfig, modeConfig);

    if (model.includes("imagen")) {
      const endpoint = `v1beta/models/${model}:predict`;
      const url = resolveEndpointUrl(baseUrl, endpoint);
      const authMode = resolveAuthMode(
        getProviderProfile(providerConfig),
        modeConfig,
        providerConfig,
      );
      const authConfig = applyAuthConfig(authMode, providerConfig, url, {
        "Content-Type": "application/json",
      });

      const data: any = {
        instances: [{ prompt: prompt || "" }],
        parameters: {
          sampleCount: 1,
          outputOptions: { mimeType: "image/png" },
        },
      };

      const response = await this.httpClient.request(
        {
          url: authConfig.url,
          method: "POST",
          headers: authConfig.headers,
          data,
        },
        token,
      );

      const predictions = response.data?.predictions || [];
      const images: AIImage[] = [];
      for (const p of predictions) {
        if (p.bytesBase64Encoded) {
          images.push({
            data: Buffer.from(p.bytesBase64Encoded, "base64"),
            mimeType: p.mimeType || "image/png",
          });
        }
      }
      if (images.length === 0) throw new Error("图片生成失败");
      return images;
    }

    const endpoint = (
      modeConfig.endpoint || "models/{model}:generateContent"
    ).replace("{model}", model);
    const url = resolveEndpointUrl(baseUrl, endpoint);