langchain 0.0.147 → 0.0.148

package/dist/chat_models/{googlevertexai.cjs → googlevertexai/common.cjs}

@@ -1,9 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ChatGoogleVertexAI = exports.GoogleVertexAIChatMessage = void 0;
-const base_js_1 = require("./base.cjs");
-const index_js_1 = require("../schema/index.cjs");
-const googlevertexai_connection_js_1 = require("../util/googlevertexai-connection.cjs");
+exports.BaseChatGoogleVertexAI = exports.GoogleVertexAIChatMessage = void 0;
+const base_js_1 = require("../base.cjs");
+const index_js_1 = require("../../schema/index.cjs");
 /**
  * Represents a chat message in the Google Vertex AI chat model.
  */
@@ -85,22 +84,15 @@ class GoogleVertexAIChatMessage {
 }
 exports.GoogleVertexAIChatMessage = GoogleVertexAIChatMessage;
 /**
- * Enables calls to the Google Cloud's Vertex AI API to access
- * Large Language Models in a chat-like fashion.
- *
- * To use, you will need to have one of the following authentication
- * methods in place:
- * - You are logged into an account permitted to the Google Cloud project
- *   using Vertex AI.
- * - You are running this on a machine using a service account permitted to
- *   the Google Cloud project using Vertex AI.
- * - The `GOOGLE_APPLICATION_CREDENTIALS` environment variable is set to the
- *   path of a credentials file for a service account permitted to the
- *   Google Cloud project using Vertex AI.
+ * Base class for Google Vertex AI chat models.
+ * Implemented subclasses must provide a GoogleVertexAILLMConnection
+ * with appropriate auth client.
  */
-class ChatGoogleVertexAI extends base_js_1.BaseChatModel {
-    static lc_name() {
-        return "ChatGoogleVertexAI";
+class BaseChatGoogleVertexAI extends base_js_1.BaseChatModel {
+    get lc_aliases() {
+        return {
+            model: "model_name",
+        };
     }
     constructor(fields) {
         super(fields ?? {});
@@ -158,10 +150,6 @@ class ChatGoogleVertexAI extends base_js_1.BaseChatModel {
         this.topP = fields?.topP ?? this.topP;
         this.topK = fields?.topK ?? this.topK;
         this.examples = fields?.examples ?? this.examples;
-        this.connection = new googlevertexai_connection_js_1.GoogleVertexAILLMConnection({
-            ...fields,
-            ...this,
-        }, this.caller);
     }
     _combineLLMOutput() {
         // TODO: Combine the safetyAttributes
@@ -177,13 +165,13 @@ class ChatGoogleVertexAI extends base_js_1.BaseChatModel {
             maxOutputTokens: this.maxOutputTokens,
         };
         const result = await this.connection.request([instance], parameters, options);
-        const generations = result?.data?.predictions?.map((prediction) => ChatGoogleVertexAI.convertPrediction(prediction)) ?? [];
+        const generations = result?.data?.predictions?.map((prediction) => BaseChatGoogleVertexAI.convertPrediction(prediction)) ?? [];
         return {
             generations,
         };
     }
     _llmType() {
-        return "googlevertexai";
+        return "vertexai";
     }
     /**
      * Creates an instance of the Google Vertex AI chat model.
@@ -238,4 +226,4 @@ class ChatGoogleVertexAI extends base_js_1.BaseChatModel {
         };
     }
 }
-exports.ChatGoogleVertexAI = ChatGoogleVertexAI;
+exports.BaseChatGoogleVertexAI = BaseChatGoogleVertexAI;

package/dist/chat_models/{googlevertexai.d.ts → googlevertexai/common.d.ts}

@@ -1,8 +1,8 @@
-import { BaseChatModel } from "./base.js";
-import { BaseMessage, ChatGeneration, ChatMessage, ChatResult, LLMResult } from "../schema/index.js";
-import { GoogleVertexAILLMConnection } from "../util/googlevertexai-connection.js";
-import { GoogleVertexAIBaseLLMInput, GoogleVertexAIBasePrediction } from "../types/googlevertexai-types.js";
-import { BaseLanguageModelCallOptions } from "../base_language/index.js";
+import { BaseChatModel } from "../base.js";
+import { BaseMessage, ChatGeneration, ChatMessage, ChatResult, LLMResult } from "../../schema/index.js";
+import { GoogleVertexAILLMConnection } from "../../util/googlevertexai-connection.js";
+import { GoogleVertexAIBaseLLMInput, GoogleVertexAIBasePrediction } from "../../types/googlevertexai-types.js";
+import { BaseLanguageModelCallOptions } from "../../base_language/index.js";
 /**
  * Represents a single "example" exchange that can be provided to
  * help illustrate what a model response should look like.
@@ -76,28 +76,18 @@ export interface GoogleVertexAIChatPrediction extends GoogleVertexAIBasePredicti
 /**
  * Defines the input to the Google Vertex AI chat model.
  */
-export interface GoogleVertexAIChatInput extends GoogleVertexAIBaseLLMInput {
+export interface GoogleVertexAIChatInput<AuthOptions> extends GoogleVertexAIBaseLLMInput<AuthOptions> {
     /** Instructions how the model should respond */
     context?: string;
     /** Help the model understand what an appropriate response is */
     examples?: ChatExample[];
 }
 /**
- * Enables calls to the Google Cloud's Vertex AI API to access
- * Large Language Models in a chat-like fashion.
- *
- * To use, you will need to have one of the following authentication
- * methods in place:
- * - You are logged into an account permitted to the Google Cloud project
- *   using Vertex AI.
- * - You are running this on a machine using a service account permitted to
- *   the Google Cloud project using Vertex AI.
- * - The `GOOGLE_APPLICATION_CREDENTIALS` environment variable is set to the
- *   path of a credentials file for a service account permitted to the
- *   Google Cloud project using Vertex AI.
+ * Base class for Google Vertex AI chat models.
+ * Implemented subclasses must provide a GoogleVertexAILLMConnection
+ * with appropriate auth client.
  */
-export declare class ChatGoogleVertexAI extends BaseChatModel implements GoogleVertexAIChatInput {
-    static lc_name(): string;
+export declare class BaseChatGoogleVertexAI<AuthOptions> extends BaseChatModel implements GoogleVertexAIChatInput<AuthOptions> {
     lc_serializable: boolean;
     model: string;
     temperature: number;
@@ -105,8 +95,9 @@ export declare class ChatGoogleVertexAI extends BaseChatModel implements GoogleV
     topP: number;
     topK: number;
     examples: ChatExample[];
-    connection: GoogleVertexAILLMConnection<BaseLanguageModelCallOptions, GoogleVertexAIChatInstance, GoogleVertexAIChatPrediction>;
-    constructor(fields?: GoogleVertexAIChatInput);
+    connection: GoogleVertexAILLMConnection<BaseLanguageModelCallOptions, GoogleVertexAIChatInstance, GoogleVertexAIChatPrediction, AuthOptions>;
+    get lc_aliases(): Record<string, string>;
+    constructor(fields?: GoogleVertexAIChatInput<AuthOptions>);
     _combineLLMOutput(): LLMResult["llmOutput"];
     _generate(messages: BaseMessage[], options: this["ParsedCallOptions"]): Promise<ChatResult>;
     _llmType(): string;

package/dist/chat_models/{googlevertexai.js → googlevertexai/common.js}

@@ -1,6 +1,5 @@
-import { BaseChatModel } from "./base.js";
-import { AIMessage, ChatMessage, } from "../schema/index.js";
-import { GoogleVertexAILLMConnection } from "../util/googlevertexai-connection.js";
+import { BaseChatModel } from "../base.js";
+import { AIMessage, ChatMessage, } from "../../schema/index.js";
 /**
  * Represents a chat message in the Google Vertex AI chat model.
  */
@@ -81,22 +80,15 @@ export class GoogleVertexAIChatMessage {
     }
 }
 /**
- * Enables calls to the Google Cloud's Vertex AI API to access
- * Large Language Models in a chat-like fashion.
- *
- * To use, you will need to have one of the following authentication
- * methods in place:
- * - You are logged into an account permitted to the Google Cloud project
- *   using Vertex AI.
- * - You are running this on a machine using a service account permitted to
- *   the Google Cloud project using Vertex AI.
- * - The `GOOGLE_APPLICATION_CREDENTIALS` environment variable is set to the
- *   path of a credentials file for a service account permitted to the
- *   Google Cloud project using Vertex AI.
+ * Base class for Google Vertex AI chat models.
+ * Implemented subclasses must provide a GoogleVertexAILLMConnection
+ * with appropriate auth client.
  */
-export class ChatGoogleVertexAI extends BaseChatModel {
-    static lc_name() {
-        return "ChatGoogleVertexAI";
+export class BaseChatGoogleVertexAI extends BaseChatModel {
+    get lc_aliases() {
+        return {
+            model: "model_name",
+        };
     }
     constructor(fields) {
         super(fields ?? {});
@@ -154,10 +146,6 @@ export class ChatGoogleVertexAI extends BaseChatModel {
         this.topP = fields?.topP ?? this.topP;
         this.topK = fields?.topK ?? this.topK;
         this.examples = fields?.examples ?? this.examples;
-        this.connection = new GoogleVertexAILLMConnection({
-            ...fields,
-            ...this,
-        }, this.caller);
     }
     _combineLLMOutput() {
         // TODO: Combine the safetyAttributes
@@ -173,13 +161,13 @@ export class ChatGoogleVertexAI extends BaseChatModel {
             maxOutputTokens: this.maxOutputTokens,
         };
         const result = await this.connection.request([instance], parameters, options);
-        const generations = result?.data?.predictions?.map((prediction) => ChatGoogleVertexAI.convertPrediction(prediction)) ?? [];
+        const generations = result?.data?.predictions?.map((prediction) => BaseChatGoogleVertexAI.convertPrediction(prediction)) ?? [];
         return {
             generations,
         };
     }
     _llmType() {
-        return "googlevertexai";
+        return "vertexai";
     }
     /**
      * Creates an instance of the Google Vertex AI chat model.

package/dist/chat_models/googlevertexai/index.cjs

@@ -0,0 +1,36 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GoogleVertexAIChatMessage = exports.ChatGoogleVertexAI = void 0;
+const google_auth_library_1 = require("google-auth-library");
+const common_js_1 = require("./common.cjs");
+const googlevertexai_connection_js_1 = require("../../util/googlevertexai-connection.cjs");
+/**
+ * Enables calls to the Google Cloud's Vertex AI API to access
+ * Large Language Models in a chat-like fashion.
+ *
+ * To use, you will need to have one of the following authentication
+ * methods in place:
+ * - You are logged into an account permitted to the Google Cloud project
+ *   using Vertex AI.
+ * - You are running this on a machine using a service account permitted to
+ *   the Google Cloud project using Vertex AI.
+ * - The `GOOGLE_APPLICATION_CREDENTIALS` environment variable is set to the
+ *   path of a credentials file for a service account permitted to the
+ *   Google Cloud project using Vertex AI.
+ */
+class ChatGoogleVertexAI extends common_js_1.BaseChatGoogleVertexAI {
+    static lc_name() {
+        return "ChatVertexAI";
+    }
+    constructor(fields) {
+        super(fields);
+        const client = new google_auth_library_1.GoogleAuth({
+            scopes: "https://www.googleapis.com/auth/cloud-platform",
+            ...fields?.authOptions,
+        });
+        this.connection = new googlevertexai_connection_js_1.GoogleVertexAILLMConnection({ ...fields, ...this }, this.caller, client);
+    }
+}
+exports.ChatGoogleVertexAI = ChatGoogleVertexAI;
+var common_js_2 = require("./common.cjs");
+Object.defineProperty(exports, "GoogleVertexAIChatMessage", { enumerable: true, get: function () { return common_js_2.GoogleVertexAIChatMessage; } });

package/dist/chat_models/googlevertexai/index.d.ts

@@ -0,0 +1,21 @@
+import { GoogleAuthOptions } from "google-auth-library";
+import { BaseChatGoogleVertexAI, GoogleVertexAIChatInput } from "./common.js";
+/**
+ * Enables calls to the Google Cloud's Vertex AI API to access
+ * Large Language Models in a chat-like fashion.
+ *
+ * To use, you will need to have one of the following authentication
+ * methods in place:
+ * - You are logged into an account permitted to the Google Cloud project
+ *   using Vertex AI.
+ * - You are running this on a machine using a service account permitted to
+ *   the Google Cloud project using Vertex AI.
+ * - The `GOOGLE_APPLICATION_CREDENTIALS` environment variable is set to the
+ *   path of a credentials file for a service account permitted to the
+ *   Google Cloud project using Vertex AI.
+ */
+export declare class ChatGoogleVertexAI extends BaseChatGoogleVertexAI<GoogleAuthOptions> {
+    static lc_name(): string;
+    constructor(fields?: GoogleVertexAIChatInput<GoogleAuthOptions>);
+}
+export { ChatExample, GoogleVertexAIChatAuthor, GoogleVertexAIChatInput, GoogleVertexAIChatInstance, GoogleVertexAIChatMessage, GoogleVertexAIChatMessageFields, GoogleVertexAIChatPrediction, } from "./common.js";

package/dist/chat_models/googlevertexai/index.js

@@ -0,0 +1,31 @@
+import { GoogleAuth } from "google-auth-library";
+import { BaseChatGoogleVertexAI } from "./common.js";
+import { GoogleVertexAILLMConnection } from "../../util/googlevertexai-connection.js";
+/**
+ * Enables calls to the Google Cloud's Vertex AI API to access
+ * Large Language Models in a chat-like fashion.
+ *
+ * To use, you will need to have one of the following authentication
+ * methods in place:
+ * - You are logged into an account permitted to the Google Cloud project
+ *   using Vertex AI.
+ * - You are running this on a machine using a service account permitted to
+ *   the Google Cloud project using Vertex AI.
+ * - The `GOOGLE_APPLICATION_CREDENTIALS` environment variable is set to the
+ *   path of a credentials file for a service account permitted to the
+ *   Google Cloud project using Vertex AI.
+ */
+export class ChatGoogleVertexAI extends BaseChatGoogleVertexAI {
+    static lc_name() {
+        return "ChatVertexAI";
+    }
+    constructor(fields) {
+        super(fields);
+        const client = new GoogleAuth({
+            scopes: "https://www.googleapis.com/auth/cloud-platform",
+            ...fields?.authOptions,
+        });
+        this.connection = new GoogleVertexAILLMConnection({ ...fields, ...this }, this.caller, client);
+    }
+}
+export { GoogleVertexAIChatMessage, } from "./common.js";

package/dist/chat_models/googlevertexai/web.cjs

@@ -0,0 +1,33 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GoogleVertexAIChatMessage = exports.ChatGoogleVertexAI = void 0;
+const googlevertexai_connection_js_1 = require("../../util/googlevertexai-connection.cjs");
+const googlevertexai_webauth_js_1 = require("../../util/googlevertexai-webauth.cjs");
+const common_js_1 = require("./common.cjs");
+/**
+ * Enables calls to the Google Cloud's Vertex AI API to access
+ * Large Language Models in a chat-like fashion.
+ *
+ * This entrypoint and class are intended to be used in web environments like Edge
+ * functions where you do not have access to the file system. It supports passing
+ * service account credentials directly as a "GOOGLE_VERTEX_AI_WEB_CREDENTIALS"
+ * environment variable or directly as "authOptions.credentials".
+ */
+class ChatGoogleVertexAI extends common_js_1.BaseChatGoogleVertexAI {
+    static lc_name() {
+        return "ChatVertexAI";
+    }
+    get lc_secrets() {
+        return {
+            "authOptions.credentials": "GOOGLE_VERTEX_AI_WEB_CREDENTIALS",
+        };
+    }
+    constructor(fields) {
+        super(fields);
+        const client = new googlevertexai_webauth_js_1.WebGoogleAuth(fields?.authOptions);
+        this.connection = new googlevertexai_connection_js_1.GoogleVertexAILLMConnection({ ...fields, ...this }, this.caller, client);
+    }
+}
+exports.ChatGoogleVertexAI = ChatGoogleVertexAI;
+var common_js_2 = require("./common.cjs");
+Object.defineProperty(exports, "GoogleVertexAIChatMessage", { enumerable: true, get: function () { return common_js_2.GoogleVertexAIChatMessage; } });

package/dist/chat_models/googlevertexai/web.d.ts

@@ -0,0 +1,19 @@
+import { WebGoogleAuthOptions } from "../../util/googlevertexai-webauth.js";
+import { BaseChatGoogleVertexAI, GoogleVertexAIChatInput } from "./common.js";
+/**
+ * Enables calls to the Google Cloud's Vertex AI API to access
+ * Large Language Models in a chat-like fashion.
+ *
+ * This entrypoint and class are intended to be used in web environments like Edge
+ * functions where you do not have access to the file system. It supports passing
+ * service account credentials directly as a "GOOGLE_VERTEX_AI_WEB_CREDENTIALS"
+ * environment variable or directly as "authOptions.credentials".
+ */
+export declare class ChatGoogleVertexAI extends BaseChatGoogleVertexAI<WebGoogleAuthOptions> {
+    static lc_name(): string;
+    get lc_secrets(): {
+        [key: string]: string;
+    };
+    constructor(fields?: GoogleVertexAIChatInput<WebGoogleAuthOptions>);
+}
+export { ChatExample, GoogleVertexAIChatAuthor, GoogleVertexAIChatInput, GoogleVertexAIChatInstance, GoogleVertexAIChatMessage, GoogleVertexAIChatMessageFields, GoogleVertexAIChatPrediction, } from "./common.js";

package/dist/chat_models/googlevertexai/web.js

@@ -0,0 +1,28 @@
+import { GoogleVertexAILLMConnection } from "../../util/googlevertexai-connection.js";
+import { WebGoogleAuth, } from "../../util/googlevertexai-webauth.js";
+import { BaseChatGoogleVertexAI } from "./common.js";
+/**
+ * Enables calls to the Google Cloud's Vertex AI API to access
+ * Large Language Models in a chat-like fashion.
+ *
+ * This entrypoint and class are intended to be used in web environments like Edge
+ * functions where you do not have access to the file system. It supports passing
+ * service account credentials directly as a "GOOGLE_VERTEX_AI_WEB_CREDENTIALS"
+ * environment variable or directly as "authOptions.credentials".
+ */
+export class ChatGoogleVertexAI extends BaseChatGoogleVertexAI {
+    static lc_name() {
+        return "ChatVertexAI";
+    }
+    get lc_secrets() {
+        return {
+            "authOptions.credentials": "GOOGLE_VERTEX_AI_WEB_CREDENTIALS",
+        };
+    }
+    constructor(fields) {
+        super(fields);
+        const client = new WebGoogleAuth(fields?.authOptions);
+        this.connection = new GoogleVertexAILLMConnection({ ...fields, ...this }, this.caller, client);
+    }
+}
+export { GoogleVertexAIChatMessage, } from "./common.js";

package/dist/document_loaders/web/notionapi.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.NotionAPILoader = void 0;
+exports.NotionAPILoader = exports.isDatabase = exports.isPage = exports.isErrorResponse = exports.isDatabaseResponse = exports.isPageResponse = void 0;
 const client_1 = require("@notionhq/client");
 const notion_to_md_1 = require("notion-to-md");
 const notion_js_1 = require("notion-to-md/build/utils/notion.js");
@@ -8,18 +8,15 @@ const document_js_1 = require("../../document.cjs");
 const base_js_1 = require("../base.cjs");
 const async_caller_js_1 = require("../../util/async_caller.cjs");
 const isPageResponse = (res) => !(0, client_1.isNotionClientError)(res) && res.object === "page";
+exports.isPageResponse = isPageResponse;
 const isDatabaseResponse = (res) => !(0, client_1.isNotionClientError)(res) && res.object === "database";
+exports.isDatabaseResponse = isDatabaseResponse;
 const isErrorResponse = (res) => (0, client_1.isNotionClientError)(res);
-const isPage = (res) => isPageResponse(res) && (0, client_1.isFullPage)(res);
-const isDatabase = (res) => isDatabaseResponse(res) && (0, client_1.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;
-};
+exports.isErrorResponse = isErrorResponse;
+const isPage = (res) => (0, exports.isPageResponse)(res) && (0, client_1.isFullPage)(res);
+exports.isPage = isPage;
+const isDatabase = (res) => (0, exports.isDatabaseResponse)(res) && (0, client_1.isFullDatabase)(res);
+exports.isDatabase = isDatabase;
 /**
  * A class that extends the BaseDocumentLoader class. It represents a
  * document loader for loading documents from Notion using the Notion API.
@@ -107,11 +104,85 @@ class NotionAPILoader extends base_js_1.BaseDocumentLoader {
         this.rootTitle = "";
         this.onDocumentLoaded = options.onDocumentLoaded ?? ((_ti, _cu) => { });
     }
+    /**
+     * Adds a selection of page ids to the pageQueue and removes duplicates.
+     * @param items An array of string ids
+     */
     addToQueue(...items) {
         const deDuped = items.filter((item) => !this.pageCompleted.concat(this.pageQueue).includes(item));
         this.pageQueue.push(...deDuped);
         this.pageQueueTotal += deDuped.length;
     }
+    /**
+     * Parses a Notion GetResponse object (page or database) and returns a string of the title.
+     * @param obj The Notion GetResponse object to parse.
+     * @returns The string of the title.
+     */
+    getTitle(obj) {
+        if ((0, exports.isPage)(obj) && obj.properties.title.type === "title") {
+            return obj.properties.title.title[0]?.plain_text;
+        }
+        if ((0, exports.isDatabase)(obj))
+            return obj.title[0]?.plain_text;
+        return null;
+    }
+    /**
+     * Parses the property type and returns a string
+     * @param page The Notion page property to parse.
+     * @returns A string of parsed property.
+     */
+    getPropValue(prop) {
+        switch (prop.type) {
+            case "number": {
+                const propNumber = prop[prop.type];
+                return propNumber !== null ? propNumber.toString() : "";
+            }
+            case "url":
+                return prop[prop.type] || "";
+            case "select":
+                return prop[prop.type]?.name ?? "";
+            case "multi_select":
+                return `[${prop[prop.type].map((v) => `"${v.name}"`).join(", ")}]`;
+            case "status":
+                return prop[prop.type]?.name ?? "";
+            case "date":
+                return `${prop[prop.type]?.start ?? ""}${prop[prop.type]?.end ? ` - ${prop[prop.type]?.end}` : ""}`;
+            case "email":
+                return prop[prop.type] || "";
+            case "phone_number":
+                return prop[prop.type] || "";
+            case "checkbox":
+                return prop[prop.type].toString();
+            case "files":
+                return `[${prop[prop.type].map((v) => `"${v.name}"`).join(", ")}]`;
+            case "created_by":
+                return `["${prop[prop.type].object}", "${prop[prop.type].id}"]`;
+            case "created_time":
+                return prop[prop.type];
+            case "last_edited_by":
+                return `["${prop[prop.type].object}", "${prop[prop.type].id}"]`;
+            case "last_edited_time":
+                return prop[prop.type];
+            case "title":
+                return prop[prop.type]
+                    .map((v) => this.n2mClient.annotatePlainText(v.plain_text, v.annotations))
+                    .join("");
+            case "rich_text":
+                return prop[prop.type]
+                    .map((v) => this.n2mClient.annotatePlainText(v.plain_text, v.annotations))
+                    .join("");
+            case "people":
+                return `[${prop[prop.type]
+                    .map((v) => `["${v.object}", "${v.id}"]`)
+                    .join(", ")}]`;
+            case "unique_id":
+                return `${prop[prop.type].prefix || ""}${prop[prop.type].number}`;
+            case "relation":
+                return `[${prop[prop.type].map((v) => `"${v.id}"`).join(", ")}]`;
+            default:
+                return `Unsupported type: ${prop.type}`;
+        }
+    }
     /**
      * Parses the properties of a Notion page and returns them as key-value
      * pairs.
@@ -119,60 +190,11 @@ class NotionAPILoader extends base_js_1.BaseDocumentLoader {
      * @returns An object containing the parsed properties as key-value pairs.
      */
     parsePageProperties(page) {
-        return Object.fromEntries(Object.entries(page.properties).map(([_, prop]) => {
-            switch (prop.type) {
-                case "number":
-                    return [prop.type, prop[prop.type]];
-                case "url":
-                    return [prop.type, prop[prop.type]];
-                case "select":
-                    return [prop.type, prop[prop.type]?.name ?? ""];
-                case "multi_select":
-                    return [
-                        prop.type,
-                        prop[prop.type].map((select) => select.name).join(", "),
-                    ];
-                case "status":
-                    return [prop.type, prop[prop.type]?.name ?? ""];
-                case "date":
-                    return [
-                        prop.type,
-                        `${prop[prop.type]?.start ?? ""}${prop[prop.type]?.end ? `-  ${prop[prop.type]?.end}` : ""}`,
-                    ];
-                case "email":
-                    return [prop.type, prop[prop.type]];
-                case "phone_number":
-                    return [prop.type, prop[prop.type]];
-                case "checkbox":
-                    return [prop.type, prop[prop.type].toString()];
-                // case "files":
-                case "created_by":
-                    return [prop.type, prop[prop.type]];
-                case "created_time":
-                    return [prop.type, prop[prop.type]];
-                case "last_edited_by":
-                    return [prop.type, prop[prop.type]];
-                case "last_edited_time":
-                    return [prop.type, prop[prop.type]];
-                // case "formula":
-                case "title":
-                    return [
-                        prop.type,
-                        prop[prop.type].map((v) => v.plain_text).join(""),
-                    ];
-                case "rich_text":
-                    return [
-                        prop.type,
-                        prop[prop.type].map((v) => v.plain_text).join(""),
-                    ];
-                case "people":
-                    return [prop.type, prop[prop.type]];
-                // case "relation":
-                // case "rollup":
-                default:
-                    return [prop.type, "Unsupported type"];
-            }
-        }));
+        return Object.entries(page.properties).reduce((accum, [propName, prop]) => {
+            const value = this.getPropValue(prop);
+            const props = { ...accum, [propName]: value };
+            return prop.type === "title" ? { ...props, _title: value } : props;
+        }, {});
     }
     /**
      * Parses the details of a Notion page and returns them as an object.
@@ -180,10 +202,10 @@ class NotionAPILoader extends base_js_1.BaseDocumentLoader {
      * @returns An object containing the parsed details of the page.
      */
     parsePageDetails(page) {
-        const metadata = Object.fromEntries(Object.entries(page).filter(([key, _]) => key !== "id"));
+        const { id, ...rest } = page;
         return {
-            ...metadata,
-            notionId: page.id,
+            ...rest,
+            notionId: id,
             properties: this.parsePageProperties(page),
         };
     }
@@ -306,7 +328,7 @@ class NotionAPILoader extends base_js_1.BaseDocumentLoader {
             resDatabasePromise,
         ]);
         // Check if both resPage and resDatabase resulted in error responses
-        const errors = [resPage, resDatabase].filter(isErrorResponse);
+        const errors = [resPage, resDatabase].filter(exports.isErrorResponse);
         if (errors.length === 2) {
             if (errors.every((e) => e.code === client_1.APIErrorCode.ObjectNotFound)) {
                 throw new AggregateError([
@@ -316,7 +338,8 @@ class NotionAPILoader extends base_js_1.BaseDocumentLoader {
             }
             throw new AggregateError(errors);
         }
-        this.rootTitle = getTitle(resPage) || getTitle(resDatabase) || this.id;
+        this.rootTitle =
+            this.getTitle(resPage) || this.getTitle(resDatabase) || this.id;
         let pageId = this.pageQueue.shift();
         while (pageId) {
             await this.loadPage(pageId);

package/dist/document_loaders/web/notionapi.d.ts

@@ -1,7 +1,22 @@
-import { Client } from "@notionhq/client";
+import { APIResponseError, Client, isFullBlock, isFullPage, isFullDatabase } from "@notionhq/client";
 import { Document } from "../../document.js";
 import { BaseDocumentLoader } from "../base.js";
 import { AsyncCaller } from "../../util/async_caller.js";
+type GuardType<T> = T extends (x: any, ...rest: any) => x is infer U ? U : never;
+export type GetBlockResponse = Parameters<typeof isFullBlock>[0];
+export type GetPageResponse = Parameters<typeof isFullPage>[0];
+export type GetDatabaseResponse = Parameters<typeof isFullDatabase>[0];
+export type BlockObjectResponse = GuardType<typeof isFullBlock>;
+export type PageObjectResponse = GuardType<typeof isFullPage>;
+export type DatabaseObjectResponse = GuardType<typeof isFullDatabase>;
+export type GetResponse = GetBlockResponse | GetPageResponse | GetDatabaseResponse | APIResponseError;
+export type PagePropertiesType = PageObjectResponse["properties"];
+export type PagePropertiesValue = PagePropertiesType[keyof PagePropertiesType];
+export declare const isPageResponse: (res: GetResponse) => res is import("@notionhq/client/build/src/api-endpoints.js").PageObjectResponse | import("@notionhq/client/build/src/api-endpoints.js").PartialPageObjectResponse;
+export declare const isDatabaseResponse: (res: GetResponse) => res is import("@notionhq/client/build/src/api-endpoints.js").DatabaseObjectResponse | import("@notionhq/client/build/src/api-endpoints.js").PartialDatabaseObjectResponse;
+export declare const isErrorResponse: (res: GetResponse) => res is APIResponseError;
+export declare const isPage: (res: GetResponse) => res is import("@notionhq/client/build/src/api-endpoints.js").PageObjectResponse;
+export declare const isDatabase: (res: GetResponse) => res is import("@notionhq/client/build/src/api-endpoints.js").DatabaseObjectResponse;
 /**
  * Represents the type of Notion API to load documents from. The options
  * are "database" or "page".
@@ -31,7 +46,23 @@ export declare class NotionAPILoader extends BaseDocumentLoader {
     private rootTitle;
     private onDocumentLoaded;
     constructor(options: NotionAPILoaderOptions);
+    /**
+     * Adds a selection of page ids to the pageQueue and removes duplicates.
+     * @param items An array of string ids
+     */
     private addToQueue;
+    /**
+     * Parses a Notion GetResponse object (page or database) and returns a string of the title.
+     * @param obj The Notion GetResponse object to parse.
+     * @returns The string of the title.
+     */
+    private getTitle;
+    /**
+     * Parses the property type and returns a string
+     * @param page The Notion page property to parse.
+     * @returns A string of parsed property.
+     */
+    private getPropValue;
     /**
      * Parses the properties of a Notion page and returns them as key-value
      * pairs.
@@ -73,3 +104,4 @@ export declare class NotionAPILoader extends BaseDocumentLoader {
      */
     load(): Promise<Document[]>;
 }
+export {};