@google-ai/generativelanguage
Version:
Generative Language API client for Node.js
490 lines (489 loc) • 24.7 kB
TypeScript
import type * as gax from 'google-gax';
import type { Callback, CallOptions, Descriptors, ClientOptions } from 'google-gax';
import * as protos from '../../protos/protos';
/**
* API for using Generative Language Models (GLMs) trained to generate text.
*
* Also known as Large Language Models (LLM)s, these generate text given an
* input prompt from the user.
* @class
* @memberof v1beta
*/
export declare class TextServiceClient {
private _terminated;
private _opts;
private _providedCustomServicePath;
private _gaxModule;
private _gaxGrpc;
private _protos;
private _defaults;
private _universeDomain;
private _servicePath;
private _log;
auth: gax.GoogleAuth;
descriptors: Descriptors;
warn: (code: string, message: string, warnType?: string) => void;
innerApiCalls: {
[name: string]: Function;
};
pathTemplates: {
[name: string]: gax.PathTemplate;
};
textServiceStub?: Promise<{
[name: string]: Function;
}>;
/**
* Construct an instance of TextServiceClient.
*
* @param {object} [options] - The configuration object.
* The options accepted by the constructor are described in detail
* in [this document](https://github.com/googleapis/gax-nodejs/blob/main/client-libraries.md#creating-the-client-instance).
* The common options are:
* @param {object} [options.credentials] - Credentials object.
* @param {string} [options.credentials.client_email]
* @param {string} [options.credentials.private_key]
* @param {string} [options.email] - Account email address. Required when
* using a .pem or .p12 keyFilename.
* @param {string} [options.keyFilename] - Full path to the a .json, .pem, or
* .p12 key downloaded from the Google Developers Console. If you provide
* a path to a JSON file, the projectId option below is not necessary.
* NOTE: .pem and .p12 require you to specify options.email as well.
* @param {number} [options.port] - The port on which to connect to
* the remote host.
* @param {string} [options.projectId] - The project ID from the Google
* Developer's Console, e.g. 'grape-spaceship-123'. We will also check
* the environment variable GCLOUD_PROJECT for your project ID. If your
* app is running in an environment which supports
* {@link https://cloud.google.com/docs/authentication/application-default-credentials Application Default Credentials},
* your project ID will be detected automatically.
* @param {string} [options.apiEndpoint] - The domain name of the
* API remote host.
* @param {gax.ClientConfig} [options.clientConfig] - Client configuration override.
* Follows the structure of {@link gapicConfig}.
* @param {boolean} [options.fallback] - Use HTTP/1.1 REST mode.
* For more information, please check the
* {@link https://github.com/googleapis/gax-nodejs/blob/main/client-libraries.md#http11-rest-api-mode documentation}.
* @param {gax} [gaxInstance]: loaded instance of `google-gax`. Useful if you
* need to avoid loading the default gRPC version and want to use the fallback
* HTTP implementation. Load only fallback version and pass it to the constructor:
* ```
* const gax = require('google-gax/build/src/fallback'); // avoids loading google-gax with gRPC
* const client = new TextServiceClient({fallback: true}, gax);
* ```
*/
constructor(opts?: ClientOptions, gaxInstance?: typeof gax | typeof gax.fallback);
/**
* Initialize the client.
* Performs asynchronous operations (such as authentication) and prepares the client.
* This function will be called automatically when any class method is called for the
* first time, but if you need to initialize it before calling an actual method,
* feel free to call initialize() directly.
*
* You can await on this method if you want to make sure the client is initialized.
*
* @returns {Promise} A promise that resolves to an authenticated service stub.
*/
initialize(): Promise<{
[name: string]: Function;
}>;
/**
* The DNS address for this API service.
* @deprecated Use the apiEndpoint method of the client instance.
* @returns {string} The DNS address for this service.
*/
static get servicePath(): string;
/**
* The DNS address for this API service - same as servicePath.
* @deprecated Use the apiEndpoint method of the client instance.
* @returns {string} The DNS address for this service.
*/
static get apiEndpoint(): string;
/**
* The DNS address for this API service.
* @returns {string} The DNS address for this service.
*/
get apiEndpoint(): string;
get universeDomain(): string;
/**
* The port for this API service.
* @returns {number} The default port for this service.
*/
static get port(): number;
/**
* The scopes needed to make gRPC calls for every method defined
* in this service.
* @returns {string[]} List of default scopes.
*/
static get scopes(): never[];
getProjectId(): Promise<string>;
getProjectId(callback: Callback<string, undefined, undefined>): void;
/**
* Generates a response from the model given an input message.
*
* @param {Object} request
* The request object that will be sent.
* @param {string} request.model
* Required. The name of the `Model` or `TunedModel` to use for generating the
* completion.
* Examples:
* models/text-bison-001
* tunedModels/sentence-translator-u3b7m
* @param {google.ai.generativelanguage.v1beta.TextPrompt} request.prompt
* Required. The free-form input text given to the model as a prompt.
*
* Given a prompt, the model will generate a TextCompletion response it
* predicts as the completion of the input text.
* @param {number} [request.temperature]
* Optional. Controls the randomness of the output.
* Note: The default value varies by model, see the `Model.temperature`
* attribute of the `Model` returned the `getModel` function.
*
* Values can range from [0.0,1.0],
* inclusive. A value closer to 1.0 will produce responses that are more
* varied and creative, while a value closer to 0.0 will typically result in
* more straightforward responses from the model.
* @param {number} [request.candidateCount]
* Optional. Number of generated responses to return.
*
* This value must be between [1, 8], inclusive. If unset, this will default
* to 1.
* @param {number} [request.maxOutputTokens]
* Optional. The maximum number of tokens to include in a candidate.
*
* If unset, this will default to output_token_limit specified in the `Model`
* specification.
* @param {number} [request.topP]
* Optional. The maximum cumulative probability of tokens to consider when
* sampling.
*
* The model uses combined Top-k and nucleus sampling.
*
* Tokens are sorted based on their assigned probabilities so that only the
* most likely tokens are considered. Top-k sampling directly limits the
* maximum number of tokens to consider, while Nucleus sampling limits number
* of tokens based on the cumulative probability.
*
* Note: The default value varies by model, see the `Model.top_p`
* attribute of the `Model` returned the `getModel` function.
* @param {number} [request.topK]
* Optional. The maximum number of tokens to consider when sampling.
*
* The model uses combined Top-k and nucleus sampling.
*
* Top-k sampling considers the set of `top_k` most probable tokens.
* Defaults to 40.
*
* Note: The default value varies by model, see the `Model.top_k`
* attribute of the `Model` returned the `getModel` function.
* @param {number[]} [request.safetySettings]
* Optional. A list of unique `SafetySetting` instances for blocking unsafe
* content.
*
* that will be enforced on the `GenerateTextRequest.prompt` and
* `GenerateTextResponse.candidates`. There should not be more than one
* setting for each `SafetyCategory` type. The API will block any prompts and
* responses that fail to meet the thresholds set by these settings. This list
* overrides the default settings for each `SafetyCategory` specified in the
* safety_settings. If there is no `SafetySetting` for a given
* `SafetyCategory` provided in the list, the API will use the default safety
* setting for that category. Harm categories HARM_CATEGORY_DEROGATORY,
* HARM_CATEGORY_TOXICITY, HARM_CATEGORY_VIOLENCE, HARM_CATEGORY_SEXUAL,
* HARM_CATEGORY_MEDICAL, HARM_CATEGORY_DANGEROUS are supported in text
* service.
* @param {string[]} request.stopSequences
* The set of character sequences (up to 5) that will stop output generation.
* If specified, the API will stop at the first appearance of a stop
* sequence. The stop sequence will not be included as part of the response.
* @param {object} [options]
* Call options. See {@link https://googleapis.dev/nodejs/google-gax/latest/interfaces/CallOptions.html|CallOptions} for more details.
* @returns {Promise} - The promise which resolves to an array.
* The first element of the array is an object representing {@link protos.google.ai.generativelanguage.v1beta.GenerateTextResponse|GenerateTextResponse}.
* Please see the {@link https://github.com/googleapis/gax-nodejs/blob/master/client-libraries.md#regular-methods | documentation }
* for more details and examples.
* @example <caption>include:samples/generated/v1beta/text_service.generate_text.js</caption>
* region_tag:generativelanguage_v1beta_generated_TextService_GenerateText_async
*/
generateText(request?: protos.google.ai.generativelanguage.v1beta.IGenerateTextRequest, options?: CallOptions): Promise<[
protos.google.ai.generativelanguage.v1beta.IGenerateTextResponse,
(protos.google.ai.generativelanguage.v1beta.IGenerateTextRequest | undefined),
{} | undefined
]>;
generateText(request: protos.google.ai.generativelanguage.v1beta.IGenerateTextRequest, options: CallOptions, callback: Callback<protos.google.ai.generativelanguage.v1beta.IGenerateTextResponse, protos.google.ai.generativelanguage.v1beta.IGenerateTextRequest | null | undefined, {} | null | undefined>): void;
generateText(request: protos.google.ai.generativelanguage.v1beta.IGenerateTextRequest, callback: Callback<protos.google.ai.generativelanguage.v1beta.IGenerateTextResponse, protos.google.ai.generativelanguage.v1beta.IGenerateTextRequest | null | undefined, {} | null | undefined>): void;
/**
* Generates an embedding from the model given an input message.
*
* @param {Object} request
* The request object that will be sent.
* @param {string} request.model
* Required. The model name to use with the format model=models/{model}.
* @param {string} [request.text]
* Optional. The free-form input text that the model will turn into an
* embedding.
* @param {object} [options]
* Call options. See {@link https://googleapis.dev/nodejs/google-gax/latest/interfaces/CallOptions.html|CallOptions} for more details.
* @returns {Promise} - The promise which resolves to an array.
* The first element of the array is an object representing {@link protos.google.ai.generativelanguage.v1beta.EmbedTextResponse|EmbedTextResponse}.
* Please see the {@link https://github.com/googleapis/gax-nodejs/blob/master/client-libraries.md#regular-methods | documentation }
* for more details and examples.
* @example <caption>include:samples/generated/v1beta/text_service.embed_text.js</caption>
* region_tag:generativelanguage_v1beta_generated_TextService_EmbedText_async
*/
embedText(request?: protos.google.ai.generativelanguage.v1beta.IEmbedTextRequest, options?: CallOptions): Promise<[
protos.google.ai.generativelanguage.v1beta.IEmbedTextResponse,
protos.google.ai.generativelanguage.v1beta.IEmbedTextRequest | undefined,
{} | undefined
]>;
embedText(request: protos.google.ai.generativelanguage.v1beta.IEmbedTextRequest, options: CallOptions, callback: Callback<protos.google.ai.generativelanguage.v1beta.IEmbedTextResponse, protos.google.ai.generativelanguage.v1beta.IEmbedTextRequest | null | undefined, {} | null | undefined>): void;
embedText(request: protos.google.ai.generativelanguage.v1beta.IEmbedTextRequest, callback: Callback<protos.google.ai.generativelanguage.v1beta.IEmbedTextResponse, protos.google.ai.generativelanguage.v1beta.IEmbedTextRequest | null | undefined, {} | null | undefined>): void;
/**
* Generates multiple embeddings from the model given input text in a
* synchronous call.
*
* @param {Object} request
* The request object that will be sent.
* @param {string} request.model
* Required. The name of the `Model` to use for generating the embedding.
* Examples:
* models/embedding-gecko-001
* @param {string[]} [request.texts]
* Optional. The free-form input texts that the model will turn into an
* embedding. The current limit is 100 texts, over which an error will be
* thrown.
* @param {number[]} [request.requests]
* Optional. Embed requests for the batch. Only one of `texts` or `requests`
* can be set.
* @param {object} [options]
* Call options. See {@link https://googleapis.dev/nodejs/google-gax/latest/interfaces/CallOptions.html|CallOptions} for more details.
* @returns {Promise} - The promise which resolves to an array.
* The first element of the array is an object representing {@link protos.google.ai.generativelanguage.v1beta.BatchEmbedTextResponse|BatchEmbedTextResponse}.
* Please see the {@link https://github.com/googleapis/gax-nodejs/blob/master/client-libraries.md#regular-methods | documentation }
* for more details and examples.
* @example <caption>include:samples/generated/v1beta/text_service.batch_embed_text.js</caption>
* region_tag:generativelanguage_v1beta_generated_TextService_BatchEmbedText_async
*/
batchEmbedText(request?: protos.google.ai.generativelanguage.v1beta.IBatchEmbedTextRequest, options?: CallOptions): Promise<[
protos.google.ai.generativelanguage.v1beta.IBatchEmbedTextResponse,
(protos.google.ai.generativelanguage.v1beta.IBatchEmbedTextRequest | undefined),
{} | undefined
]>;
batchEmbedText(request: protos.google.ai.generativelanguage.v1beta.IBatchEmbedTextRequest, options: CallOptions, callback: Callback<protos.google.ai.generativelanguage.v1beta.IBatchEmbedTextResponse, protos.google.ai.generativelanguage.v1beta.IBatchEmbedTextRequest | null | undefined, {} | null | undefined>): void;
batchEmbedText(request: protos.google.ai.generativelanguage.v1beta.IBatchEmbedTextRequest, callback: Callback<protos.google.ai.generativelanguage.v1beta.IBatchEmbedTextResponse, protos.google.ai.generativelanguage.v1beta.IBatchEmbedTextRequest | null | undefined, {} | null | undefined>): void;
/**
* Runs a model's tokenizer on a text and returns the token count.
*
* @param {Object} request
* The request object that will be sent.
* @param {string} request.model
* Required. The model's resource name. This serves as an ID for the Model to
* use.
*
* This name should match a model name returned by the `ListModels` method.
*
* Format: `models/{model}`
* @param {google.ai.generativelanguage.v1beta.TextPrompt} request.prompt
* Required. The free-form input text given to the model as a prompt.
* @param {object} [options]
* Call options. See {@link https://googleapis.dev/nodejs/google-gax/latest/interfaces/CallOptions.html|CallOptions} for more details.
* @returns {Promise} - The promise which resolves to an array.
* The first element of the array is an object representing {@link protos.google.ai.generativelanguage.v1beta.CountTextTokensResponse|CountTextTokensResponse}.
* Please see the {@link https://github.com/googleapis/gax-nodejs/blob/master/client-libraries.md#regular-methods | documentation }
* for more details and examples.
* @example <caption>include:samples/generated/v1beta/text_service.count_text_tokens.js</caption>
* region_tag:generativelanguage_v1beta_generated_TextService_CountTextTokens_async
*/
countTextTokens(request?: protos.google.ai.generativelanguage.v1beta.ICountTextTokensRequest, options?: CallOptions): Promise<[
protos.google.ai.generativelanguage.v1beta.ICountTextTokensResponse,
(protos.google.ai.generativelanguage.v1beta.ICountTextTokensRequest | undefined),
{} | undefined
]>;
countTextTokens(request: protos.google.ai.generativelanguage.v1beta.ICountTextTokensRequest, options: CallOptions, callback: Callback<protos.google.ai.generativelanguage.v1beta.ICountTextTokensResponse, protos.google.ai.generativelanguage.v1beta.ICountTextTokensRequest | null | undefined, {} | null | undefined>): void;
countTextTokens(request: protos.google.ai.generativelanguage.v1beta.ICountTextTokensRequest, callback: Callback<protos.google.ai.generativelanguage.v1beta.ICountTextTokensResponse, protos.google.ai.generativelanguage.v1beta.ICountTextTokensRequest | null | undefined, {} | null | undefined>): void;
/**
* Return a fully-qualified cachedContent resource name string.
*
* @param {string} id
* @returns {string} Resource name string.
*/
cachedContentPath(id: string): string;
/**
* Parse the id from CachedContent resource.
*
* @param {string} cachedContentName
* A fully-qualified path representing CachedContent resource.
* @returns {string} A string representing the id.
*/
matchIdFromCachedContentName(cachedContentName: string): string | number;
/**
* Return a fully-qualified chunk resource name string.
*
* @param {string} corpus
* @param {string} document
* @param {string} chunk
* @returns {string} Resource name string.
*/
chunkPath(corpus: string, document: string, chunk: string): string;
/**
* Parse the corpus from Chunk resource.
*
* @param {string} chunkName
* A fully-qualified path representing Chunk resource.
* @returns {string} A string representing the corpus.
*/
matchCorpusFromChunkName(chunkName: string): string | number;
/**
* Parse the document from Chunk resource.
*
* @param {string} chunkName
* A fully-qualified path representing Chunk resource.
* @returns {string} A string representing the document.
*/
matchDocumentFromChunkName(chunkName: string): string | number;
/**
* Parse the chunk from Chunk resource.
*
* @param {string} chunkName
* A fully-qualified path representing Chunk resource.
* @returns {string} A string representing the chunk.
*/
matchChunkFromChunkName(chunkName: string): string | number;
/**
* Return a fully-qualified corpus resource name string.
*
* @param {string} corpus
* @returns {string} Resource name string.
*/
corpusPath(corpus: string): string;
/**
* Parse the corpus from Corpus resource.
*
* @param {string} corpusName
* A fully-qualified path representing Corpus resource.
* @returns {string} A string representing the corpus.
*/
matchCorpusFromCorpusName(corpusName: string): string | number;
/**
* Return a fully-qualified corpusPermission resource name string.
*
* @param {string} corpus
* @param {string} permission
* @returns {string} Resource name string.
*/
corpusPermissionPath(corpus: string, permission: string): string;
/**
* Parse the corpus from CorpusPermission resource.
*
* @param {string} corpusPermissionName
* A fully-qualified path representing corpus_permission resource.
* @returns {string} A string representing the corpus.
*/
matchCorpusFromCorpusPermissionName(corpusPermissionName: string): string | number;
/**
* Parse the permission from CorpusPermission resource.
*
* @param {string} corpusPermissionName
* A fully-qualified path representing corpus_permission resource.
* @returns {string} A string representing the permission.
*/
matchPermissionFromCorpusPermissionName(corpusPermissionName: string): string | number;
/**
* Return a fully-qualified document resource name string.
*
* @param {string} corpus
* @param {string} document
* @returns {string} Resource name string.
*/
documentPath(corpus: string, document: string): string;
/**
* Parse the corpus from Document resource.
*
* @param {string} documentName
* A fully-qualified path representing Document resource.
* @returns {string} A string representing the corpus.
*/
matchCorpusFromDocumentName(documentName: string): string | number;
/**
* Parse the document from Document resource.
*
* @param {string} documentName
* A fully-qualified path representing Document resource.
* @returns {string} A string representing the document.
*/
matchDocumentFromDocumentName(documentName: string): string | number;
/**
* Return a fully-qualified file resource name string.
*
* @param {string} file
* @returns {string} Resource name string.
*/
filePath(file: string): string;
/**
* Parse the file from File resource.
*
* @param {string} fileName
* A fully-qualified path representing File resource.
* @returns {string} A string representing the file.
*/
matchFileFromFileName(fileName: string): string | number;
/**
* Return a fully-qualified model resource name string.
*
* @param {string} model
* @returns {string} Resource name string.
*/
modelPath(model: string): string;
/**
* Parse the model from Model resource.
*
* @param {string} modelName
* A fully-qualified path representing Model resource.
* @returns {string} A string representing the model.
*/
matchModelFromModelName(modelName: string): string | number;
/**
* Return a fully-qualified tunedModel resource name string.
*
* @param {string} tuned_model
* @returns {string} Resource name string.
*/
tunedModelPath(tunedModel: string): string;
/**
* Parse the tuned_model from TunedModel resource.
*
* @param {string} tunedModelName
* A fully-qualified path representing TunedModel resource.
* @returns {string} A string representing the tuned_model.
*/
matchTunedModelFromTunedModelName(tunedModelName: string): string | number;
/**
* Return a fully-qualified tunedModelPermission resource name string.
*
* @param {string} tuned_model
* @param {string} permission
* @returns {string} Resource name string.
*/
tunedModelPermissionPath(tunedModel: string, permission: string): string;
/**
* Parse the tuned_model from TunedModelPermission resource.
*
* @param {string} tunedModelPermissionName
* A fully-qualified path representing tuned_model_permission resource.
* @returns {string} A string representing the tuned_model.
*/
matchTunedModelFromTunedModelPermissionName(tunedModelPermissionName: string): string | number;
/**
* Parse the permission from TunedModelPermission resource.
*
* @param {string} tunedModelPermissionName
* A fully-qualified path representing tuned_model_permission resource.
* @returns {string} A string representing the permission.
*/
matchPermissionFromTunedModelPermissionName(tunedModelPermissionName: string): string | number;
/**
* Terminate the gRPC channel and close the client.
*
* The client will no longer be usable and all future behavior is undefined.
* @returns {Promise} A promise that resolves when the client is closed.
*/
close(): Promise<void>;
}