 Ver en ai.google.dev
|
 Ejecutar en Google Colab
|
 Ver código fuente en GitHub
|
Descripción general
En esta guía de inicio rápido, se muestra cómo usar la API de PaLM, que te da acceso a los modelos grandes de lenguaje más recientes de Google, con el SDK de PaLM Node.js específicamente para casos de uso centrados en diálogos, como chatbots.
Obtén una clave de API
Para comenzar, deberás obtener una clave de API. Configúralo como una variable de entorno:
import os
os.environ["API_KEY"] = "<YOUR API KEY>"
Instala el cliente de la API
En un directorio nuevo, inicializa un proyecto de Node.js con npm y, luego, instala la biblioteca 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.
A continuación, instala la biblioteca cliente:
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.
Cómo obtener una clave de API
Sigue las instrucciones de la página de configuración a fin de crear una clave de API para tu app. La necesitarás en el siguiente paso.
Generar mensajes
Crea un archivo nuevo index.js y agrega el siguiente código para proporcionar tu clave de API de PaLM a través de la variable de entorno API_KEY:
%%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
Luego, ejecuta la secuencia de comandos:
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.
Continúa la conversación
Para continuar una conversación después de un mensaje inicial, debes adjuntar el mensaje del candidato que se muestra, así como el siguiente mensaje:
%%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.