在 SociaLite 範例應用程式中新增生成式 AI 回覆

動畫螢幕截圖,顯示 SociaLite 聊天機器人運作情形

SociaLite 範例應用程式示範如何使用 Android 平台 API 實作社群網路和通訊應用程式中常見的功能。我們已使用 Google AI 用戶端 SDK 整合 Gemini API,以示範如何在您自己的 Android 應用程式中實作聊天機器人功能。

這個範例程式碼使用 Gemini Flash,可快速且經濟實惠地完成轉換。進一步瞭解 Gemini 模型。為了在 Socialite 示範中實作 AI 驅動的聊天機器人,我們使用了 Gemini API 的 系統指示功能,修改模型的行為。在這種情況下,我們會使用「Please respond to this chat conversation like a friendly cat」提示。這個含有 Gemini 的 SociaLite 版本也使用模型的多模態功能,讓聊天機器人回應圖片。

實作 Gemini API

聊天機器人實作內容主要位於 ChatRepository 類別中。GenerativeModel 類別可讓您與 Gemini API 互動,其例項化方式如下:

val generativeModel = GenerativeModel(
  // Set the model name to the latest Gemini model.
  modelName = "gemini-1.5-pro-latest",
  // Set your Gemini API key in the API_KEY variable in your
  // local.properties file
  apiKey = BuildConfig.API_KEY,
  // Set a system instruction to set the behavior of the model.
  systemInstruction = content {
    text("Please respond to this chat conversation like a friendly cat.")
  },
)

在協同程式範圍中,透過將 pastMessages 傳遞至 startChat() 來啟動即時通訊,確保模型可存取對話記錄。這樣一來,聊天機器人就能維持上下文,並根據先前的對話內容產生一致的回覆。

val pastMessages = getMessageHistory(chatId)
val chat = generativeModel.startChat(
  history = pastMessages,
)

使用 sendMessage() 方法將訊息傳遞至模型。

測試 AI 聊天機器人

您可以按照下列步驟自行測試:

  1. 查看 SociaLite 範例應用程式的程式碼,並在 Android Studio 中開啟該程式碼。
  2. Google AI Studio 中建立 Gemini API 金鑰。
  3. 將 Gemini API 金鑰做為 API_KEY 變數新增至 local.properties 檔案。
  4. 同步處理並執行應用程式。
  5. 在 SociaLite 應用程式中,依序輕觸「設定」 和「AI 聊天機器人」,直到按鈕標籤顯示「AI 聊天機器人:已啟用」

你現在可以開始即時通訊了!

其他資源

進一步瞭解 Google AI SDK。如果您的應用程式已整合 Firebase,您也可以透過 Vertex AI in Firebase 存取 Gemini API。