langchain 0.0.150 → 0.0.152

package/cache/cloudflare_kv.cjs

@@ -0,0 +1 @@
+module.exports = require('../dist/cache/cloudflare_kv.cjs');

package/cache/cloudflare_kv.d.ts

@@ -0,0 +1 @@
+export * from '../dist/cache/cloudflare_kv.js'

package/cache/cloudflare_kv.js

@@ -0,0 +1 @@
+export * from '../dist/cache/cloudflare_kv.js'

package/dist/agents/chat/index.cjs

@@ -91,7 +91,7 @@ class ChatAgent extends agent_js_1.Agent {
             chat_js_1.SystemMessagePromptTemplate.fromTemplate(template),
             chat_js_1.HumanMessagePromptTemplate.fromTemplate(humanMessageTemplate),
         ];
-        return chat_js_1.ChatPromptTemplate.fromPromptMessages(messages);
+        return chat_js_1.ChatPromptTemplate.fromMessages(messages);
     }
     /**
      * Creates a ChatAgent instance using a language model, tools, and

package/dist/agents/chat/index.js

@@ -88,7 +88,7 @@ export class ChatAgent extends Agent {
             SystemMessagePromptTemplate.fromTemplate(template),
             HumanMessagePromptTemplate.fromTemplate(humanMessageTemplate),
         ];
-        return ChatPromptTemplate.fromPromptMessages(messages);
+        return ChatPromptTemplate.fromMessages(messages);
     }
     /**
      * Creates a ChatAgent instance using a language model, tools, and

package/dist/agents/chat_convo/index.cjs

@@ -110,7 +110,7 @@ class ChatConversationalAgent extends agent_js_1.Agent {
             chat_js_1.HumanMessagePromptTemplate.fromTemplate(renderedHumanMessage),
             new chat_js_1.MessagesPlaceholder("agent_scratchpad"),
         ];
-        return chat_js_1.ChatPromptTemplate.fromPromptMessages(messages);
+        return chat_js_1.ChatPromptTemplate.fromMessages(messages);
     }
     /**
      * Creates an instance of the ChatConversationalAgent class from a

package/dist/agents/chat_convo/index.js

@@ -107,7 +107,7 @@ export class ChatConversationalAgent extends Agent {
             HumanMessagePromptTemplate.fromTemplate(renderedHumanMessage),
             new MessagesPlaceholder("agent_scratchpad"),
         ];
-        return ChatPromptTemplate.fromPromptMessages(messages);
+        return ChatPromptTemplate.fromMessages(messages);
     }
     /**
      * Creates an instance of the ChatConversationalAgent class from a

package/dist/agents/openai/index.cjs

@@ -102,7 +102,7 @@ class OpenAIAgent extends agent_js_1.Agent {
      */
     static createPrompt(_tools, fields) {
         const { prefix = prompt_js_1.PREFIX } = fields || {};
-        return chat_js_1.ChatPromptTemplate.fromPromptMessages([
+        return chat_js_1.ChatPromptTemplate.fromMessages([
             chat_js_1.SystemMessagePromptTemplate.fromTemplate(prefix),
             new chat_js_1.MessagesPlaceholder("chat_history"),
             chat_js_1.HumanMessagePromptTemplate.fromTemplate("{input}"),

package/dist/agents/openai/index.js

@@ -98,7 +98,7 @@ export class OpenAIAgent extends Agent {
      */
     static createPrompt(_tools, fields) {
         const { prefix = PREFIX } = fields || {};
-        return ChatPromptTemplate.fromPromptMessages([
+        return ChatPromptTemplate.fromMessages([
             SystemMessagePromptTemplate.fromTemplate(prefix),
             new MessagesPlaceholder("chat_history"),
             HumanMessagePromptTemplate.fromTemplate("{input}"),

package/dist/agents/structured_chat/index.cjs

@@ -120,7 +120,7 @@ class StructuredChatAgent extends agent_js_1.Agent {
                 inputVariables,
             })),
         ];
-        return chat_js_1.ChatPromptTemplate.fromPromptMessages(messages);
+        return chat_js_1.ChatPromptTemplate.fromMessages(messages);
     }
     /**
      * Creates a StructuredChatAgent from an LLM and a list of tools.

package/dist/agents/structured_chat/index.js

@@ -117,7 +117,7 @@ export class StructuredChatAgent extends Agent {
                 inputVariables,
             })),
         ];
-        return ChatPromptTemplate.fromPromptMessages(messages);
+        return ChatPromptTemplate.fromMessages(messages);
     }
     /**
      * Creates a StructuredChatAgent from an LLM and a list of tools.

package/dist/agents/xml/index.cjs

@@ -65,7 +65,7 @@ class XMLAgent extends agent_js_1.BaseSingleActionAgent {
         return ["input"];
     }
     static createPrompt() {
-        return chat_js_1.ChatPromptTemplate.fromPromptMessages([
+        return chat_js_1.ChatPromptTemplate.fromMessages([
             chat_js_1.HumanMessagePromptTemplate.fromTemplate(prompt_js_1.AGENT_INSTRUCTIONS),
             chat_js_1.AIMessagePromptTemplate.fromTemplate("{intermediate_steps}"),
         ]);

package/dist/agents/xml/index.js

@@ -61,7 +61,7 @@ export class XMLAgent extends BaseSingleActionAgent {
         return ["input"];
     }
     static createPrompt() {
-        return ChatPromptTemplate.fromPromptMessages([
+        return ChatPromptTemplate.fromMessages([
             HumanMessagePromptTemplate.fromTemplate(AGENT_INSTRUCTIONS),
             AIMessagePromptTemplate.fromTemplate("{intermediate_steps}"),
         ]);

package/dist/base_language/count_tokens.cjs

@@ -63,6 +63,7 @@ const calculateMaxTokens = async ({ prompt, modelName, }) => {
     catch (error) {
         console.warn("Failed to calculate number of tokens, falling back to approximate count");
         // fallback to approximate calculation if tiktoken is not available
+        // each token is ~4 characters: https://help.openai.com/en/articles/4936856-what-are-tokens-and-how-to-count-them#
         numTokens = Math.ceil(prompt.length / 4);
     }
     const maxTokens = (0, exports.getModelContextSize)(modelName);

package/dist/base_language/count_tokens.js

@@ -57,6 +57,7 @@ export const calculateMaxTokens = async ({ prompt, modelName, }) => {
     catch (error) {
         console.warn("Failed to calculate number of tokens, falling back to approximate count");
         // fallback to approximate calculation if tiktoken is not available
+        // each token is ~4 characters: https://help.openai.com/en/articles/4936856-what-are-tokens-and-how-to-count-them#
         numTokens = Math.ceil(prompt.length / 4);
     }
     const maxTokens = getModelContextSize(modelName);

package/dist/base_language/index.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.calculateMaxTokens = exports.BaseLanguageModel = exports.BaseLangChain = void 0;
+exports.getModelContextSize = exports.calculateMaxTokens = exports.BaseLanguageModel = exports.BaseLangChain = void 0;
 const index_js_1 = require("../schema/index.cjs");
 const async_caller_js_1 = require("../util/async_caller.cjs");
 const count_tokens_js_1 = require("./count_tokens.cjs");
@@ -155,8 +155,10 @@ class BaseLanguageModel extends BaseLangChain {
 }
 exports.BaseLanguageModel = BaseLanguageModel;
 /*
- * Calculate max tokens for given model and prompt.
- * That is the model size - number of tokens in prompt.
+ * Export utility functions for token calculations:
+ * - calculateMaxTokens: Calculate max tokens for a given model and prompt (the model context size - tokens in prompt).
+ * - getModelContextSize: Get the context size for a specific model.
  */
 var count_tokens_js_2 = require("./count_tokens.cjs");
 Object.defineProperty(exports, "calculateMaxTokens", { enumerable: true, get: function () { return count_tokens_js_2.calculateMaxTokens; } });
+Object.defineProperty(exports, "getModelContextSize", { enumerable: true, get: function () { return count_tokens_js_2.getModelContextSize; } });

package/dist/base_language/index.d.ts

@@ -101,4 +101,4 @@ export declare abstract class BaseLanguageModel<RunOutput = any, CallOptions ext
      */
     static deserialize(data: SerializedLLM): Promise<BaseLanguageModel>;
 }
-export { calculateMaxTokens } from "./count_tokens.js";
+export { calculateMaxTokens, getModelContextSize } from "./count_tokens.js";

package/dist/base_language/index.js

@@ -150,7 +150,8 @@ export class BaseLanguageModel extends BaseLangChain {
     }
 }
 /*
- * Calculate max tokens for given model and prompt.
- * That is the model size - number of tokens in prompt.
+ * Export utility functions for token calculations:
+ * - calculateMaxTokens: Calculate max tokens for a given model and prompt (the model context size - tokens in prompt).
+ * - getModelContextSize: Get the context size for a specific model.
  */
-export { calculateMaxTokens } from "./count_tokens.js";
+export { calculateMaxTokens, getModelContextSize } from "./count_tokens.js";

package/dist/cache/cloudflare_kv.cjs

@@ -0,0 +1,61 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CloudflareKVCache = void 0;
+const index_js_1 = require("../schema/index.cjs");
+const base_js_1 = require("./base.cjs");
+/**
+ * Represents a specific implementation of a caching mechanism using Cloudflare KV
+ * as the underlying storage system. It extends the `BaseCache` class and
+ * overrides its methods to provide the Cloudflare KV-specific logic.
+ */
+class CloudflareKVCache extends index_js_1.BaseCache {
+    constructor(binding) {
+        super();
+        Object.defineProperty(this, "binding", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.binding = binding;
+    }
+    /**
+     * Retrieves data from the cache. It constructs a cache key from the given
+     * `prompt` and `llmKey`, and retrieves the corresponding value from the
+     * Cloudflare KV namespace.
+     * @param prompt The prompt used to construct the cache key.
+     * @param llmKey The LLM key used to construct the cache key.
+     * @returns An array of Generations if found, null otherwise.
+     */
+    async lookup(prompt, llmKey) {
+        let idx = 0;
+        let key = (0, base_js_1.getCacheKey)(prompt, llmKey, String(idx));
+        let value = await this.binding.get(key);
+        const generations = [];
+        while (value) {
+            if (!value) {
+                break;
+            }
+            generations.push({ text: value });
+            idx += 1;
+            key = (0, base_js_1.getCacheKey)(prompt, llmKey, String(idx));
+            value = await this.binding.get(key);
+        }
+        return generations.length > 0 ? generations : null;
+    }
+    /**
+     * Updates the cache with new data. It constructs a cache key from the
+     * given `prompt` and `llmKey`, and stores the `value` in the Cloudflare KV
+     * namespace.
+     * @param prompt The prompt used to construct the cache key.
+     * @param llmKey The LLM key used to construct the cache key.
+     * @param value The value to be stored in the cache.
+     */
+    async update(prompt, llmKey, value) {
+        for (let i = 0; i < value.length; i += 1) {
+            const key = (0, base_js_1.getCacheKey)(prompt, llmKey, String(i));
+            await this.binding.put(key, value[i].text);
+        }
+    }
+}
+exports.CloudflareKVCache = CloudflareKVCache;

package/dist/cache/cloudflare_kv.d.ts

@@ -0,0 +1,29 @@
+import type { KVNamespace } from "@cloudflare/workers-types";
+import { BaseCache, Generation } from "../schema/index.js";
+/**
+ * Represents a specific implementation of a caching mechanism using Cloudflare KV
+ * as the underlying storage system. It extends the `BaseCache` class and
+ * overrides its methods to provide the Cloudflare KV-specific logic.
+ */
+export declare class CloudflareKVCache extends BaseCache {
+    private binding;
+    constructor(binding: KVNamespace);
+    /**
+     * Retrieves data from the cache. It constructs a cache key from the given
+     * `prompt` and `llmKey`, and retrieves the corresponding value from the
+     * Cloudflare KV namespace.
+     * @param prompt The prompt used to construct the cache key.
+     * @param llmKey The LLM key used to construct the cache key.
+     * @returns An array of Generations if found, null otherwise.
+     */
+    lookup(prompt: string, llmKey: string): Promise<Generation[] | null>;
+    /**
+     * Updates the cache with new data. It constructs a cache key from the
+     * given `prompt` and `llmKey`, and stores the `value` in the Cloudflare KV
+     * namespace.
+     * @param prompt The prompt used to construct the cache key.
+     * @param llmKey The LLM key used to construct the cache key.
+     * @param value The value to be stored in the cache.
+     */
+    update(prompt: string, llmKey: string, value: Generation[]): Promise<void>;
+}

package/dist/cache/cloudflare_kv.js

@@ -0,0 +1,57 @@
+import { BaseCache } from "../schema/index.js";
+import { getCacheKey } from "./base.js";
+/**
+ * Represents a specific implementation of a caching mechanism using Cloudflare KV
+ * as the underlying storage system. It extends the `BaseCache` class and
+ * overrides its methods to provide the Cloudflare KV-specific logic.
+ */
+export class CloudflareKVCache extends BaseCache {
+    constructor(binding) {
+        super();
+        Object.defineProperty(this, "binding", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.binding = binding;
+    }
+    /**
+     * Retrieves data from the cache. It constructs a cache key from the given
+     * `prompt` and `llmKey`, and retrieves the corresponding value from the
+     * Cloudflare KV namespace.
+     * @param prompt The prompt used to construct the cache key.
+     * @param llmKey The LLM key used to construct the cache key.
+     * @returns An array of Generations if found, null otherwise.
+     */
+    async lookup(prompt, llmKey) {
+        let idx = 0;
+        let key = getCacheKey(prompt, llmKey, String(idx));
+        let value = await this.binding.get(key);
+        const generations = [];
+        while (value) {
+            if (!value) {
+                break;
+            }
+            generations.push({ text: value });
+            idx += 1;
+            key = getCacheKey(prompt, llmKey, String(idx));
+            value = await this.binding.get(key);
+        }
+        return generations.length > 0 ? generations : null;
+    }
+    /**
+     * Updates the cache with new data. It constructs a cache key from the
+     * given `prompt` and `llmKey`, and stores the `value` in the Cloudflare KV
+     * namespace.
+     * @param prompt The prompt used to construct the cache key.
+     * @param llmKey The LLM key used to construct the cache key.
+     * @param value The value to be stored in the cache.
+     */
+    async update(prompt, llmKey, value) {
+        for (let i = 0; i < value.length; i += 1) {
+            const key = getCacheKey(prompt, llmKey, String(i));
+            await this.binding.put(key, value[i].text);
+        }
+    }
+}

package/dist/chains/openai_functions/openapi.cjs

@@ -375,7 +375,7 @@ async function createOpenAPIChain(spec, options = {}) {
     if (defaultExecutionMethod === undefined) {
         throw new Error(`Could not parse any valid operations from the provided spec.`);
     }
-    const { llm = new openai_js_1.ChatOpenAI({ modelName: "gpt-3.5-turbo-0613" }), prompt = chat_js_1.ChatPromptTemplate.fromPromptMessages([
+    const { llm = new openai_js_1.ChatOpenAI({ modelName: "gpt-3.5-turbo-0613" }), prompt = chat_js_1.ChatPromptTemplate.fromMessages([
         chat_js_1.HumanMessagePromptTemplate.fromTemplate("Use the provided API's to respond to this user query:\n\n{query}"),
     ]), requestChain = new SimpleRequestChain({
         requestMethod: async (name, args) => defaultExecutionMethod(name, args, {

package/dist/chains/openai_functions/openapi.js

@@ -371,7 +371,7 @@ export async function createOpenAPIChain(spec, options = {}) {
     if (defaultExecutionMethod === undefined) {
         throw new Error(`Could not parse any valid operations from the provided spec.`);
     }
-    const { llm = new ChatOpenAI({ modelName: "gpt-3.5-turbo-0613" }), prompt = ChatPromptTemplate.fromPromptMessages([
+    const { llm = new ChatOpenAI({ modelName: "gpt-3.5-turbo-0613" }), prompt = ChatPromptTemplate.fromMessages([
         HumanMessagePromptTemplate.fromTemplate("Use the provided API's to respond to this user query:\n\n{query}"),
     ]), requestChain = new SimpleRequestChain({
         requestMethod: async (name, args) => defaultExecutionMethod(name, args, {

package/dist/chains/question_answering/map_reduce_prompts.cjs

@@ -21,8 +21,7 @@ const messages = [
     /*#__PURE__*/ chat_js_1.SystemMessagePromptTemplate.fromTemplate(system_template),
     /*#__PURE__*/ chat_js_1.HumanMessagePromptTemplate.fromTemplate("{question}"),
 ];
-const CHAT_QA_PROMPT = 
-/*#__PURE__*/ chat_js_1.ChatPromptTemplate.fromPromptMessages(messages);
+const CHAT_QA_PROMPT = /*#__PURE__*/ chat_js_1.ChatPromptTemplate.fromMessages(messages);
 exports.COMBINE_QA_PROMPT_SELECTOR = 
 /*#__PURE__*/ new conditional_js_1.ConditionalPromptSelector(exports.DEFAULT_COMBINE_QA_PROMPT, [
     [conditional_js_1.isChatModel, CHAT_QA_PROMPT],
@@ -68,7 +67,7 @@ const combine_messages = [
     /*#__PURE__*/ chat_js_1.HumanMessagePromptTemplate.fromTemplate("{question}"),
 ];
 const CHAT_COMBINE_PROMPT = 
-/*#__PURE__*/ chat_js_1.ChatPromptTemplate.fromPromptMessages(combine_messages);
+/*#__PURE__*/ chat_js_1.ChatPromptTemplate.fromMessages(combine_messages);
 exports.COMBINE_PROMPT_SELECTOR = 
 /*#__PURE__*/ new conditional_js_1.ConditionalPromptSelector(exports.COMBINE_PROMPT, [
     [conditional_js_1.isChatModel, CHAT_COMBINE_PROMPT],

package/dist/chains/question_answering/map_reduce_prompts.js

@@ -18,8 +18,7 @@ const messages = [
     /*#__PURE__*/ SystemMessagePromptTemplate.fromTemplate(system_template),
     /*#__PURE__*/ HumanMessagePromptTemplate.fromTemplate("{question}"),
 ];
-const CHAT_QA_PROMPT = 
-/*#__PURE__*/ ChatPromptTemplate.fromPromptMessages(messages);
+const CHAT_QA_PROMPT = /*#__PURE__*/ ChatPromptTemplate.fromMessages(messages);
 export const COMBINE_QA_PROMPT_SELECTOR = 
 /*#__PURE__*/ new ConditionalPromptSelector(DEFAULT_COMBINE_QA_PROMPT, [
     [isChatModel, CHAT_QA_PROMPT],
@@ -65,7 +64,7 @@ const combine_messages = [
     /*#__PURE__*/ HumanMessagePromptTemplate.fromTemplate("{question}"),
 ];
 const CHAT_COMBINE_PROMPT = 
-/*#__PURE__*/ ChatPromptTemplate.fromPromptMessages(combine_messages);
+/*#__PURE__*/ ChatPromptTemplate.fromMessages(combine_messages);
 export const COMBINE_PROMPT_SELECTOR = 
 /*#__PURE__*/ new ConditionalPromptSelector(COMBINE_PROMPT, [
     [isChatModel, CHAT_COMBINE_PROMPT],

package/dist/chains/question_answering/refine_prompts.cjs

@@ -32,7 +32,7 @@ const messages = [
     /*#__PURE__*/ index_js_1.HumanMessagePromptTemplate.fromTemplate(refineTemplate),
 ];
 exports.CHAT_REFINE_PROMPT = 
-/*#__PURE__*/ index_js_1.ChatPromptTemplate.fromPromptMessages(messages);
+/*#__PURE__*/ index_js_1.ChatPromptTemplate.fromMessages(messages);
 exports.REFINE_PROMPT_SELECTOR = 
 /*#__PURE__*/ new conditional_js_1.ConditionalPromptSelector(exports.DEFAULT_REFINE_PROMPT, [
     [conditional_js_1.isChatModel, exports.CHAT_REFINE_PROMPT],
@@ -56,7 +56,7 @@ const chat_messages = [
     /*#__PURE__*/ index_js_1.HumanMessagePromptTemplate.fromTemplate("{question}"),
 ];
 exports.CHAT_QUESTION_PROMPT = 
-/*#__PURE__*/ index_js_1.ChatPromptTemplate.fromPromptMessages(chat_messages);
+/*#__PURE__*/ index_js_1.ChatPromptTemplate.fromMessages(chat_messages);
 exports.QUESTION_PROMPT_SELECTOR = 
 /*#__PURE__*/ new conditional_js_1.ConditionalPromptSelector(exports.DEFAULT_TEXT_QA_PROMPT, [
     [conditional_js_1.isChatModel, exports.CHAT_QUESTION_PROMPT],

package/dist/chains/question_answering/refine_prompts.js

@@ -29,7 +29,7 @@ const messages = [
     /*#__PURE__*/ HumanMessagePromptTemplate.fromTemplate(refineTemplate),
 ];
 export const CHAT_REFINE_PROMPT = 
-/*#__PURE__*/ ChatPromptTemplate.fromPromptMessages(messages);
+/*#__PURE__*/ ChatPromptTemplate.fromMessages(messages);
 export const REFINE_PROMPT_SELECTOR = 
 /*#__PURE__*/ new ConditionalPromptSelector(DEFAULT_REFINE_PROMPT, [
     [isChatModel, CHAT_REFINE_PROMPT],
@@ -53,7 +53,7 @@ const chat_messages = [
     /*#__PURE__*/ HumanMessagePromptTemplate.fromTemplate("{question}"),
 ];
 export const CHAT_QUESTION_PROMPT = 
-/*#__PURE__*/ ChatPromptTemplate.fromPromptMessages(chat_messages);
+/*#__PURE__*/ ChatPromptTemplate.fromMessages(chat_messages);
 export const QUESTION_PROMPT_SELECTOR = 
 /*#__PURE__*/ new ConditionalPromptSelector(DEFAULT_TEXT_QA_PROMPT, [
     [isChatModel, CHAT_QUESTION_PROMPT],

package/dist/chains/question_answering/stuff_prompts.cjs

@@ -17,6 +17,5 @@ const messages = [
     /*#__PURE__*/ chat_js_1.SystemMessagePromptTemplate.fromTemplate(system_template),
     /*#__PURE__*/ chat_js_1.HumanMessagePromptTemplate.fromTemplate("{question}"),
 ];
-const CHAT_PROMPT = 
-/*#__PURE__*/ chat_js_1.ChatPromptTemplate.fromPromptMessages(messages);
+const CHAT_PROMPT = /*#__PURE__*/ chat_js_1.ChatPromptTemplate.fromMessages(messages);
 exports.QA_PROMPT_SELECTOR = new conditional_js_1.ConditionalPromptSelector(exports.DEFAULT_QA_PROMPT, [[conditional_js_1.isChatModel, CHAT_PROMPT]]);

package/dist/chains/question_answering/stuff_prompts.js

@@ -14,6 +14,5 @@ const messages = [
     /*#__PURE__*/ SystemMessagePromptTemplate.fromTemplate(system_template),
     /*#__PURE__*/ HumanMessagePromptTemplate.fromTemplate("{question}"),
 ];
-const CHAT_PROMPT = 
-/*#__PURE__*/ ChatPromptTemplate.fromPromptMessages(messages);
+const CHAT_PROMPT = /*#__PURE__*/ ChatPromptTemplate.fromMessages(messages);
 export const QA_PROMPT_SELECTOR = /*#__PURE__*/ new ConditionalPromptSelector(DEFAULT_QA_PROMPT, [[isChatModel, CHAT_PROMPT]]);

package/dist/chat_models/ollama.cjs

@@ -336,14 +336,10 @@ class ChatOllama extends base_js_1.SimpleChatModel {
         return formattedMessages;
     }
     /** @ignore */
-    async _call(messages, options) {
-        const stream = await this.caller.call(async () => (0, ollama_js_1.createOllamaStream)(this.baseUrl, {
-            ...this.invocationParams(options),
-            prompt: this._formatMessagesAsPrompt(messages),
-        }, options));
+    async _call(messages, options, runManager) {
         const chunks = [];
-        for await (const chunk of stream) {
-            chunks.push(chunk.response);
+        for await (const chunk of this._streamResponseChunks(messages, options, runManager)) {
+            chunks.push(chunk.message.content);
         }
         return chunks.join("");
     }

package/dist/chat_models/ollama.d.ts

@@ -97,5 +97,5 @@ export declare class ChatOllama extends SimpleChatModel implements OllamaInput {
     _streamResponseChunks(input: BaseMessage[], options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): AsyncGenerator<ChatGenerationChunk>;
     protected _formatMessagesAsPrompt(messages: BaseMessage[]): string;
     /** @ignore */
-    _call(messages: BaseMessage[], options: this["ParsedCallOptions"]): Promise<string>;
+    _call(messages: BaseMessage[], options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): Promise<string>;
 }

package/dist/chat_models/ollama.js

@@ -333,14 +333,10 @@ export class ChatOllama extends SimpleChatModel {
         return formattedMessages;
     }
     /** @ignore */
-    async _call(messages, options) {
-        const stream = await this.caller.call(async () => createOllamaStream(this.baseUrl, {
-            ...this.invocationParams(options),
-            prompt: this._formatMessagesAsPrompt(messages),
-        }, options));
+    async _call(messages, options, runManager) {
         const chunks = [];
-        for await (const chunk of stream) {
-            chunks.push(chunk.response);
+        for await (const chunk of this._streamResponseChunks(messages, options, runManager)) {
+            chunks.push(chunk.message.content);
         }
         return chunks.join("");
     }

package/dist/document_loaders/web/pdf.cjs

@@ -0,0 +1,87 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WebPDFLoader = void 0;
+const pdf_js_1 = require("pdf-parse/lib/pdf.js/v1.10.100/build/pdf.js");
+const document_js_1 = require("../../document.cjs");
+const base_js_1 = require("../base.cjs");
+/**
+ * A document loader for loading data from PDFs.
+ */
+class WebPDFLoader extends base_js_1.BaseDocumentLoader {
+    constructor(blob, { splitPages = true } = {}) {
+        super();
+        Object.defineProperty(this, "blob", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "splitPages", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: true
+        });
+        this.blob = blob;
+        this.splitPages = splitPages ?? this.splitPages;
+    }
+    /**
+     * Loads the contents of the PDF as documents.
+     * @returns An array of Documents representing the retrieved data.
+     */
+    async load() {
+        const parsedPdf = await (0, pdf_js_1.getDocument)({
+            data: new Uint8Array(await this.blob.arrayBuffer()),
+            useWorkerFetch: false,
+            isEvalSupported: false,
+            useSystemFonts: true,
+        }).promise;
+        const meta = await parsedPdf.getMetadata().catch(() => null);
+        const documents = [];
+        for (let i = 1; i <= parsedPdf.numPages; i += 1) {
+            const page = await parsedPdf.getPage(i);
+            const content = await page.getTextContent();
+            if (content.items.length === 0) {
+                continue;
+            }
+            const text = content.items
+                .map((item) => item.str)
+                .join("\n");
+            documents.push(new document_js_1.Document({
+                pageContent: text,
+                metadata: {
+                    pdf: {
+                        version: pdf_js_1.version,
+                        info: meta?.info,
+                        metadata: meta?.metadata,
+                        totalPages: parsedPdf.numPages,
+                    },
+                    loc: {
+                        pageNumber: i,
+                    },
+                },
+            }));
+        }
+        if (this.splitPages) {
+            return documents;
+        }
+        if (documents.length === 0) {
+            return [];
+        }
+        return [
+            new document_js_1.Document({
+                pageContent: documents.map((doc) => doc.pageContent).join("\n\n"),
+                metadata: {
+                    pdf: {
+                        version: pdf_js_1.version,
+                        info: meta?.info,
+                        metadata: meta?.metadata,
+                        totalPages: parsedPdf.numPages,
+                    },
+                },
+            }),
+        ];
+        return documents;
+    }
+}
+exports.WebPDFLoader = WebPDFLoader;

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

@@ -0,0 +1,17 @@
+import { Document } from "../../document.js";
+import { BaseDocumentLoader } from "../base.js";
+/**
+ * A document loader for loading data from PDFs.
+ */
+export declare class WebPDFLoader extends BaseDocumentLoader {
+    protected blob: Blob;
+    protected splitPages: boolean;
+    constructor(blob: Blob, { splitPages }?: {
+        splitPages?: boolean | undefined;
+    });
+    /**
+     * Loads the contents of the PDF as documents.
+     * @returns An array of Documents representing the retrieved data.
+     */
+    load(): Promise<Document[]>;
+}