langchain 0.0.132 → 0.0.133

package/dist/document_loaders/web/notionapi.js

@@ -1,8 +1,22 @@
-import { Client, isFullBlock, isFullPage, iteratePaginatedAPI, } from "@notionhq/client";
+import { Client, isFullBlock, isFullPage, iteratePaginatedAPI, APIErrorCode, isNotionClientError, isFullDatabase, } from "@notionhq/client";
 import { NotionToMarkdown } from "notion-to-md";
 import { getBlockChildren } from "notion-to-md/build/utils/notion.js";
-import { BaseDocumentLoader } from "../base.js";
 import { Document } from "../../document.js";
+import { BaseDocumentLoader } from "../base.js";
+import { AsyncCaller } from "../../util/async_caller.js";
+const isPageResponse = (res) => !isNotionClientError(res) && res.object === "page";
+const isDatabaseResponse = (res) => !isNotionClientError(res) && res.object === "database";
+const isErrorResponse = (res) => isNotionClientError(res);
+const isPage = (res) => isPageResponse(res) && isFullPage(res);
+const isDatabase = (res) => isDatabaseResponse(res) && isFullDatabase(res);
+const getTitle = (obj) => {
+    if (isPage(obj) && obj.properties.title.type === "title") {
+        return obj.properties.title.title[0]?.plain_text;
+    }
+    if (isDatabase(obj))
+        return obj.title[0]?.plain_text;
+    return null;
+};
 /**
  * A class that extends the BaseDocumentLoader class. It represents a
  * document loader for loading documents from Notion using the Notion API.
@@ -10,6 +24,12 @@ import { Document } from "../../document.js";
 export class NotionAPILoader extends BaseDocumentLoader {
     constructor(options) {
         super();
+        Object.defineProperty(this, "caller", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "notionClient", {
             enumerable: true,
             configurable: true,
@@ -28,19 +48,66 @@ export class NotionAPILoader extends BaseDocumentLoader {
             writable: true,
             value: void 0
         });
-        Object.defineProperty(this, "type", {
+        Object.defineProperty(this, "pageQueue", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "pageCompleted", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "pageQueueTotal", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "documents", {
             enumerable: true,
             configurable: true,
             writable: true,
             value: void 0
         });
-        this.notionClient = new Client(options.clientOptions);
+        Object.defineProperty(this, "rootTitle", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "onDocumentLoaded", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.caller = new AsyncCaller({
+            maxConcurrency: 64,
+            ...options.callerOptions,
+        });
+        this.notionClient = new Client({
+            logger: () => { },
+            ...options.clientOptions,
+        });
         this.n2mClient = new NotionToMarkdown({
             notionClient: this.notionClient,
             config: { parseChildPages: false, convertImagesToBase64: false },
         });
         this.id = options.id;
-        this.type = options.type;
+        this.pageQueue = [];
+        this.pageCompleted = [];
+        this.pageQueueTotal = 0;
+        this.documents = [];
+        this.rootTitle = "";
+        this.onDocumentLoaded = options.onDocumentLoaded ?? ((_ti, _cu) => { });
+    }
+    addToQueue(...items) {
+        const deDuped = items.filter((item) => !this.pageCompleted.concat(this.pageQueue).includes(item));
+        this.pageQueue.push(...deDuped);
+        this.pageQueueTotal += deDuped.length;
     }
     /**
      * Parses the properties of a Notion page and returns them as key-value
@@ -123,123 +190,135 @@ export class NotionAPILoader extends BaseDocumentLoader {
      * @returns A Promise that resolves to an MdBlock object.
      */
     async loadBlock(block) {
-        return {
+        const mdBlock = {
             type: block.type,
             blockId: block.id,
-            parent: await this.n2mClient.blockToMarkdown(block),
+            parent: await this.caller.call(() => this.n2mClient.blockToMarkdown(block)),
             children: [],
         };
+        if (block.has_children) {
+            const block_id = block.type === "synced_block" &&
+                block.synced_block?.synced_from?.block_id
+                ? block.synced_block.synced_from.block_id
+                : block.id;
+            const childBlocks = await this.loadBlocks(await this.caller.call(() => getBlockChildren(this.notionClient, block_id, null)));
+            mdBlock.children = childBlocks;
+        }
+        return mdBlock;
     }
     /**
-     * Loads Notion blocks and their child documents recursively.
+     * Loads Notion blocks and their children recursively.
      * @param blocksResponse The response from the Notion API containing the blocks to load.
-     * @returns A Promise that resolves to an object containing the loaded MdBlocks and child Documents.
+     * @returns A Promise that resolves to an array containing the loaded MdBlocks.
      */
-    async loadBlocksAndDocs(blocksResponse) {
+    async loadBlocks(blocksResponse) {
         const blocks = blocksResponse.filter(isFullBlock);
-        const [childPageDocuments, childDatabaseDocuments, blocksDocsArray] = await Promise.all([
-            Promise.all(blocks
-                .filter((block) => block.type.includes("child_page"))
-                .map((block) => this.loadPage(block.id))),
-            Promise.all(blocks
-                .filter((block) => block.type.includes("child_database"))
-                .map((block) => this.loadDatabase(block.id))),
-            Promise.all(blocks
-                .filter((block) => !["child_page", "child_database"].includes(block.type))
-                .map(async (block) => {
-                const mdBlock = await this.loadBlock(block);
-                let childDocuments = [];
-                if (block.has_children) {
-                    const block_id = block.type === "synced_block" &&
-                        block.synced_block?.synced_from?.block_id
-                        ? block.synced_block.synced_from.block_id
-                        : block.id;
-                    const childBlocksDocs = await this.loadBlocksAndDocs(await getBlockChildren(this.notionClient, block_id, null));
-                    mdBlock.children = childBlocksDocs.mdBlocks;
-                    childDocuments = childBlocksDocs.childDocuments;
-                }
-                return {
-                    mdBlocks: [mdBlock],
-                    childDocuments,
-                };
-            })),
+        // Add child pages to queue
+        const childPages = blocks
+            .filter((block) => block.type.includes("child_page"))
+            .map((block) => block.id);
+        if (childPages.length > 0)
+            this.addToQueue(...childPages);
+        // Add child database pages to queue
+        const childDatabases = blocks
+            .filter((block) => block.type.includes("child_database"))
+            .map((block) => this.caller.call(() => this.loadDatabase(block.id)));
+        // Load this block and child blocks
+        const loadingMdBlocks = blocks
+            .filter((block) => !["child_page", "child_database"].includes(block.type))
+            .map((block) => this.loadBlock(block));
+        const [mdBlocks] = await Promise.all([
+            Promise.all(loadingMdBlocks),
+            Promise.all(childDatabases),
         ]);
-        const allMdBlocks = blocksDocsArray
-            .flat()
-            .map((blockDoc) => blockDoc.mdBlocks);
-        const childDocuments = blocksDocsArray
-            .flat()
-            .map((blockDoc) => blockDoc.childDocuments);
-        return {
-            mdBlocks: [...allMdBlocks.flat()],
-            childDocuments: [
-                ...childPageDocuments.flat(),
-                ...childDatabaseDocuments.flat(),
-                ...childDocuments.flat(),
-            ],
-        };
+        return mdBlocks;
     }
     /**
-     * Loads a Notion page and its child documents.
+     * Loads a Notion page and its child documents, then adds it to the completed documents array.
      * @param page The Notion page or page ID to load.
-     * @returns A Promise that resolves to an array of Documents.
      */
     async loadPage(page) {
-        // Check page is a page ID or a GetPageResponse
+        // Check page is a page ID or a PageObjectResponse
         const [pageData, pageId] = typeof page === "string"
-            ? [this.notionClient.pages.retrieve({ page_id: page }), page]
+            ? [
+                this.caller.call(() => this.notionClient.pages.retrieve({ page_id: page })),
+                page,
+            ]
             : [page, page.id];
         const [pageDetails, pageBlocks] = await Promise.all([
             pageData,
-            getBlockChildren(this.notionClient, pageId, null),
+            this.caller.call(() => getBlockChildren(this.notionClient, pageId, null)),
         ]);
         if (!isFullPage(pageDetails))
-            return [];
-        const { mdBlocks, childDocuments } = await this.loadBlocksAndDocs(pageBlocks);
+            return;
+        const mdBlocks = await this.loadBlocks(pageBlocks);
         const mdStringObject = this.n2mClient.toMarkdownString(mdBlocks);
         const pageDocument = new Document({
             pageContent: mdStringObject.parent,
             metadata: this.parsePageDetails(pageDetails),
         });
-        return [pageDocument, ...childDocuments];
+        this.documents.push(pageDocument);
+        this.pageCompleted.push(pageId);
+        this.onDocumentLoaded(this.documents.length, this.pageQueueTotal, pageDocument.metadata.properties.title, this.rootTitle);
     }
     /**
-     * Loads a Notion database and its documents.
+     * Loads a Notion database and adds it's pages to the queue.
      * @param id The ID of the Notion database to load.
-     * @returns A Promise that resolves to an array of Documents.
      */
     async loadDatabase(id) {
-        const documents = [];
         try {
             for await (const page of iteratePaginatedAPI(this.notionClient.databases.query, {
                 database_id: id,
+                page_size: 50,
             })) {
-                if (!isFullPage(page))
-                    continue;
-                documents.push(...(await this.loadPage(page)));
+                this.addToQueue(page.id);
             }
         }
         catch (e) {
             console.log(e);
             // TODO: Catch and report api request errors
         }
-        return documents;
     }
     /**
      * Loads the documents from Notion based on the specified options.
      * @returns A Promise that resolves to an array of Documents.
      */
     async load() {
-        const documents = [];
-        switch (this.type) {
-            case "page":
-                documents.push(...(await this.loadPage(this.id)));
-                break;
-            case "database":
-                documents.push(...(await this.loadDatabase(this.id)));
-                break;
-            default:
+        const resPagePromise = this.notionClient.pages
+            .retrieve({ page_id: this.id })
+            .then((res) => {
+            this.addToQueue(this.id);
+            return res;
+        })
+            .catch((error) => error);
+        const resDatabasePromise = this.notionClient.databases
+            .retrieve({ database_id: this.id })
+            .then(async (res) => {
+            await this.loadDatabase(this.id);
+            return res;
+        })
+            .catch((error) => error);
+        const [resPage, resDatabase] = await Promise.all([
+            resPagePromise,
+            resDatabasePromise,
+        ]);
+        // Check if both resPage and resDatabase resulted in error responses
+        const errors = [resPage, resDatabase].filter(isErrorResponse);
+        if (errors.length === 2) {
+            if (errors.every((e) => e.code === APIErrorCode.ObjectNotFound)) {
+                throw new AggregateError([
+                    Error(`Could not find object with ID: ${this.id}. Make sure the relevant pages and databases are shared with your integration.`),
+                    ...errors,
+                ]);
+            }
+            throw new AggregateError(errors);
+        }
+        this.rootTitle = getTitle(resPage) || getTitle(resDatabase) || this.id;
+        let pageId = this.pageQueue.shift();
+        while (pageId) {
+            await this.loadPage(pageId);
+            pageId = this.pageQueue.shift();
         }
-        return documents;
+        return this.documents;
     }
 }

package/dist/embeddings/googlevertexai.cjs

@@ -34,7 +34,7 @@ class GoogleVertexAIEmbeddings extends base_js_1.Embeddings {
             value: void 0
         });
         this.model = fields?.model ?? this.model;
-        this.connection = new googlevertexai_connection_js_1.GoogleVertexAIConnection({ ...fields, ...this }, this.caller);
+        this.connection = new googlevertexai_connection_js_1.GoogleVertexAILLMConnection({ ...fields, ...this }, this.caller);
     }
     /**
      * Takes an array of documents as input and returns a promise that

package/dist/embeddings/googlevertexai.d.ts

@@ -1,11 +1,11 @@
 import { Embeddings, EmbeddingsParams } from "./base.js";
-import { GoogleVertexAIConnectionParams } from "../types/googlevertexai-types.js";
+import { GoogleVertexAIBaseLLMInput } from "../types/googlevertexai-types.js";
 /**
  * Defines the parameters required to initialize a
  * GoogleVertexAIEmbeddings instance. It extends EmbeddingsParams and
  * GoogleVertexAIConnectionParams.
  */
-export interface GoogleVertexAIEmbeddingsParams extends EmbeddingsParams, GoogleVertexAIConnectionParams {
+export interface GoogleVertexAIEmbeddingsParams extends EmbeddingsParams, GoogleVertexAIBaseLLMInput {
 }
 /**
  * Enables calls to the Google Cloud's Vertex AI API to access

package/dist/embeddings/googlevertexai.js

@@ -1,5 +1,5 @@
 import { Embeddings } from "./base.js";
-import { GoogleVertexAIConnection } from "../util/googlevertexai-connection.js";
+import { GoogleVertexAILLMConnection } from "../util/googlevertexai-connection.js";
 import { chunkArray } from "../util/chunk.js";
 /**
  * Enables calls to the Google Cloud's Vertex AI API to access
@@ -31,7 +31,7 @@ export class GoogleVertexAIEmbeddings extends Embeddings {
             value: void 0
         });
         this.model = fields?.model ?? this.model;
-        this.connection = new GoogleVertexAIConnection({ ...fields, ...this }, this.caller);
+        this.connection = new GoogleVertexAILLMConnection({ ...fields, ...this }, this.caller);
     }
     /**
      * Takes an array of documents as input and returns a promise that

package/dist/experimental/multimodal_embeddings/googlevertexai.cjs

@@ -24,7 +24,7 @@ class GoogleVertexAIMultimodalEmbeddings extends base_js_1.Embeddings {
             value: void 0
         });
         this.model = fields?.model ?? this.model;
-        this.connection = new googlevertexai_connection_js_1.GoogleVertexAIConnection({ ...fields, ...this }, this.caller);
+        this.connection = new googlevertexai_connection_js_1.GoogleVertexAILLMConnection({ ...fields, ...this }, this.caller);
     }
     /**
      * Converts media (text or image) to an instance that can be used for

package/dist/experimental/multimodal_embeddings/googlevertexai.d.ts

@@ -1,11 +1,11 @@
 /// <reference types="node" resolution-mode="require"/>
 import { Embeddings, EmbeddingsParams } from "../../embeddings/base.js";
-import { GoogleVertexAIBasePrediction, GoogleVertexAIConnectionParams, GoogleVertexAILLMResponse } from "../../types/googlevertexai-types.js";
+import { GoogleVertexAIBaseLLMInput, GoogleVertexAIBasePrediction, GoogleVertexAILLMResponse } from "../../types/googlevertexai-types.js";
 /**
  * Parameters for the GoogleVertexAIMultimodalEmbeddings class, extending
  * both EmbeddingsParams and GoogleVertexAIConnectionParams.
  */
-export interface GoogleVertexAIMultimodalEmbeddingsParams extends EmbeddingsParams, GoogleVertexAIConnectionParams {
+export interface GoogleVertexAIMultimodalEmbeddingsParams extends EmbeddingsParams, GoogleVertexAIBaseLLMInput {
 }
 /**
  * An instance of media (text or image) that can be used for generating

package/dist/experimental/multimodal_embeddings/googlevertexai.js

@@ -1,5 +1,5 @@
 import { Embeddings } from "../../embeddings/base.js";
-import { GoogleVertexAIConnection } from "../../util/googlevertexai-connection.js";
+import { GoogleVertexAILLMConnection } from "../../util/googlevertexai-connection.js";
 /**
  * Class for generating embeddings for text and images using Google's
  * Vertex AI. It extends the Embeddings base class and implements the
@@ -21,7 +21,7 @@ export class GoogleVertexAIMultimodalEmbeddings extends Embeddings {
             value: void 0
         });
         this.model = fields?.model ?? this.model;
-        this.connection = new GoogleVertexAIConnection({ ...fields, ...this }, this.caller);
+        this.connection = new GoogleVertexAILLMConnection({ ...fields, ...this }, this.caller);
     }
     /**
      * Converts media (text or image) to an instance that can be used for

package/dist/llms/googlevertexai.cjs

@@ -68,7 +68,7 @@ class GoogleVertexAI extends base_js_1.BaseLLM {
         this.maxOutputTokens = fields?.maxOutputTokens ?? this.maxOutputTokens;
         this.topP = fields?.topP ?? this.topP;
         this.topK = fields?.topK ?? this.topK;
-        this.connection = new googlevertexai_connection_js_1.GoogleVertexAIConnection({ ...fields, ...this }, this.caller);
+        this.connection = new googlevertexai_connection_js_1.GoogleVertexAILLMConnection({ ...fields, ...this }, this.caller);
     }
     _llmType() {
         return "googlevertexai";

package/dist/llms/googlevertexai.js

@@ -1,5 +1,5 @@
 import { BaseLLM } from "./base.js";
-import { GoogleVertexAIConnection } from "../util/googlevertexai-connection.js";
+import { GoogleVertexAILLMConnection } from "../util/googlevertexai-connection.js";
 /**
  * Enables calls to the Google Cloud's Vertex AI API to access
  * Large Language Models.
@@ -65,7 +65,7 @@ export class GoogleVertexAI extends BaseLLM {
         this.maxOutputTokens = fields?.maxOutputTokens ?? this.maxOutputTokens;
         this.topP = fields?.topP ?? this.topP;
         this.topK = fields?.topK ?? this.topK;
-        this.connection = new GoogleVertexAIConnection({ ...fields, ...this }, this.caller);
+        this.connection = new GoogleVertexAILLMConnection({ ...fields, ...this }, this.caller);
     }
     _llmType() {
         return "googlevertexai";

package/dist/load/import_constants.cjs

@@ -33,6 +33,7 @@ exports.optionalImportEntrypoints = [
     "langchain/vectorstores/analyticdb",
     "langchain/vectorstores/elasticsearch",
     "langchain/vectorstores/chroma",
+    "langchain/vectorstores/googlevertexai",
     "langchain/vectorstores/hnswlib",
     "langchain/vectorstores/faiss",
     "langchain/vectorstores/weaviate",

package/dist/load/import_constants.js

@@ -30,6 +30,7 @@ export const optionalImportEntrypoints = [
     "langchain/vectorstores/analyticdb",
     "langchain/vectorstores/elasticsearch",
     "langchain/vectorstores/chroma",
+    "langchain/vectorstores/googlevertexai",
     "langchain/vectorstores/hnswlib",
     "langchain/vectorstores/faiss",
     "langchain/vectorstores/weaviate",

package/dist/schema/output_parser.cjs

@@ -96,7 +96,7 @@ class StringOutputParser extends BaseTransformOutputParser {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: ["schema", "output_parser"]
+            value: ["langchain", "schema", "output_parser"]
         });
         Object.defineProperty(this, "lc_serializable", {
             enumerable: true,
@@ -135,7 +135,7 @@ class BytesOutputParser extends BaseTransformOutputParser {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: ["schema", "output_parser"]
+            value: ["langchain", "schema", "output_parser"]
         });
         Object.defineProperty(this, "lc_serializable", {
             enumerable: true,

package/dist/schema/output_parser.js

@@ -90,7 +90,7 @@ export class StringOutputParser extends BaseTransformOutputParser {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: ["schema", "output_parser"]
+            value: ["langchain", "schema", "output_parser"]
         });
         Object.defineProperty(this, "lc_serializable", {
             enumerable: true,
@@ -128,7 +128,7 @@ export class BytesOutputParser extends BaseTransformOutputParser {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: ["schema", "output_parser"]
+            value: ["langchain", "schema", "output_parser"]
         });
         Object.defineProperty(this, "lc_serializable", {
             enumerable: true,

package/dist/tools/base.cjs

@@ -1,9 +1,27 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.Tool = exports.StructuredTool = void 0;
+exports.Tool = exports.StructuredTool = exports.ToolInputParsingException = void 0;
 const zod_1 = require("zod");
 const manager_js_1 = require("../callbacks/manager.cjs");
 const index_js_1 = require("../base_language/index.cjs");
+/**
+ * Custom error class used to handle exceptions related to tool input parsing.
+ * It extends the built-in `Error` class and adds an optional `output`
+ * property that can hold the output that caused the exception.
+ */
+class ToolInputParsingException extends Error {
+    constructor(message, output) {
+        super(message);
+        Object.defineProperty(this, "output", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.output = output;
+    }
+}
+exports.ToolInputParsingException = ToolInputParsingException;
 /**
  * Base class for Tools that accept input of any shape defined by a Zod schema.
  */
@@ -41,7 +59,13 @@ class StructuredTool extends index_js_1.BaseLangChain {
     async call(arg, configArg, 
     /** @deprecated */
     tags) {
-        const parsed = await this.schema.parseAsync(arg);
+        let parsed;
+        try {
+            parsed = await this.schema.parseAsync(arg);
+        }
+        catch (e) {
+            throw new ToolInputParsingException(`Received tool input did not match expected schema`, JSON.stringify(arg));
+        }
         const config = (0, manager_js_1.parseCallbackConfigArg)(configArg);
         const callbackManager_ = await manager_js_1.CallbackManager.configure(config.callbacks, this.callbacks, config.tags || tags, this.tags, config.metadata, this.metadata, { verbose: this.verbose });
         const runManager = await callbackManager_?.handleToolStart(this.toJSON(), typeof parsed === "string" ? parsed : JSON.stringify(parsed));

package/dist/tools/base.d.ts

@@ -7,6 +7,15 @@ import { RunnableConfig } from "../schema/runnable.js";
  */
 export interface ToolParams extends BaseLangChainParams {
 }
+/**
+ * Custom error class used to handle exceptions related to tool input parsing.
+ * It extends the built-in `Error` class and adds an optional `output`
+ * property that can hold the output that caused the exception.
+ */
+export declare class ToolInputParsingException extends Error {
+    output?: string;
+    constructor(message: string, output?: string);
+}
 /**
  * Base class for Tools that accept input of any shape defined by a Zod schema.
  */

package/dist/tools/base.js

@@ -1,6 +1,23 @@
 import { z } from "zod";
 import { CallbackManager, parseCallbackConfigArg, } from "../callbacks/manager.js";
 import { BaseLangChain } from "../base_language/index.js";
+/**
+ * Custom error class used to handle exceptions related to tool input parsing.
+ * It extends the built-in `Error` class and adds an optional `output`
+ * property that can hold the output that caused the exception.
+ */
+export class ToolInputParsingException extends Error {
+    constructor(message, output) {
+        super(message);
+        Object.defineProperty(this, "output", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.output = output;
+    }
+}
 /**
  * Base class for Tools that accept input of any shape defined by a Zod schema.
  */
@@ -38,7 +55,13 @@ export class StructuredTool extends BaseLangChain {
     async call(arg, configArg, 
     /** @deprecated */
     tags) {
-        const parsed = await this.schema.parseAsync(arg);
+        let parsed;
+        try {
+            parsed = await this.schema.parseAsync(arg);
+        }
+        catch (e) {
+            throw new ToolInputParsingException(`Received tool input did not match expected schema`, JSON.stringify(arg));
+        }
         const config = parseCallbackConfigArg(configArg);
         const callbackManager_ = await CallbackManager.configure(config.callbacks, this.callbacks, config.tags || tags, this.tags, config.metadata, this.metadata, { verbose: this.verbose });
         const runManager = await callbackManager_?.handleToolStart(this.toJSON(), typeof parsed === "string" ? parsed : JSON.stringify(parsed));

package/dist/types/googlevertexai-types.d.ts

@@ -6,10 +6,12 @@ export interface GoogleVertexAIConnectionParams {
     authOptions?: GoogleAuthOptions;
     /** Region where the LLM is stored */
     location?: string;
-    /** Model to use */
-    model?: string;
+    /** The version of the API functions. Part of the path. */
+    apiVersion?: string;
 }
 export interface GoogleVertexAIModelParams {
+    /** Model to use */
+    model?: string;
     /** Sampling temperature to use */
     temperature?: number;
     /**
@@ -39,7 +41,10 @@ export interface GoogleVertexAIModelParams {
 }
 export interface GoogleVertexAIBaseLLMInput extends BaseLLMParams, GoogleVertexAIConnectionParams, GoogleVertexAIModelParams {
 }
-export interface GoogleVertexAIBasePrediction {
+export interface GoogleVertexAIResponse {
+    data: any;
+}
+export interface GoogleVertexAIBasePrediction extends GoogleVertexAIResponse {
     safetyAttributes?: any;
 }
 export interface GoogleVertexAILLMResponse<PredictionType extends GoogleVertexAIBasePrediction> {