firestore-genai-chatbot/extension.yaml

name: firestore-genai-chatbot
version: 0.0.13
specVersion: v1beta

icon: icon.png

displayName: Build Chatbot with the Gemini API

description: >-
  Deploys customizable chatbots using Gemini models and Firestore.

license: Apache-2.0 # https://spdx.org/licenses/

tags:
  [
    ai,
    chatbot-ai,
    generative-ai,
    text-ai,
    generative-ai,
    vertex-ai,
    generative-models,
    llm,
    nlp,
    google-ai,
  ]

apis:
  - apiName: aiplatform.googleapis.com
    reason: For access to the Vertex AI Gemini API if this provider is chosen.

author:
  authorName: Google Cloud
  url: https://cloud.google.com/

contributors:
  - authorName: Invertase
    email: oss@invertase.io
    url: https://github.com/invertase
  - authorName: Jacob Cable
    email: jacob@invertase.io
    url: https://github.com/cabljac
  - authorName: Mais Alheraki
    email: mais@invertase.io
    url: https://github.com/pr-Mais

sourceUrl: https://github.com/GoogleCloudPlatform/firebase-extensions/tree/main/firestore-genai-chatbot
releaseNotesUrl: https://github.com/GoogleCloudPlatform/firebase-extensions/tree/main/firestore-genai-chatbot/CHANGELOG.md

billingRequired: true

roles:
  - role: datastore.user
    reason:
      Allows this extension to access Cloud Firestore to read and process added
      messages.
  - role: aiplatform.user
    reason:
      Allows this extension to access the Vertex AI API if this provider is
      chosen.

resources:
  - name: generateMessage
    type: firebaseextensions.v1beta.function
    description: >-
      Listens to Firestore data writes to generate conversations.
    properties:
      location: ${LOCATION}
      eventTrigger:
        eventType: providers/cloud.firestore/eventTypes/document.write
        resource: projects/${PROJECT_ID}/databases/(default)/documents/${COLLECTION_NAME}/{messageId}
      runtime: 'nodejs20'

params:
  - param: GENERATIVE_AI_PROVIDER
    label: Gemini API Provider
    description: >-
      This extension makes use of the Gemini family of generative models. For
      Google AI you will require an API key, whereas Vertex AI will authenticate
      using application default credentials. For more information see the
      [docs](https://firebase.google.com/docs/admin/setup#initialize-sdk).
    type: select
    options:
      - label: Google AI
        value: google-ai
      - label: Vertex AI
        value: vertex-ai
    required: true
    default: google-ai
    immutable: false

  - param: API_KEY
    label: Google AI API Key
    description: >-
      If you have selected Google AI as your provider, then this parameteris
      required. If you have instead selected Vertex AI, then this parameter is
      not required, and application default credentials will be used.
    type: secret
    required: false
    immutable: false

  - param: MODEL
    label: Gemini model
    description: >-
      Input the name of the Gemini model you would like to use. To view
      available models for each provider, see: [Vertex AI Gemini
      models](https://cloud.google.com/vertex-ai/docs/generative-ai/learn/models),
      [Google AI Gemini models](https://ai.google.dev/models/gemini)
    type: string
    required: true
    default: gemini-2.0-flash
    immutable: false

  - param: COLLECTION_NAME
    label: Firestore Collection Path
    description: >-
      Used to store conversation history represented as documents. This
      extension will listen to the specified collection(s) for new message
      documents.
    type: string
    validationRegex: '^[^/]+(/[^/]+/[^/]+)*$'
    validationErrorMessage: Must be a valid Cloud Firestore Collection
    default: generate
    required: true
    immutable: false

  - param: PROMPT_FIELD
    label: Prompt Field
    description: >-
      The field in the message document that contains the prompt.
    type: string
    default: prompt
    example: prompt
    required: true
    immutable: false

  - param: RESPONSE_FIELD
    label: Response Field
    description: >-
      The field in the message document into which to put the response.
    type: string
    default: response
    example: response
    required: true
    immutable: false

  - param: ORDER_FIELD
    label: Order Field
    description: >-
      The field by which to order when fetching conversation history. If absent
      when processing begins, the current timestamp will be written to this
      field. Sorting will be in descending order.
    type: string
    default: createTime
    example: createTime
    required: false
    immutable: false

  - param: LOCATION
    label: Cloud Functions location
    description: >-
      Where do you want to deploy the functions created for this extension? For
      help selecting a location, refer to the [location selection
      guide](https://firebase.google.com/docs/functions/locations). Note that
      Generative AI on Vertex AI is only available in the regions listed
      [here](https://cloud.google.com/vertex-ai/docs/generative-ai/learn/locations-genai).
      A list of languages and regions supported by the Gemini API on Google AI
      is [here](https://ai.google.dev/available_regions).
    type: select
    options:
      - label: Iowa (us-central1)
        value: us-central1
      - label: South Carolina (us-east1)
        value: us-east1
      - label: Northern Virginia (us-east4)
        value: us-east4
      - label: Los Angeles (us-west2)
        value: us-west2
      - label: Salt Lake City (us-west3)
        value: us-west3
      - label: Las Vegas (us-west4)
        value: us-west4
      - label: Warsaw (europe-central2)
        value: europe-central2
      - label: Belgium (europe-west1)
        value: europe-west1
      - label: London (europe-west2)
        value: europe-west2
      - label: Frankfurt (europe-west3)
        value: europe-west3
      - label: Zurich (europe-west6)
        value: europe-west6
      - label: Hong Kong (asia-east2)
        value: asia-east2
      - label: Tokyo (asia-northeast1)
        value: asia-northeast1
      - label: Osaka (asia-northeast2)
        value: asia-northeast2
      - label: Seoul (asia-northeast3)
        value: asia-northeast3
      - label: Mumbai (asia-south1)
        value: asia-south1
      - label: Jakarta (asia-southeast2)
        value: asia-southeast2
      - label: Montreal (northamerica-northeast1)
        value: northamerica-northeast1
      - label: Sao Paulo (southamerica-east1)
        value: southamerica-east1
      - label: Sydney (australia-southeast1)
        value: australia-southeast1
    required: true
    immutable: true

  - param: CONTEXT
    label: Context
    description: >-
      Contextual preamble for the generative AI model. A string giving context
      for the discussion.
    type: string
    required: false
    immutable: false

  - param: TEMPERATURE
    label: Temperature
    description: >-
      Controls the randomness of the output. Values can range over [0,1],
      inclusive. A value closer to 1 will produce responses that are more
      varied, while a value closer to 0 will typically result in less surprising
      responses from the model.
    type: string
    validationRegex: ^(?:0*(?:\.\d+)?|1(\.0*)?)$
    validationErrorMessage:
      Please specify a decimal representation of a number between 0 and 1.
    required: false
    immutable: false

  - param: TOP_P
    label: Nucleus sampling probability
    description: >-
      If specified, nucleus sampling will be used as the decoding strategy.
      Nucleus sampling considers the smallest set of tokens whose probability
      sum is at least a fixed value. Enter a value between 0 and 1.
    type: string
    validationRegex: ^(?:0*(?:\.\d+)?|1(\.0*)?)$
    validationErrorMessage:
      Please specify a decimal representation of a number between 0 and 1.
    required: false
    immutable: false

  - param: TOP_K
    label: Sampling strategy parameter
    description: >-
      If specified, top-k sampling will be used as the decoding strategy. Top-k
      sampling considers the set of topK most probable tokens.
    type: string
    validationRegex: ^[1-9][0-9]*
    validationErrorMessage: Please specify a positive integer.
    required: false
    immutable: false

  - param: CANDIDATE_COUNT
    label: Candidate count
    description: >-
      The default value is one. When set to an integer higher than one,
      additional candidate responses, up to the specified number, will be stored
      in Firestore under the 'candidates' field.
    type: string
    validationRegex: ^[1-9][0-9]*
    validationErrorMessage: Please specify a positive integer.
    default: 1
    required: false
    immutable: false

  - param: MAX_OUTPUT_TOKENS
    label: Max Output Tokens
    description: >-
      If specified, this parameter is passed to the Gemini API to control the
      length of the response.
    type: string
    validationRegex: ^[1-9][0-9]*
    validationErrorMessage: Please specify a positive integer.
    required: false
    immutable: false

  - param: CANDIDATES_FIELD
    label: Candidates field
    description: >-
      The field in the message document into which to put the other candidate
      responses if the candidate count parameter is greater than one.
    type: string
    default: candidates
    required: false
    immutable: false

  - param: ENABLE_DISCUSSION_OPTION_OVERRIDES
    label: Enable per document overrides.
    description: >-
      If set to \"Yes\", discussion parameters may be overwritten by fields in
      the discussion collection.
    type: select
    options:
      - label: Yes
        value: yes
      - label: No
        value: no
    default: no
    required: true

  - param: HARM_CATEGORY_HATE_SPEECH
    label: Hate Speech Threshold
    description: >-
      Threshold for hate speech content. Specify what probability level of hate
      speech content is blocked by the Gemini provider.
    type: select
    options:
      - label: Default
        value: HARM_BLOCK_THRESHOLD_UNSPECIFIED
      - label: Block low and above
        value: BLOCK_LOW_AND_ABOVE
      - label: Block medium and above
        value: BLOCK_MEDIUM_AND_ABOVE
      - label: Block only high
        value: BLOCK_ONLY_HIGH
      - label: Block none
        value: BLOCK_NONE
    required: false
    default: HARM_BLOCK_THRESHOLD_UNSPECIFIED
    immutable: false

  - param: HARM_CATEGORY_DANGEROUS_CONTENT
    label: Dangerous Content Threshold
    description: >-
      Threshold for dangerous content. Specify what probability level of
      dangerous content is blocked by the Gemini provider.
    type: select
    options:
      - label: Default
        value: HARM_BLOCK_THRESHOLD_UNSPECIFIED
      - label: Block low and above
        value: BLOCK_LOW_AND_ABOVE
      - label: Block medium and above
        value: BLOCK_MEDIUM_AND_ABOVE
      - label: Block only high
        value: BLOCK_ONLY_HIGH
      - label: Block none
        value: BLOCK_NONE
    required: false
    default: HARM_BLOCK_THRESHOLD_UNSPECIFIED
    immutable: false

  - param: HARM_CATEGORY_HARASSMENT
    label: harassment Content Threshold
    description: >-
      Threshold for harassment content. Specify what probability level of
      harassment content is blocked by the Gemini provider.
    type: select
    options:
      - label: Default
        value: HARM_BLOCK_THRESHOLD_UNSPECIFIED
      - label: Block low and above
        value: BLOCK_LOW_AND_ABOVE
      - label: Block medium and above
        value: BLOCK_MEDIUM_AND_ABOVE
      - label: Block only high
        value: BLOCK_ONLY_HIGH
      - label: Block none
        value: BLOCK_NONE
    required: false
    default: HARM_BLOCK_THRESHOLD_UNSPECIFIED
    immutable: false

  - param: HARM_CATEGORY_SEXUALLY_EXPLICIT
    label: Sexual Content Threshold
    description: >-
      Threshold for sexually explicit content. Specify what probability level of
      sexual content is blocked by the Gemini provider.
    type: select
    options:
      - label: Default
        value: HARM_BLOCK_THRESHOLD_UNSPECIFIED
      - label: Block low and above
        value: BLOCK_LOW_AND_ABOVE
      - label: Block medium and above
        value: BLOCK_MEDIUM_AND_ABOVE
      - label: Block only high
        value: BLOCK_ONLY_HIGH
      - label: Block none
        value: BLOCK_NONE
    required: false
    default: HARM_BLOCK_THRESHOLD_UNSPECIFIED
    immutable: false