langchain 0.0.155 → 0.0.157

package/dist/llms/bedrock.d.ts

@@ -1,60 +1,34 @@
 import { EventStreamCodec } from "@smithy/eventstream-codec";
-import type { AwsCredentialIdentity, Provider } from "@aws-sdk/types";
+import { BaseBedrockInput, type CredentialType } from "../util/bedrock.js";
 import { LLM, BaseLLMParams } from "./base.js";
 import { CallbackManagerForLLMRun } from "../callbacks/manager.js";
 import { GenerationChunk } from "../schema/index.js";
-type CredentialType = AwsCredentialIdentity | Provider<AwsCredentialIdentity>;
-/** Bedrock models.
-    To authenticate, the AWS client uses the following methods to automatically load credentials:
-    https://boto3.amazonaws.com/v1/documentation/api/latest/guide/credentials.html
-    If a specific credential profile should be used, you must pass the name of the profile from the ~/.aws/credentials file that is to be used.
-    Make sure the credentials / roles used have the required policies to access the Bedrock service.
-*/
-export interface BedrockInput {
-    /** Model to use.
-        For example, "amazon.titan-tg1-large", this is equivalent to the modelId property in the list-foundation-models api.
-    */
-    model: string;
-    /** The AWS region e.g. `us-west-2`.
-        Fallback to AWS_DEFAULT_REGION env variable or region specified in ~/.aws/config in case it is not provided here.
-    */
-    region?: string;
-    /** AWS Credentials.
-        If no credentials are provided, the default credentials from `@aws-sdk/credential-provider-node` will be used.
-     */
-    credentials?: CredentialType;
-    /** Temperature */
-    temperature?: number;
-    /** Max tokens */
-    maxTokens?: number;
-    /** A custom fetch function for low-level access to AWS API. Defaults to fetch() */
-    fetchFn?: typeof fetch;
-    /** Override the default endpoint url */
-    endpointUrl?: string;
-}
 /**
  * A type of Large Language Model (LLM) that interacts with the Bedrock
  * service. It extends the base `LLM` class and implements the
- * `BedrockInput` interface. The class is designed to authenticate and
+ * `BaseBedrockInput` interface. The class is designed to authenticate and
  * interact with the Bedrock service, which is a part of Amazon Web
  * Services (AWS). It uses AWS credentials for authentication and can be
  * configured with various parameters such as the model to use, the AWS
  * region, and the maximum number of tokens to generate.
  */
-export declare class Bedrock extends LLM implements BedrockInput {
+export declare class Bedrock extends LLM implements BaseBedrockInput {
     model: string;
     region: string;
     credentials: CredentialType;
     temperature?: number | undefined;
     maxTokens?: number | undefined;
     fetchFn: typeof fetch;
-    endpointUrl?: string;
+    endpointHost?: string;
+    stopSequences?: string[];
+    modelKwargs?: Record<string, unknown>;
     codec: EventStreamCodec;
     get lc_secrets(): {
         [key: string]: string;
     } | undefined;
     _llmType(): string;
-    constructor(fields?: Partial<BedrockInput> & BaseLLMParams);
+    static lc_name(): string;
+    constructor(fields?: Partial<BaseBedrockInput> & BaseLLMParams);
     /** Call out to Bedrock service model.
       Arguments:
         prompt: The prompt to pass into the model.
@@ -71,4 +45,3 @@ export declare class Bedrock extends LLM implements BedrockInput {
         [Symbol.asyncIterator](): AsyncGenerator<any, void, unknown>;
     };
 }
-export {};

package/dist/llms/bedrock.js

@@ -1,64 +1,17 @@
-import { SignatureV4 } from "@aws-sdk/signature-v4";
+import { SignatureV4 } from "@smithy/signature-v4";
 import { defaultProvider } from "@aws-sdk/credential-provider-node";
-import { HttpRequest } from "@aws-sdk/protocol-http";
+import { HttpRequest } from "@smithy/protocol-http";
 import { EventStreamCodec } from "@smithy/eventstream-codec";
 import { fromUtf8, toUtf8 } from "@smithy/util-utf8";
 import { Sha256 } from "@aws-crypto/sha256-js";
+import { BedrockLLMInputOutputAdapter, } from "../util/bedrock.js";
 import { getEnvironmentVariable } from "../util/env.js";
 import { LLM } from "./base.js";
 import { GenerationChunk } from "../schema/index.js";
-/**
- * A helper class used within the `Bedrock` class. It is responsible for
- * preparing the input and output for the Bedrock service. It formats the
- * input prompt based on the provider (e.g., "anthropic", "ai21",
- * "amazon") and extracts the generated text from the service response.
- */
-class BedrockLLMInputOutputAdapter {
-    /** Adapter class to prepare the inputs from Langchain to a format
-    that LLM model expects. Also, provides a helper function to extract
-    the generated text from the model response. */
-    static prepareInput(provider, prompt, maxTokens = 50, temperature = 0) {
-        const inputBody = {};
-        if (provider === "anthropic") {
-            inputBody.prompt = prompt;
-            inputBody.max_tokens_to_sample = maxTokens;
-            inputBody.temperature = temperature;
-        }
-        else if (provider === "ai21") {
-            inputBody.prompt = prompt;
-            inputBody.maxTokens = maxTokens;
-            inputBody.temperature = temperature;
-        }
-        else if (provider === "amazon") {
-            inputBody.inputText = prompt;
-            inputBody.textGenerationConfig = {
-                maxTokenCount: maxTokens,
-                temperature,
-            };
-        }
-        return inputBody;
-    }
-    /**
-     * Extracts the generated text from the service response.
-     * @param provider The provider name.
-     * @param responseBody The response body from the service.
-     * @returns The generated text.
-     */
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    static prepareOutput(provider, responseBody) {
-        if (provider === "anthropic") {
-            return responseBody.completion;
-        }
-        else if (provider === "ai21") {
-            return responseBody.data.text;
-        }
-        return responseBody.outputText;
-    }
-}
 /**
  * A type of Large Language Model (LLM) that interacts with the Bedrock
  * service. It extends the base `LLM` class and implements the
- * `BedrockInput` interface. The class is designed to authenticate and
+ * `BaseBedrockInput` interface. The class is designed to authenticate and
  * interact with the Bedrock service, which is a part of Amazon Web
  * Services (AWS). It uses AWS credentials for authentication and can be
  * configured with various parameters such as the model to use, the AWS
@@ -71,6 +24,9 @@ export class Bedrock extends LLM {
     _llmType() {
         return "bedrock";
     }
+    static lc_name() {
+        return "Bedrock";
+    }
     constructor(fields) {
         super(fields ?? {});
         Object.defineProperty(this, "model", {
@@ -109,7 +65,19 @@ export class Bedrock extends LLM {
             writable: true,
             value: void 0
         });
-        Object.defineProperty(this, "endpointUrl", {
+        Object.defineProperty(this, "endpointHost", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "stopSequences", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "modelKwargs", {
             enumerable: true,
             configurable: true,
             writable: true,
@@ -135,7 +103,9 @@ export class Bedrock extends LLM {
         this.temperature = fields?.temperature ?? this.temperature;
         this.maxTokens = fields?.maxTokens ?? this.maxTokens;
         this.fetchFn = fields?.fetchFn ?? fetch;
-        this.endpointUrl = fields?.endpointUrl;
+        this.endpointHost = fields?.endpointHost ?? fields?.endpointUrl;
+        this.stopSequences = fields?.stopSequences;
+        this.modelKwargs = fields?.modelKwargs;
     }
     /** Call out to Bedrock service model.
       Arguments:
@@ -156,10 +126,11 @@ export class Bedrock extends LLM {
     }
     async *_streamResponseChunks(prompt, options, runManager) {
         const provider = this.model.split(".")[0];
-        const service = "bedrock";
-        const inputBody = BedrockLLMInputOutputAdapter.prepareInput(provider, prompt, this.maxTokens, this.temperature);
-        const endpointUrl = this.endpointUrl ?? `${service}.${this.region}.amazonaws.com`;
-        const url = new URL(`https://${endpointUrl}/model/${this.model}/invoke-with-response-stream`);
+        const service = "bedrock-runtime";
+        const inputBody = BedrockLLMInputOutputAdapter.prepareInput(provider, prompt, this.maxTokens, this.temperature, this.stopSequences, this.modelKwargs);
+        const endpointHost = this.endpointHost ?? `${service}.${this.region}.amazonaws.com`;
+        const amazonMethod = provider === "anthropic" ? "invoke-with-response-stream" : "invoke";
+        const url = new URL(`https://${endpointHost}/model/${this.model}/${amazonMethod}`);
         const request = new HttpRequest({
             hostname: url.hostname,
             path: url.pathname,
@@ -171,12 +142,12 @@ export class Bedrock extends LLM {
                 // host is required by AWS Signature V4: https://docs.aws.amazon.com/general/latest/gr/sigv4-create-canonical-request.html
                 host: url.host,
                 accept: "application/json",
-                "Content-Type": "application/json",
+                "content-type": "application/json",
             },
         });
         const signer = new SignatureV4({
             credentials: this.credentials,
-            service,
+            service: "bedrock",
             region: this.region,
             sha256: Sha256,
         });
@@ -190,15 +161,34 @@ export class Bedrock extends LLM {
         if (response.status < 200 || response.status >= 300) {
             throw Error(`Failed to access underlying url '${url}': got ${response.status} ${response.statusText}: ${await response.text()}`);
         }
-        const reader = response.body?.getReader();
-        for await (const chunk of this._readChunks(reader)) {
-            const event = this.codec.decode(chunk);
-            if (event.headers[":event-type"].value !== "chunk" ||
-                event.headers[":content-type"].value !== "application/json") {
-                throw Error(`Failed to get event chunk: got ${chunk}`);
+        if (provider === "anthropic") {
+            const reader = response.body?.getReader();
+            const decoder = new TextDecoder();
+            for await (const chunk of this._readChunks(reader)) {
+                const event = this.codec.decode(chunk);
+                if ((event.headers[":event-type"] !== undefined &&
+                    event.headers[":event-type"].value !== "chunk") ||
+                    event.headers[":content-type"].value !== "application/json") {
+                    throw Error(`Failed to get event chunk: got ${chunk}`);
+                }
+                const body = JSON.parse(decoder.decode(event.body));
+                if (body.message) {
+                    throw new Error(body.message);
+                }
+                if (body.bytes !== undefined) {
+                    const chunkResult = JSON.parse(Buffer.from(body.bytes, "base64").toString());
+                    const text = BedrockLLMInputOutputAdapter.prepareOutput(provider, chunkResult);
+                    yield new GenerationChunk({
+                        text,
+                        generationInfo: {},
+                    });
+                    await runManager?.handleLLMNewToken(text);
+                }
             }
-            const body = JSON.parse(Buffer.from(JSON.parse(new TextDecoder("utf-8").decode(event.body)).bytes, "base64").toString());
-            const text = BedrockLLMInputOutputAdapter.prepareOutput(provider, body);
+        }
+        else {
+            const json = await response.json();
+            const text = BedrockLLMInputOutputAdapter.prepareOutput(provider, json);
             yield new GenerationChunk({
                 text,
                 generationInfo: {},

package/dist/llms/openai-chat.d.ts

@@ -3,7 +3,7 @@ import { CallbackManagerForLLMRun } from "../callbacks/manager.js";
 import { GenerationChunk, LLMResult } from "../schema/index.js";
 import { AzureOpenAIInput, OpenAICallOptions, OpenAIChatInput, OpenAICoreRequestOptions, LegacyOpenAIInput } from "../types/openai-types.js";
 import { BaseLLMParams, LLM } from "./base.js";
-export { AzureOpenAIInput, OpenAIChatInput };
+export { type AzureOpenAIInput, type OpenAIChatInput };
 /**
  * Interface that extends the OpenAICallOptions interface and includes an
  * optional promptIndex property. It represents the options that can be

package/dist/llms/openai.d.ts

@@ -3,7 +3,7 @@ import { CallbackManagerForLLMRun } from "../callbacks/manager.js";
 import { GenerationChunk, LLMResult } from "../schema/index.js";
 import { AzureOpenAIInput, OpenAICallOptions, OpenAICoreRequestOptions, OpenAIInput, LegacyOpenAIInput } from "../types/openai-types.js";
 import { BaseLLM, BaseLLMParams } from "./base.js";
-export { AzureOpenAIInput, OpenAICallOptions, OpenAIInput };
+export type { AzureOpenAIInput, OpenAICallOptions, OpenAIInput };
 /**
  * Wrapper around OpenAI large language models.
  *

package/dist/load/import_constants.cjs

@@ -15,6 +15,8 @@ exports.optionalImportEntrypoints = [
     "langchain/chains/query_constructor",
     "langchain/chains/query_constructor/ir",
     "langchain/chains/sql_db",
+    "langchain/chains/graph_qa/cypher",
+    "langchain/embeddings/cloudflare_workersai",
     "langchain/embeddings/cohere",
     "langchain/embeddings/tensorflow",
     "langchain/embeddings/hf",
@@ -36,6 +38,7 @@ exports.optionalImportEntrypoints = [
     "langchain/prompts/load",
     "langchain/vectorstores/analyticdb",
     "langchain/vectorstores/elasticsearch",
+    "langchain/vectorstores/cloudflare_vectorize",
     "langchain/vectorstores/chroma",
     "langchain/vectorstores/googlevertexai",
     "langchain/vectorstores/hnswlib",
@@ -96,6 +99,7 @@ exports.optionalImportEntrypoints = [
     "langchain/document_loaders/fs/openai_whisper_audio",
     "langchain/document_transformers/html_to_text",
     "langchain/document_transformers/mozilla_readability",
+    "langchain/chat_models/bedrock",
     "langchain/chat_models/googlevertexai",
     "langchain/chat_models/googlevertexai/web",
     "langchain/chat_models/googlepalm",
@@ -130,6 +134,7 @@ exports.optionalImportEntrypoints = [
     "langchain/stores/message/planetscale",
     "langchain/stores/message/xata",
     "langchain/storage/ioredis",
+    "langchain/graphs/neo4j_graph",
     "langchain/hub",
     "langchain/experimental/multimodal_embeddings/googlevertexai",
     "langchain/experimental/chat_models/anthropic_functions",

package/dist/load/import_constants.js

@@ -12,6 +12,8 @@ export const optionalImportEntrypoints = [
     "langchain/chains/query_constructor",
     "langchain/chains/query_constructor/ir",
     "langchain/chains/sql_db",
+    "langchain/chains/graph_qa/cypher",
+    "langchain/embeddings/cloudflare_workersai",
     "langchain/embeddings/cohere",
     "langchain/embeddings/tensorflow",
     "langchain/embeddings/hf",
@@ -33,6 +35,7 @@ export const optionalImportEntrypoints = [
     "langchain/prompts/load",
     "langchain/vectorstores/analyticdb",
     "langchain/vectorstores/elasticsearch",
+    "langchain/vectorstores/cloudflare_vectorize",
     "langchain/vectorstores/chroma",
     "langchain/vectorstores/googlevertexai",
     "langchain/vectorstores/hnswlib",
@@ -93,6 +96,7 @@ export const optionalImportEntrypoints = [
     "langchain/document_loaders/fs/openai_whisper_audio",
     "langchain/document_transformers/html_to_text",
     "langchain/document_transformers/mozilla_readability",
+    "langchain/chat_models/bedrock",
     "langchain/chat_models/googlevertexai",
     "langchain/chat_models/googlevertexai/web",
     "langchain/chat_models/googlepalm",
@@ -127,6 +131,7 @@ export const optionalImportEntrypoints = [
     "langchain/stores/message/planetscale",
     "langchain/stores/message/xata",
     "langchain/storage/ioredis",
+    "langchain/graphs/neo4j_graph",
     "langchain/hub",
     "langchain/experimental/multimodal_embeddings/googlevertexai",
     "langchain/experimental/chat_models/anthropic_functions",

package/dist/memory/index.d.ts

@@ -1,12 +1,12 @@
-export { BufferMemory, BufferMemoryInput } from "./buffer_memory.js";
+export { BufferMemory, type BufferMemoryInput } from "./buffer_memory.js";
 export { BaseMemory, getInputValue, getBufferString } from "./base.js";
-export { ConversationSummaryMemory, ConversationSummaryMemoryInput, BaseConversationSummaryMemory, BaseConversationSummaryMemoryInput, } from "./summary.js";
-export { BufferWindowMemory, BufferWindowMemoryInput, } from "./buffer_window_memory.js";
-export { BaseChatMemory, BaseChatMemoryInput } from "./chat_memory.js";
+export { ConversationSummaryMemory, type ConversationSummaryMemoryInput, BaseConversationSummaryMemory, type BaseConversationSummaryMemoryInput, } from "./summary.js";
+export { BufferWindowMemory, type BufferWindowMemoryInput, } from "./buffer_window_memory.js";
+export { BaseChatMemory, type BaseChatMemoryInput } from "./chat_memory.js";
 export { ChatMessageHistory } from "../stores/message/in_memory.js";
-export { MotorheadMemory, MotorheadMemoryInput } from "./motorhead_memory.js";
-export { VectorStoreRetrieverMemory, VectorStoreRetrieverMemoryParams, } from "./vector_store.js";
+export { MotorheadMemory, type MotorheadMemoryInput, } from "./motorhead_memory.js";
+export { VectorStoreRetrieverMemory, type VectorStoreRetrieverMemoryParams, } from "./vector_store.js";
 export { EntityMemory } from "./entity_memory.js";
 export { ENTITY_MEMORY_CONVERSATION_TEMPLATE } from "./prompt.js";
-export { CombinedMemoryInput, CombinedMemory } from "./combined_memory.js";
-export { ConversationSummaryBufferMemory, ConversationSummaryBufferMemoryInput, } from "./summary_buffer.js";
+export { type CombinedMemoryInput, CombinedMemory } from "./combined_memory.js";
+export { ConversationSummaryBufferMemory, type ConversationSummaryBufferMemoryInput, } from "./summary_buffer.js";

package/dist/memory/index.js

@@ -4,7 +4,7 @@ export { ConversationSummaryMemory, BaseConversationSummaryMemory, } from "./sum
 export { BufferWindowMemory, } from "./buffer_window_memory.js";
 export { BaseChatMemory } from "./chat_memory.js";
 export { ChatMessageHistory } from "../stores/message/in_memory.js";
-export { MotorheadMemory } from "./motorhead_memory.js";
+export { MotorheadMemory, } from "./motorhead_memory.js";
 export { VectorStoreRetrieverMemory, } from "./vector_store.js";
 export { EntityMemory } from "./entity_memory.js";
 export { ENTITY_MEMORY_CONVERSATION_TEMPLATE } from "./prompt.js";

package/dist/output_parsers/index.d.ts

@@ -1,8 +1,8 @@
 export { ListOutputParser, CommaSeparatedListOutputParser } from "./list.js";
 export { RegexParser } from "./regex.js";
-export { StructuredOutputParser, AsymmetricStructuredOutputParser, JsonMarkdownStructuredOutputParser, JsonMarkdownFormatInstructionsOptions, JsonMarkdownStructuredOutputParserInput, } from "./structured.js";
+export { StructuredOutputParser, AsymmetricStructuredOutputParser, JsonMarkdownStructuredOutputParser, type JsonMarkdownFormatInstructionsOptions, type JsonMarkdownStructuredOutputParserInput, } from "./structured.js";
 export { OutputFixingParser } from "./fix.js";
 export { CombiningOutputParser } from "./combining.js";
-export { RouterOutputParser, RouterOutputParserInput } from "./router.js";
+export { RouterOutputParser, type RouterOutputParserInput } from "./router.js";
 export { CustomListOutputParser } from "./list.js";
-export { FunctionParameters, OutputFunctionsParser, JsonOutputFunctionsParser, JsonKeyOutputFunctionsParser, } from "../output_parsers/openai_functions.js";
+export { type FunctionParameters, OutputFunctionsParser, JsonOutputFunctionsParser, JsonKeyOutputFunctionsParser, } from "../output_parsers/openai_functions.js";

package/dist/prompts/index.d.ts

@@ -1,10 +1,10 @@
-export { BaseExampleSelector, BasePromptTemplate, BasePromptTemplateInput, StringPromptValue, BaseStringPromptTemplate, } from "./base.js";
-export { PromptTemplate, PromptTemplateInput } from "./prompt.js";
+export { BaseExampleSelector, BasePromptTemplate, type BasePromptTemplateInput, StringPromptValue, BaseStringPromptTemplate, } from "./base.js";
+export { PromptTemplate, type PromptTemplateInput } from "./prompt.js";
 export { BasePromptSelector, ConditionalPromptSelector, isChatModel, isLLM, } from "./selectors/conditional.js";
-export { LengthBasedExampleSelector, LengthBasedExampleSelectorInput, } from "./selectors/LengthBasedExampleSelector.js";
-export { SemanticSimilarityExampleSelector, SemanticSimilarityExampleSelectorInput, } from "./selectors/SemanticSimilarityExampleSelector.js";
-export { FewShotPromptTemplate, FewShotPromptTemplateInput, } from "./few_shot.js";
+export { LengthBasedExampleSelector, type LengthBasedExampleSelectorInput, } from "./selectors/LengthBasedExampleSelector.js";
+export { SemanticSimilarityExampleSelector, type SemanticSimilarityExampleSelectorInput, } from "./selectors/SemanticSimilarityExampleSelector.js";
+export { FewShotPromptTemplate, type FewShotPromptTemplateInput, } from "./few_shot.js";
 export { ChatPromptTemplate, HumanMessagePromptTemplate, AIMessagePromptTemplate, SystemMessagePromptTemplate, ChatMessagePromptTemplate, MessagesPlaceholder, BaseChatPromptTemplate, } from "./chat.js";
-export { SerializedPromptTemplate, SerializedBasePromptTemplate, SerializedFewShotTemplate, } from "./serde.js";
-export { parseTemplate, renderTemplate, checkValidTemplate, TemplateFormat, } from "./template.js";
-export { PipelinePromptParams, PipelinePromptTemplate, PipelinePromptTemplateInput, } from "./pipeline.js";
+export { type SerializedPromptTemplate, type SerializedBasePromptTemplate, type SerializedFewShotTemplate, } from "./serde.js";
+export { parseTemplate, renderTemplate, checkValidTemplate, type TemplateFormat, } from "./template.js";
+export { type PipelinePromptParams, PipelinePromptTemplate, type PipelinePromptTemplateInput, } from "./pipeline.js";

package/dist/retrievers/remote/index.d.ts

@@ -1,3 +1,3 @@
-export { RemoteRetriever, RemoteRetrieverParams, RemoteRetrieverAuth, RemoteRetrieverValues, } from "./base.js";
-export { ChatGPTPluginRetriever, ChatGPTPluginRetrieverFilter, ChatGPTPluginRetrieverParams, } from "./chatgpt-plugin.js";
-export { RemoteLangChainRetriever, RemoteLangChainRetrieverParams, } from "./remote-retriever.js";
+export { RemoteRetriever, type RemoteRetrieverParams, type RemoteRetrieverAuth, type RemoteRetrieverValues, } from "./base.js";
+export { ChatGPTPluginRetriever, type ChatGPTPluginRetrieverFilter, type ChatGPTPluginRetrieverParams, } from "./chatgpt-plugin.js";
+export { RemoteLangChainRetriever, type RemoteLangChainRetrieverParams, } from "./remote-retriever.js";

package/dist/schema/runnable/index.d.ts

@@ -1,5 +1,5 @@
-export { RunnableFunc, RunnableLike, RunnableBatchOptions, RunnableRetryFailedAttemptHandler, Runnable, RunnableBindingArgs, RunnableBinding, RunnableEach, RunnableRetry, RunnableSequence, RunnableMap, RunnableLambda, RunnableWithFallbacks, } from "./base.js";
-export { RunnableConfig } from "./config.js";
+export { type RunnableFunc, type RunnableLike, type RunnableBatchOptions, type RunnableRetryFailedAttemptHandler, Runnable, type RunnableBindingArgs, RunnableBinding, RunnableEach, RunnableRetry, RunnableSequence, RunnableMap, RunnableLambda, RunnableWithFallbacks, } from "./base.js";
+export type { RunnableConfig } from "./config.js";
 export { RunnablePassthrough } from "./passthrough.js";
 export { RouterRunnable } from "./router.js";
-export { RunnableBranch, Branch, BranchLike } from "./branch.js";
+export { RunnableBranch, type Branch, type BranchLike } from "./branch.js";

package/dist/sql_db.d.ts

@@ -1,7 +1,7 @@
 import type { DataSource as DataSourceT, DataSourceOptions } from "typeorm";
 import { SerializedSqlDatabase, SqlDatabaseDataSourceParams, SqlDatabaseOptionsParams, SqlTable } from "./util/sql_utils.js";
 import { Serializable } from "./load/serializable.js";
-export { SqlDatabaseDataSourceParams, SqlDatabaseOptionsParams };
+export type { SqlDatabaseDataSourceParams, SqlDatabaseOptionsParams };
 export declare class SqlDatabase extends Serializable implements SqlDatabaseOptionsParams, SqlDatabaseDataSourceParams {
     lc_namespace: string[];
     toJSON(): import("./load/serializable.js").SerializedNotImplemented;

package/dist/tools/index.d.ts

@@ -1,21 +1,21 @@
-export { SerpAPI, SerpAPIParameters } from "./serpapi.js";
+export { SerpAPI, type SerpAPIParameters } from "./serpapi.js";
 export { DadJokeAPI } from "./dadjokeapi.js";
 export { BingSerpAPI } from "./bingserpapi.js";
-export { Tool, ToolParams, StructuredTool } from "./base.js";
-export { DynamicTool, DynamicToolInput, DynamicStructuredTool, DynamicStructuredToolInput, } from "./dynamic.js";
+export { Tool, type ToolParams, StructuredTool } from "./base.js";
+export { DynamicTool, type DynamicToolInput, DynamicStructuredTool, type DynamicStructuredToolInput, } from "./dynamic.js";
 export { IFTTTWebhook } from "./IFTTTWebhook.js";
-export { ChainTool, ChainToolInput } from "./chain.js";
-export { JsonSpec, JsonListKeysTool, JsonGetValueTool, JsonObject, Json, } from "./json.js";
+export { ChainTool, type ChainToolInput } from "./chain.js";
+export { JsonSpec, JsonListKeysTool, JsonGetValueTool, type JsonObject, type Json, } from "./json.js";
 export { RequestsGetTool, RequestsPostTool } from "./requests.js";
 export { VectorStoreQATool } from "./vectorstore.js";
-export { ZapierNLARunAction, ZapierNLAWrapper, ZapierNLAWrapperParams, } from "./zapier.js";
-export { Serper, SerperParameters } from "./serper.js";
-export { GoogleCustomSearch, GoogleCustomSearchParams, } from "./google_custom_search.js";
+export { ZapierNLARunAction, ZapierNLAWrapper, type ZapierNLAWrapperParams, } from "./zapier.js";
+export { Serper, type SerperParameters } from "./serper.js";
+export { GoogleCustomSearch, type GoogleCustomSearchParams, } from "./google_custom_search.js";
 export { AIPluginTool } from "./aiplugin.js";
 export { ReadFileTool, WriteFileTool } from "./fs.js";
-export { BraveSearch, BraveSearchParams } from "./brave_search.js";
-export { WikipediaQueryRun, WikipediaQueryRunParams, } from "./wikipedia_query_run.js";
+export { BraveSearch, type BraveSearchParams } from "./brave_search.js";
+export { WikipediaQueryRun, type WikipediaQueryRunParams, } from "./wikipedia_query_run.js";
 export { WolframAlphaTool } from "./wolframalpha.js";
-export { DataForSeoAPISearch, DataForSeoApiConfig, } from "./dataforseo_api_search.js";
+export { DataForSeoAPISearch, type DataForSeoApiConfig, } from "./dataforseo_api_search.js";
 export { SearxngSearch } from "./searxng_search.js";
-export { SearchApi, SearchApiParameters } from "./searchapi.js";
+export { SearchApi, type SearchApiParameters } from "./searchapi.js";

package/dist/util/bedrock.cjs

@@ -0,0 +1,54 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BedrockLLMInputOutputAdapter = void 0;
+/**
+ * A helper class used within the `Bedrock` class. It is responsible for
+ * preparing the input and output for the Bedrock service. It formats the
+ * input prompt based on the provider (e.g., "anthropic", "ai21",
+ * "amazon") and extracts the generated text from the service response.
+ */
+class BedrockLLMInputOutputAdapter {
+    /** Adapter class to prepare the inputs from Langchain to a format
+    that LLM model expects. Also, provides a helper function to extract
+    the generated text from the model response. */
+    static prepareInput(provider, prompt, maxTokens = 50, temperature = 0, stopSequences = undefined, modelKwargs = {}) {
+        const inputBody = {};
+        if (provider === "anthropic") {
+            inputBody.prompt = prompt;
+            inputBody.max_tokens_to_sample = maxTokens;
+            inputBody.temperature = temperature;
+            inputBody.stop_sequences = stopSequences;
+        }
+        else if (provider === "ai21") {
+            inputBody.prompt = prompt;
+            inputBody.maxTokens = maxTokens;
+            inputBody.temperature = temperature;
+            inputBody.stopSequences = stopSequences;
+        }
+        else if (provider === "amazon") {
+            inputBody.inputText = prompt;
+            inputBody.textGenerationConfig = {
+                maxTokenCount: maxTokens,
+                temperature,
+            };
+        }
+        return { ...inputBody, ...modelKwargs };
+    }
+    /**
+     * Extracts the generated text from the service response.
+     * @param provider The provider name.
+     * @param responseBody The response body from the service.
+     * @returns The generated text.
+     */
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    static prepareOutput(provider, responseBody) {
+        if (provider === "anthropic") {
+            return responseBody.completion;
+        }
+        else if (provider === "ai21") {
+            return responseBody?.completions?.[0]?.data?.text ?? "";
+        }
+        return responseBody.outputText;
+    }
+}
+exports.BedrockLLMInputOutputAdapter = BedrockLLMInputOutputAdapter;

package/dist/util/bedrock.d.ts

@@ -0,0 +1,59 @@
+import type { AwsCredentialIdentity, Provider } from "@aws-sdk/types";
+export type CredentialType = AwsCredentialIdentity | Provider<AwsCredentialIdentity>;
+/** Bedrock models.
+    To authenticate, the AWS client uses the following methods to automatically load credentials:
+    https://boto3.amazonaws.com/v1/documentation/api/latest/guide/credentials.html
+    If a specific credential profile should be used, you must pass the name of the profile from the ~/.aws/credentials file that is to be used.
+    Make sure the credentials / roles used have the required policies to access the Bedrock service.
+*/
+export interface BaseBedrockInput {
+    /** Model to use.
+        For example, "amazon.titan-tg1-large", this is equivalent to the modelId property in the list-foundation-models api.
+    */
+    model: string;
+    /** The AWS region e.g. `us-west-2`.
+        Fallback to AWS_DEFAULT_REGION env variable or region specified in ~/.aws/config in case it is not provided here.
+    */
+    region?: string;
+    /** AWS Credentials.
+        If no credentials are provided, the default credentials from `@aws-sdk/credential-provider-node` will be used.
+     */
+    credentials?: CredentialType;
+    /** Temperature. */
+    temperature?: number;
+    /** Max tokens. */
+    maxTokens?: number;
+    /** A custom fetch function for low-level access to AWS API. Defaults to fetch(). */
+    fetchFn?: typeof fetch;
+    /** @deprecated Use endpointHost instead Override the default endpoint url. */
+    endpointUrl?: string;
+    /** Override the default endpoint hostname. */
+    endpointHost?: string;
+    /** Optional additional stop sequences to pass to the model. Currently only supported for Anthropic and AI21. */
+    stopSequences?: string[];
+    /** Additional kwargs to pass to the model. */
+    modelKwargs?: Record<string, unknown>;
+}
+type Dict = {
+    [key: string]: unknown;
+};
+/**
+ * A helper class used within the `Bedrock` class. It is responsible for
+ * preparing the input and output for the Bedrock service. It formats the
+ * input prompt based on the provider (e.g., "anthropic", "ai21",
+ * "amazon") and extracts the generated text from the service response.
+ */
+export declare class BedrockLLMInputOutputAdapter {
+    /** Adapter class to prepare the inputs from Langchain to a format
+    that LLM model expects. Also, provides a helper function to extract
+    the generated text from the model response. */
+    static prepareInput(provider: string, prompt: string, maxTokens?: number, temperature?: number, stopSequences?: string[] | undefined, modelKwargs?: Record<string, unknown>): Dict;
+    /**
+     * Extracts the generated text from the service response.
+     * @param provider The provider name.
+     * @param responseBody The response body from the service.
+     * @returns The generated text.
+     */
+    static prepareOutput(provider: string, responseBody: any): string;
+}
+export {};

package/dist/util/bedrock.js

@@ -0,0 +1,50 @@
+/**
+ * A helper class used within the `Bedrock` class. It is responsible for
+ * preparing the input and output for the Bedrock service. It formats the
+ * input prompt based on the provider (e.g., "anthropic", "ai21",
+ * "amazon") and extracts the generated text from the service response.
+ */
+export class BedrockLLMInputOutputAdapter {
+    /** Adapter class to prepare the inputs from Langchain to a format
+    that LLM model expects. Also, provides a helper function to extract
+    the generated text from the model response. */
+    static prepareInput(provider, prompt, maxTokens = 50, temperature = 0, stopSequences = undefined, modelKwargs = {}) {
+        const inputBody = {};
+        if (provider === "anthropic") {
+            inputBody.prompt = prompt;
+            inputBody.max_tokens_to_sample = maxTokens;
+            inputBody.temperature = temperature;
+            inputBody.stop_sequences = stopSequences;
+        }
+        else if (provider === "ai21") {
+            inputBody.prompt = prompt;
+            inputBody.maxTokens = maxTokens;
+            inputBody.temperature = temperature;
+            inputBody.stopSequences = stopSequences;
+        }
+        else if (provider === "amazon") {
+            inputBody.inputText = prompt;
+            inputBody.textGenerationConfig = {
+                maxTokenCount: maxTokens,
+                temperature,
+            };
+        }
+        return { ...inputBody, ...modelKwargs };
+    }
+    /**
+     * Extracts the generated text from the service response.
+     * @param provider The provider name.
+     * @param responseBody The response body from the service.
+     * @returns The generated text.
+     */
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    static prepareOutput(provider, responseBody) {
+        if (provider === "anthropic") {
+            return responseBody.completion;
+        }
+        else if (provider === "ai21") {
+            return responseBody?.completions?.[0]?.data?.text ?? "";
+        }
+        return responseBody.outputText;
+    }
+}