function getGenerativeSafetySettings()

in firestore-palm-gen-text/functions/src/config.ts [55:97]


function getGenerativeSafetySettings() {
  const {
    UNSPECIFIED_THRESHOLD,
    DEROGATORY_THRESHOLD,
    TOXICITY_THRESHOLD,
    VIOLENCE_THRESHOLD,
    SEXUAL_THRESHOLD,
    MEDICAL_THRESHOLD,
    DANGEROUS_THRESHOLD,
  } = process.env as Record<string, keyof typeof GLHarmBlockThreshold>;

  // Array to map categories to their environmental variables
  return [
    {
      category: GLHarmCategory.HARM_CATEGORY_UNSPECIFIED,
      threshold: UNSPECIFIED_THRESHOLD!,
    },
    {
      category: GLHarmCategory.HARM_CATEGORY_DEROGATORY,
      threshold: DEROGATORY_THRESHOLD!,
    },
    {
      category: GLHarmCategory.HARM_CATEGORY_TOXICITY,
      threshold: TOXICITY_THRESHOLD!,
    },
    {
      category: GLHarmCategory.HARM_CATEGORY_VIOLENCE,
      threshold: VIOLENCE_THRESHOLD!,
    },
    {
      category: GLHarmCategory.HARM_CATEGORY_SEXUAL,
      threshold: SEXUAL_THRESHOLD!,
    },
    {
      category: GLHarmCategory.HARM_CATEGORY_MEDICAL,
      threshold: MEDICAL_THRESHOLD!,
    },
    {
      category: GLHarmCategory.HARM_CATEGORY_DANGEROUS,
      threshold: DANGEROUS_THRESHOLD!,
    },
  ];
}