 前往 ai.google.dev 查看
|
 在 Google Colab 中執行
|
 在 GitHub 上查看原始碼 |
總覽
本快速入門導覽課程說明如何使用 PaLM API，這個 API 可讓您存取 Google 最新的大型語言模型，並搭配 PaLM Node.js SDK 專為以對話方塊為主的用途，例如聊天機器人。
取得 API 金鑰
如要開始使用，請取得 API 金鑰。將其設為環境變數：
import os
os.environ["API_KEY"] = "<YOUR API KEY>"
安裝 API 用戶端
在新目錄中，使用 npm 初始化 Node.js 專案，並安裝 google-auth 程式庫：
npm init -ynpm install google-auth-library
Wrote to /content/package.json:
{
"name": "content",
"version": "1.0.0",
"main": "index.js",
"scripts": {
"test": "echo \"Error: no test specified\" && exit 1"
},
"keywords": [],
"author": "",
"license": "ISC",
"dependencies": {
"@google-ai/generativelanguage": "^1.0.1",
"google-auth-library": "^9.0.0"
},
"devDependencies": {},
"description": ""
}
+ google-auth-library@9.0.0
updated 1 package and audited 74 packages in 0.905s
3 packages are looking for funding
run `npm fund` for details
found 0 vulnerabilities
npm WARN content@1.0.0 No description
npm WARN content@1.0.0 No repository field.
接著，安裝用戶端程式庫：
npm install @google-ai/generativelanguage
+ @google-ai/generativelanguage@1.0.1 updated 1 package and audited 74 packages in 2.274s 3 packages are looking for funding run `npm fund` for details found 0 vulnerabilities npm WARN content@1.0.0 No description npm WARN content@1.0.0 No repository field.
取得 API 金鑰
按照設定頁面中的操作說明，為應用程式建立 API 金鑰。下一步會用到這組 API 金鑰。
產生訊息
建立新檔案 index.js 並加入下列程式碼，透過 API_KEY 環境變數提供 PaLM API 金鑰：
%%writefile index.js
const { DiscussServiceClient } = require("@google-ai/generativelanguage");
const { GoogleAuth } = require("google-auth-library");
const MODEL_NAME = "models/chat-bison-001";
const API_KEY = process.env.API_KEY;
const client = new DiscussServiceClient({
authClient: new GoogleAuth().fromAPIKey(API_KEY),
});
async function main() {
const result = await client.generateMessage({
model: MODEL_NAME, // Required. The model to use to generate the result.
temperature: 0.5, // Optional. Value `0.0` always uses the highest-probability result.
candidateCount: 1, // Optional. The number of candidate results to generate.
prompt: {
// optional, preamble context to prime responses
context: "Respond to all questions with a rhyming poem.",
// Optional. Examples for further fine-tuning of responses.
examples: [
{
input: { content: "What is the capital of California?" },
output: {
content:
`If the capital of California is what you seek,
Sacramento is where you ought to peek.`,
},
},
],
// Required. Alternating prompt/response messages.
messages: [{ content: "How tall is the Eiffel Tower?" }],
},
});
console.log(result[0].candidates[0].content);
}
main();
Overwriting index.js
接著執行指令碼：
node index.js
The Eiffel Tower is 324 meters tall, It's a sight to behold, tall and not small. It's made of iron and weighs 10,100 tons, It's a wonder of the world, it's a must-see for all.
延續對話
如要在傳送初始提示訊息後繼續對話，您必須附加傳回的候選訊息，以及下一個提示：
%%writefile index.js
const { DiscussServiceClient } = require("@google-ai/generativelanguage");
const { GoogleAuth } = require("google-auth-library");
const MODEL_NAME = "models/chat-bison-001";
const API_KEY = process.env.API_KEY;
const client = new DiscussServiceClient({
authClient: new GoogleAuth().fromAPIKey(API_KEY),
});
async function main() {
let first = "Tell me a one short animal fact."
let messages = [{ content: first }];
const result = await client.generateMessage({
model: MODEL_NAME,
prompt: { messages },
});
console.log("User:\n\n", first, "\n\n")
console.log("Palm:\n\n", result[0].candidates[0].content, "\n\n");
let second = "Oh, where do those live?"
messages.push({ content: result[0].candidates[0].content });
messages.push({ content: second });
const secondResult = await client.generateMessage({
model: MODEL_NAME,
prompt: { messages },
});
console.log("User:\n\n", second, "\n\n")
console.log("Palm:\n\n", secondResult[0].candidates[0].content, "\n\n");
}
main();
Overwriting index.js
node index.js
User: Tell me a one short animal fact. Palm: The world's smallest mammal is the bumblebee bat, which weighs less than a penny. User: Oh, where do those live? Palm: The bumblebee bat is found in the rainforests of Thailand, Myanmar, and Laos. It is a small, nocturnal bat that feeds on insects. The bumblebee bat is the smallest mammal in the world, weighing only about 2 grams. It is about the size of a bumblebee, hence its name. The bumblebee bat is a very important part of the rainforest ecosystem. It helps to control insect populations and pollinate plants.