async generate()

in firestore-multimodal-genai/functions/src/generative-client/generative_ai.ts [25:95]
59 lines of code
8 McCabe index (conditional complexity)

  async generate(promptText: string, options: any): Promise<any> {
    if (!this.client) {
      throw new Error('Gemini Client not initialized.');
    }

    const model = this.client.getGenerativeModel({
      model: this.modelName,
    });
    const textPart: Part = {
      text: promptText,
    };
    const promptParts: Part[] = [textPart];

    if (config.imageField) {
      if (!options.image) {
        throw new Error('Vision model selected, but missing Image Field');
      }

      const base64String = await getImageBase64(options.image, 'google-ai');

      const imagePart = {
        inlineData: {
          mimeType: 'image/png',
          data: base64String,
        },
      };

      promptParts.push(imagePart);
    }
    let result;
    try {
      result = await model.generateContent({
        contents: [
          {
            role: Role.USER,
            parts: promptParts,
          },
        ],
        generationConfig: {
          topK: options.topK,
          topP: options.topP,
          temperature: options.temperature,
          candidateCount: options.candidateCount,
          maxOutputTokens: options.maxOutputTokens,
        },
        safetySettings: options.safetySettings,
      });
    } catch (e) {
      logger.error(e);
      // TODO: the error message provided exposes the API key, so we should handle this/ get the Gemini team to fix it their side.
      throw new Error(
        'failed to generate content, see function logs for details'
      );
    }

    const candidates =
      result.response.candidates?.map(c => c.content.parts[0].text) || [];

    if (!candidates || candidates.length === 0) {
      // TODO: handle blocked responses
      throw new Error('No candidates returned');
    }
    const firstCandidate = candidates[0];

    return {
      response: firstCandidate!,
      candidates: candidates,
      // TODO: add this as a feature:
      // safetyMetadata: promptFeedback,
    };
  }