@firebase/ai 1.4.0 → 1.4.1-canary.25b60fdaa

package/dist/esm/src/types/enums.d.ts

@@ -28,221 +28,263 @@ export declare const POSSIBLE_ROLES: readonly ["user", "model", "function", "sys
  * Harm categories that would cause prompts or candidates to be blocked.
  * @public
  */
-export declare enum HarmCategory {
-    HARM_CATEGORY_HATE_SPEECH = "HARM_CATEGORY_HATE_SPEECH",
-    HARM_CATEGORY_SEXUALLY_EXPLICIT = "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-    HARM_CATEGORY_HARASSMENT = "HARM_CATEGORY_HARASSMENT",
-    HARM_CATEGORY_DANGEROUS_CONTENT = "HARM_CATEGORY_DANGEROUS_CONTENT"
-}
+export declare const HarmCategory: {
+    readonly HARM_CATEGORY_HATE_SPEECH: "HARM_CATEGORY_HATE_SPEECH";
+    readonly HARM_CATEGORY_SEXUALLY_EXPLICIT: "HARM_CATEGORY_SEXUALLY_EXPLICIT";
+    readonly HARM_CATEGORY_HARASSMENT: "HARM_CATEGORY_HARASSMENT";
+    readonly HARM_CATEGORY_DANGEROUS_CONTENT: "HARM_CATEGORY_DANGEROUS_CONTENT";
+};
+/**
+ * Harm categories that would cause prompts or candidates to be blocked.
+ * @public
+ */
+export type HarmCategory = (typeof HarmCategory)[keyof typeof HarmCategory];
 /**
  * Threshold above which a prompt or candidate will be blocked.
  * @public
  */
-export declare enum HarmBlockThreshold {
+export declare const HarmBlockThreshold: {
     /**
      * Content with `NEGLIGIBLE` will be allowed.
      */
-    BLOCK_LOW_AND_ABOVE = "BLOCK_LOW_AND_ABOVE",
+    readonly BLOCK_LOW_AND_ABOVE: "BLOCK_LOW_AND_ABOVE";
     /**
      * Content with `NEGLIGIBLE` and `LOW` will be allowed.
      */
-    BLOCK_MEDIUM_AND_ABOVE = "BLOCK_MEDIUM_AND_ABOVE",
+    readonly BLOCK_MEDIUM_AND_ABOVE: "BLOCK_MEDIUM_AND_ABOVE";
     /**
      * Content with `NEGLIGIBLE`, `LOW`, and `MEDIUM` will be allowed.
      */
-    BLOCK_ONLY_HIGH = "BLOCK_ONLY_HIGH",
+    readonly BLOCK_ONLY_HIGH: "BLOCK_ONLY_HIGH";
     /**
      * All content will be allowed.
      */
-    BLOCK_NONE = "BLOCK_NONE",
+    readonly BLOCK_NONE: "BLOCK_NONE";
     /**
      * All content will be allowed. This is the same as `BLOCK_NONE`, but the metadata corresponding
      * to the {@link HarmCategory} will not be present in the response.
      */
-    OFF = "OFF"
-}
+    readonly OFF: "OFF";
+};
+/**
+ * Threshold above which a prompt or candidate will be blocked.
+ * @public
+ */
+export type HarmBlockThreshold = (typeof HarmBlockThreshold)[keyof typeof HarmBlockThreshold];
 /**
  * This property is not supported in the Gemini Developer API ({@link GoogleAIBackend}).
  *
  * @public
  */
-export declare enum HarmBlockMethod {
+export declare const HarmBlockMethod: {
     /**
      * The harm block method uses both probability and severity scores.
      */
-    SEVERITY = "SEVERITY",
+    readonly SEVERITY: "SEVERITY";
     /**
      * The harm block method uses the probability score.
      */
-    PROBABILITY = "PROBABILITY"
-}
+    readonly PROBABILITY: "PROBABILITY";
+};
+/**
+ * This property is not supported in the Gemini Developer API ({@link GoogleAIBackend}).
+ *
+ * @public
+ */
+export type HarmBlockMethod = (typeof HarmBlockMethod)[keyof typeof HarmBlockMethod];
 /**
  * Probability that a prompt or candidate matches a harm category.
  * @public
  */
-export declare enum HarmProbability {
+export declare const HarmProbability: {
     /**
      * Content has a negligible chance of being unsafe.
      */
-    NEGLIGIBLE = "NEGLIGIBLE",
+    readonly NEGLIGIBLE: "NEGLIGIBLE";
     /**
      * Content has a low chance of being unsafe.
      */
-    LOW = "LOW",
+    readonly LOW: "LOW";
     /**
      * Content has a medium chance of being unsafe.
      */
-    MEDIUM = "MEDIUM",
+    readonly MEDIUM: "MEDIUM";
     /**
      * Content has a high chance of being unsafe.
      */
-    HIGH = "HIGH"
-}
+    readonly HIGH: "HIGH";
+};
+/**
+ * Probability that a prompt or candidate matches a harm category.
+ * @public
+ */
+export type HarmProbability = (typeof HarmProbability)[keyof typeof HarmProbability];
 /**
  * Harm severity levels.
  * @public
  */
-export declare enum HarmSeverity {
+export declare const HarmSeverity: {
     /**
      * Negligible level of harm severity.
      */
-    HARM_SEVERITY_NEGLIGIBLE = "HARM_SEVERITY_NEGLIGIBLE",
+    readonly HARM_SEVERITY_NEGLIGIBLE: "HARM_SEVERITY_NEGLIGIBLE";
     /**
      * Low level of harm severity.
      */
-    HARM_SEVERITY_LOW = "HARM_SEVERITY_LOW",
+    readonly HARM_SEVERITY_LOW: "HARM_SEVERITY_LOW";
     /**
      * Medium level of harm severity.
      */
-    HARM_SEVERITY_MEDIUM = "HARM_SEVERITY_MEDIUM",
+    readonly HARM_SEVERITY_MEDIUM: "HARM_SEVERITY_MEDIUM";
     /**
      * High level of harm severity.
      */
-    HARM_SEVERITY_HIGH = "HARM_SEVERITY_HIGH",
+    readonly HARM_SEVERITY_HIGH: "HARM_SEVERITY_HIGH";
     /**
      * Harm severity is not supported.
      *
      * @remarks
      * The GoogleAI backend does not support `HarmSeverity`, so this value is used as a fallback.
      */
-    HARM_SEVERITY_UNSUPPORTED = "HARM_SEVERITY_UNSUPPORTED"
-}
+    readonly HARM_SEVERITY_UNSUPPORTED: "HARM_SEVERITY_UNSUPPORTED";
+};
+/**
+ * Harm severity levels.
+ * @public
+ */
+export type HarmSeverity = (typeof HarmSeverity)[keyof typeof HarmSeverity];
 /**
  * Reason that a prompt was blocked.
  * @public
  */
-export declare enum BlockReason {
+export declare const BlockReason: {
     /**
      * Content was blocked by safety settings.
      */
-    SAFETY = "SAFETY",
+    readonly SAFETY: "SAFETY";
     /**
      * Content was blocked, but the reason is uncategorized.
      */
-    OTHER = "OTHER",
+    readonly OTHER: "OTHER";
     /**
      * Content was blocked because it contained terms from the terminology blocklist.
      */
-    BLOCKLIST = "BLOCKLIST",
+    readonly BLOCKLIST: "BLOCKLIST";
     /**
      * Content was blocked due to prohibited content.
      */
-    PROHIBITED_CONTENT = "PROHIBITED_CONTENT"
-}
+    readonly PROHIBITED_CONTENT: "PROHIBITED_CONTENT";
+};
+/**
+ * Reason that a prompt was blocked.
+ * @public
+ */
+export type BlockReason = (typeof BlockReason)[keyof typeof BlockReason];
 /**
  * Reason that a candidate finished.
  * @public
  */
-export declare enum FinishReason {
+export declare const FinishReason: {
     /**
      * Natural stop point of the model or provided stop sequence.
      */
-    STOP = "STOP",
+    readonly STOP: "STOP";
     /**
      * The maximum number of tokens as specified in the request was reached.
      */
-    MAX_TOKENS = "MAX_TOKENS",
+    readonly MAX_TOKENS: "MAX_TOKENS";
     /**
      * The candidate content was flagged for safety reasons.
      */
-    SAFETY = "SAFETY",
+    readonly SAFETY: "SAFETY";
     /**
      * The candidate content was flagged for recitation reasons.
      */
-    RECITATION = "RECITATION",
+    readonly RECITATION: "RECITATION";
     /**
      * Unknown reason.
      */
-    OTHER = "OTHER",
+    readonly OTHER: "OTHER";
     /**
      * The candidate content contained forbidden terms.
      */
-    BLOCKLIST = "BLOCKLIST",
+    readonly BLOCKLIST: "BLOCKLIST";
     /**
      * The candidate content potentially contained prohibited content.
      */
-    PROHIBITED_CONTENT = "PROHIBITED_CONTENT",
+    readonly PROHIBITED_CONTENT: "PROHIBITED_CONTENT";
     /**
      * The candidate content potentially contained Sensitive Personally Identifiable Information (SPII).
      */
-    SPII = "SPII",
+    readonly SPII: "SPII";
     /**
      * The function call generated by the model was invalid.
      */
-    MALFORMED_FUNCTION_CALL = "MALFORMED_FUNCTION_CALL"
-}
+    readonly MALFORMED_FUNCTION_CALL: "MALFORMED_FUNCTION_CALL";
+};
+/**
+ * Reason that a candidate finished.
+ * @public
+ */
+export type FinishReason = (typeof FinishReason)[keyof typeof FinishReason];
 /**
  * @public
  */
-export declare enum FunctionCallingMode {
+export declare const FunctionCallingMode: {
     /**
      * Default model behavior; model decides to predict either a function call
      * or a natural language response.
      */
-    AUTO = "AUTO",
+    readonly AUTO: "AUTO";
     /**
      * Model is constrained to always predicting a function call only.
      * If `allowed_function_names` is set, the predicted function call will be
      * limited to any one of `allowed_function_names`, else the predicted
      * function call will be any one of the provided `function_declarations`.
      */
-    ANY = "ANY",
+    readonly ANY: "ANY";
     /**
      * Model will not predict any function call. Model behavior is same as when
      * not passing any function declarations.
      */
-    NONE = "NONE"
-}
+    readonly NONE: "NONE";
+};
+export type FunctionCallingMode = (typeof FunctionCallingMode)[keyof typeof FunctionCallingMode];
 /**
  * Content part modality.
  * @public
  */
-export declare enum Modality {
+export declare const Modality: {
     /**
      * Unspecified modality.
      */
-    MODALITY_UNSPECIFIED = "MODALITY_UNSPECIFIED",
+    readonly MODALITY_UNSPECIFIED: "MODALITY_UNSPECIFIED";
     /**
      * Plain text.
      */
-    TEXT = "TEXT",
+    readonly TEXT: "TEXT";
     /**
      * Image.
      */
-    IMAGE = "IMAGE",
+    readonly IMAGE: "IMAGE";
     /**
      * Video.
      */
-    VIDEO = "VIDEO",
+    readonly VIDEO: "VIDEO";
     /**
      * Audio.
      */
-    AUDIO = "AUDIO",
+    readonly AUDIO: "AUDIO";
     /**
      * Document (for example, PDF).
      */
-    DOCUMENT = "DOCUMENT"
-}
+    readonly DOCUMENT: "DOCUMENT";
+};
+/**
+ * Content part modality.
+ * @public
+ */
+export type Modality = (typeof Modality)[keyof typeof Modality];
 /**
  * Generation modalities to be returned in generation responses.
  *

package/dist/esm/src/types/error.d.ts

@@ -51,31 +51,37 @@ export interface CustomErrorData {
  *
  * @public
  */
-export declare const enum AIErrorCode {
+export declare const AIErrorCode: {
     /** A generic error occurred. */
-    ERROR = "error",
+    readonly ERROR: "error";
     /** An error occurred in a request. */
-    REQUEST_ERROR = "request-error",
+    readonly REQUEST_ERROR: "request-error";
     /** An error occurred in a response. */
-    RESPONSE_ERROR = "response-error",
+    readonly RESPONSE_ERROR: "response-error";
     /** An error occurred while performing a fetch. */
-    FETCH_ERROR = "fetch-error",
+    readonly FETCH_ERROR: "fetch-error";
     /** An error associated with a Content object.  */
-    INVALID_CONTENT = "invalid-content",
+    readonly INVALID_CONTENT: "invalid-content";
     /** An error due to the Firebase API not being enabled in the Console. */
-    API_NOT_ENABLED = "api-not-enabled",
+    readonly API_NOT_ENABLED: "api-not-enabled";
     /** An error due to invalid Schema input.  */
-    INVALID_SCHEMA = "invalid-schema",
+    readonly INVALID_SCHEMA: "invalid-schema";
     /** An error occurred due to a missing Firebase API key. */
-    NO_API_KEY = "no-api-key",
+    readonly NO_API_KEY: "no-api-key";
     /** An error occurred due to a missing Firebase app ID. */
-    NO_APP_ID = "no-app-id",
+    readonly NO_APP_ID: "no-app-id";
     /** An error occurred due to a model name not being specified during initialization. */
-    NO_MODEL = "no-model",
+    readonly NO_MODEL: "no-model";
     /** An error occurred due to a missing project ID. */
-    NO_PROJECT_ID = "no-project-id",
+    readonly NO_PROJECT_ID: "no-project-id";
     /** An error occurred while parsing. */
-    PARSE_FAILED = "parse-failed",
+    readonly PARSE_FAILED: "parse-failed";
     /** An error occurred due an attempt to use an unsupported feature. */
-    UNSUPPORTED = "unsupported"
-}
+    readonly UNSUPPORTED: "unsupported";
+};
+/**
+ * Standardized error codes that {@link AIError} can have.
+ *
+ * @public
+ */
+export type AIErrorCode = (typeof AIErrorCode)[keyof typeof AIErrorCode];

package/dist/esm/src/types/imagen/requests.d.ts

@@ -70,7 +70,7 @@ export interface ImagenGenerationConfig {
     numberOfImages?: number;
     /**
      * The aspect ratio of the generated images. The default value is square 1:1.
-     * Supported aspect ratios depend on the Imagen model, see {@link ImagenAspectRatio}
+     * Supported aspect ratios depend on the Imagen model, see {@link (ImagenAspectRatio:type)}
      * for more details.
      */
     aspectRatio?: ImagenAspectRatio;
@@ -106,27 +106,40 @@ export interface ImagenGenerationConfig {
  *
  * @beta
  */
-export declare enum ImagenSafetyFilterLevel {
+export declare const ImagenSafetyFilterLevel: {
     /**
      * The most aggressive filtering level; most strict blocking.
      */
-    BLOCK_LOW_AND_ABOVE = "block_low_and_above",
+    readonly BLOCK_LOW_AND_ABOVE: "block_low_and_above";
     /**
      * Blocks some sensitive prompts and responses.
      */
-    BLOCK_MEDIUM_AND_ABOVE = "block_medium_and_above",
+    readonly BLOCK_MEDIUM_AND_ABOVE: "block_medium_and_above";
     /**
      * Blocks few sensitive prompts and responses.
      */
-    BLOCK_ONLY_HIGH = "block_only_high",
+    readonly BLOCK_ONLY_HIGH: "block_only_high";
     /**
      * The least aggressive filtering level; blocks very few sensitive prompts and responses.
      *
      * Access to this feature is restricted and may require your case to be reviewed and approved by
      * Cloud support.
      */
-    BLOCK_NONE = "block_none"
-}
+    readonly BLOCK_NONE: "block_none";
+};
+/**
+ * A filter level controlling how aggressively to filter sensitive content.
+ *
+ * Text prompts provided as inputs and images (generated or uploaded) through Imagen on Vertex AI
+ * are assessed against a list of safety filters, which include 'harmful categories' (for example,
+ * `violence`, `sexual`, `derogatory`, and `toxic`). This filter level controls how aggressively to
+ * filter out potentially harmful content from responses. See the {@link http://firebase.google.com/docs/vertex-ai/generate-images | documentation }
+ * and the {@link https://cloud.google.com/vertex-ai/generative-ai/docs/image/responsible-ai-imagen#safety-filters | Responsible AI and usage guidelines}
+ * for more details.
+ *
+ * @beta
+ */
+export type ImagenSafetyFilterLevel = (typeof ImagenSafetyFilterLevel)[keyof typeof ImagenSafetyFilterLevel];
 /**
  * A filter level controlling whether generation of images containing people or faces is allowed.
  *
@@ -135,11 +148,11 @@ export declare enum ImagenSafetyFilterLevel {
  *
  * @beta
  */
-export declare enum ImagenPersonFilterLevel {
+export declare const ImagenPersonFilterLevel: {
     /**
      * Disallow generation of images containing people or faces; images of people are filtered out.
      */
-    BLOCK_ALL = "dont_allow",
+    readonly BLOCK_ALL: "dont_allow";
     /**
      * Allow generation of images containing adults only; images of children are filtered out.
      *
@@ -147,7 +160,7 @@ export declare enum ImagenPersonFilterLevel {
      * reviewed and approved by Cloud support; see the {@link https://cloud.google.com/vertex-ai/generative-ai/docs/image/responsible-ai-imagen#person-face-gen | Responsible AI and usage guidelines}
      * for more details.
      */
-    ALLOW_ADULT = "allow_adult",
+    readonly ALLOW_ADULT: "allow_adult";
     /**
      * Allow generation of images containing adults only; images of children are filtered out.
      *
@@ -155,8 +168,17 @@ export declare enum ImagenPersonFilterLevel {
      * reviewed and approved by Cloud support; see the {@link https://cloud.google.com/vertex-ai/generative-ai/docs/image/responsible-ai-imagen#person-face-gen | Responsible AI and usage guidelines}
      * for more details.
      */
-    ALLOW_ALL = "allow_all"
-}
+    readonly ALLOW_ALL: "allow_all";
+};
+/**
+ * A filter level controlling whether generation of images containing people or faces is allowed.
+ *
+ * See the <a href="http://firebase.google.com/docs/vertex-ai/generate-images">personGeneration</a>
+ * documentation for more details.
+ *
+ * @beta
+ */
+export type ImagenPersonFilterLevel = (typeof ImagenPersonFilterLevel)[keyof typeof ImagenPersonFilterLevel];
 /**
  * Settings for controlling the aggressiveness of filtering out sensitive content.
  *
@@ -187,25 +209,37 @@ export interface ImagenSafetySettings {
  *
  * @beta
  */
-export declare enum ImagenAspectRatio {
+export declare const ImagenAspectRatio: {
     /**
      * Square (1:1) aspect ratio.
      */
-    SQUARE = "1:1",
+    readonly SQUARE: "1:1";
     /**
      * Landscape (3:4) aspect ratio.
      */
-    LANDSCAPE_3x4 = "3:4",
+    readonly LANDSCAPE_3x4: "3:4";
     /**
      * Portrait (4:3) aspect ratio.
      */
-    PORTRAIT_4x3 = "4:3",
+    readonly PORTRAIT_4x3: "4:3";
     /**
      * Landscape (16:9) aspect ratio.
      */
-    LANDSCAPE_16x9 = "16:9",
+    readonly LANDSCAPE_16x9: "16:9";
     /**
      * Portrait (9:16) aspect ratio.
      */
-    PORTRAIT_9x16 = "9:16"
-}
+    readonly PORTRAIT_9x16: "9:16";
+};
+/**
+ * Aspect ratios for Imagen images.
+ *
+ * To specify an aspect ratio for generated images, set the `aspectRatio` property in your
+ * {@link ImagenGenerationConfig}.
+ *
+ * See the the {@link http://firebase.google.com/docs/vertex-ai/generate-images | documentation }
+ * for more details and examples of the supported aspect ratios.
+ *
+ * @beta
+ */
+export type ImagenAspectRatio = (typeof ImagenAspectRatio)[keyof typeof ImagenAspectRatio];

package/dist/esm/src/types/imagen/responses.d.ts

@@ -67,8 +67,8 @@ export interface ImagenGenerationResponse<T extends ImagenInlineImage | ImagenGC
      * The reason that images were filtered out. This property will only be defined if one
      * or more images were filtered.
      *
-     * Images may be filtered out due to the {@link ImagenSafetyFilterLevel},
-     * {@link ImagenPersonFilterLevel}, or filtering included in the model.
+     * Images may be filtered out due to the {@link (ImagenSafetyFilterLevel:type)},
+     * {@link (ImagenPersonFilterLevel:type)}, or filtering included in the model.
      * The filter levels may be adjusted in your {@link ImagenSafetySettings}.
      *
      * See the {@link https://cloud.google.com/vertex-ai/generative-ai/docs/image/responsible-ai-imagen | Responsible AI and usage guidelines for Imagen}

package/dist/esm/src/types/requests.d.ts

@@ -14,10 +14,10 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-import { TypedSchema } from '../requests/schema-builder';
+import { ObjectSchema, TypedSchema } from '../requests/schema-builder';
 import { Content, Part } from './content';
 import { FunctionCallingMode, HarmBlockMethod, HarmBlockThreshold, HarmCategory, ResponseModality } from './enums';
-import { ObjectSchemaInterface, SchemaRequest } from './schema';
+import { ObjectSchemaRequest, SchemaRequest } from './schema';
 /**
  * Base parameters for a number of methods.
  * @public
@@ -87,7 +87,7 @@ export interface GenerationConfig {
      * value can be a class generated with a {@link Schema} static method
      * like `Schema.string()` or `Schema.object()` or it can be a plain
      * JS object matching the {@link SchemaRequest} interface.
-     * <br/>Note: This only applies when the specified `responseMIMEType` supports a schema; currently
+     * <br/>Note: This only applies when the specified `responseMimeType` supports a schema; currently
      * this is limited to `application/json` and `text/x.enum`.
      */
     responseSchema?: TypedSchema | SchemaRequest;
@@ -101,6 +101,10 @@ export interface GenerationConfig {
      * @beta
      */
     responseModalities?: ResponseModality[];
+    /**
+     * Configuration for "thinking" behavior of compatible Gemini models.
+     */
+    thinkingConfig?: ThinkingConfig;
 }
 /**
  * Params for {@link GenerativeModel.startChat}.
@@ -149,7 +153,7 @@ export interface RequestOptions {
  * Defines a tool that model can call to access external knowledge.
  * @public
  */
-export declare type Tool = FunctionDeclarationsTool;
+export type Tool = FunctionDeclarationsTool | GoogleSearchTool;
 /**
  * Structured representation of a function declaration as defined by the
  * {@link https://spec.openapis.org/oas/v3.0.3 | OpenAPI 3.0 specification}.
@@ -159,7 +163,7 @@ export declare type Tool = FunctionDeclarationsTool;
  * as a Tool by the model and executed by the client.
  * @public
  */
-export declare interface FunctionDeclaration {
+export interface FunctionDeclaration {
     /**
      * The name of the function to call. Must start with a letter or an
      * underscore. Must be a-z, A-Z, 0-9, or contain underscores and dashes, with
@@ -176,7 +180,41 @@ export declare interface FunctionDeclaration {
      * format. Reflects the Open API 3.03 Parameter Object. Parameter names are
      * case-sensitive. For a function with no parameters, this can be left unset.
      */
-    parameters?: ObjectSchemaInterface;
+    parameters?: ObjectSchema | ObjectSchemaRequest;
+}
+/**
+ * A tool that allows a Gemini model to connect to Google Search to access and incorporate
+ * up-to-date information from the web into its responses.
+ *
+ * Important: If using Grounding with Google Search, you are required to comply with the
+ * "Grounding with Google Search" usage requirements for your chosen API provider: {@link https://ai.google.dev/gemini-api/terms#grounding-with-google-search | Gemini Developer API}
+ * or Vertex AI Gemini API (see {@link https://cloud.google.com/terms/service-terms | Service Terms}
+ * section within the Service Specific Terms).
+ *
+ * @public
+ */
+export interface GoogleSearchTool {
+    /**
+     * Specifies the Google Search configuration.
+     * Currently, this is an empty object, but it's reserved for future configuration options.
+     * Specifies the Google Search configuration. Currently, this is an empty object, but it's
+     * reserved for future configuration options.
+     *
+     * When using this feature, you are required to comply with the "Grounding with Google Search"
+     * usage requirements for your chosen API provider: {@link https://ai.google.dev/gemini-api/terms#grounding-with-google-search | Gemini Developer API}
+     * or Vertex AI Gemini API (see {@link https://cloud.google.com/terms/service-terms | Service Terms}
+     * section within the Service Specific Terms).
+     */
+    googleSearch: GoogleSearch;
+}
+/**
+ * Specifies the Google Search configuration.
+ *
+ * @remarks Currently, this is an empty object, but it's reserved for future configuration options.
+ *
+ * @public
+ */
+export interface GoogleSearch {
 }
 /**
  * A `FunctionDeclarationsTool` is a piece of code that enables the system to
@@ -184,7 +222,7 @@ export declare interface FunctionDeclaration {
  * outside of knowledge and scope of the model.
  * @public
  */
-export declare interface FunctionDeclarationsTool {
+export interface FunctionDeclarationsTool {
     /**
      * Optional. One or more function declarations
      * to be passed to the model along with the current user query. Model may
@@ -211,3 +249,27 @@ export interface FunctionCallingConfig {
     mode?: FunctionCallingMode;
     allowedFunctionNames?: string[];
 }
+/**
+ * Configuration for "thinking" behavior of compatible Gemini models.
+ *
+ * Certain models utilize a thinking process before generating a response. This allows them to
+ * reason through complex problems and plan a more coherent and accurate answer.
+ *
+ * @public
+ */
+export interface ThinkingConfig {
+    /**
+     * The thinking budget, in tokens.
+     *
+     * This parameter sets an upper limit on the number of tokens the model can use for its internal
+     * "thinking" process. A higher budget may result in higher quality responses for complex tasks
+     * but can also increase latency and cost.
+     *
+     * If you don't specify a budget, the model will determine the appropriate amount
+     * of thinking based on the complexity of the prompt.
+     *
+     * An error will be thrown if you set a thinking budget for a model that does not support this
+     * feature or if the specified budget is not within the model's supported range.
+     */
+    thinkingBudget?: number;
+}