langchain 0.0.148 → 0.0.150

package/dist/chains/question_answering/load.cjs

@@ -42,9 +42,13 @@ exports.loadQAStuffChain = loadQAStuffChain;
  * @returns A MapReduceQAChain instance.
  */
 function loadQAMapReduceChain(llm, params = {}) {
-    const { combineMapPrompt = map_reduce_prompts_js_1.COMBINE_QA_PROMPT_SELECTOR.getPrompt(llm), combinePrompt = map_reduce_prompts_js_1.COMBINE_PROMPT_SELECTOR.getPrompt(llm), verbose, returnIntermediateSteps, } = params;
+    const { combineMapPrompt = map_reduce_prompts_js_1.COMBINE_QA_PROMPT_SELECTOR.getPrompt(llm), combinePrompt = map_reduce_prompts_js_1.COMBINE_PROMPT_SELECTOR.getPrompt(llm), verbose, combineLLM, returnIntermediateSteps, } = params;
     const llmChain = new llm_chain_js_1.LLMChain({ prompt: combineMapPrompt, llm, verbose });
-    const combineLLMChain = new llm_chain_js_1.LLMChain({ prompt: combinePrompt, llm, verbose });
+    const combineLLMChain = new llm_chain_js_1.LLMChain({
+        prompt: combinePrompt,
+        llm: combineLLM ?? llm,
+        verbose,
+    });
     const combineDocumentChain = new combine_docs_chain_js_1.StuffDocumentsChain({
         llmChain: combineLLMChain,
         documentVariableName: "summaries",
@@ -67,9 +71,13 @@ exports.loadQAMapReduceChain = loadQAMapReduceChain;
  * @returns A RefineQAChain instance.
  */
 function loadQARefineChain(llm, params = {}) {
-    const { questionPrompt = refine_prompts_js_1.QUESTION_PROMPT_SELECTOR.getPrompt(llm), refinePrompt = refine_prompts_js_1.REFINE_PROMPT_SELECTOR.getPrompt(llm), verbose, } = params;
+    const { questionPrompt = refine_prompts_js_1.QUESTION_PROMPT_SELECTOR.getPrompt(llm), refinePrompt = refine_prompts_js_1.REFINE_PROMPT_SELECTOR.getPrompt(llm), refineLLM, verbose, } = params;
     const llmChain = new llm_chain_js_1.LLMChain({ prompt: questionPrompt, llm, verbose });
-    const refineLLMChain = new llm_chain_js_1.LLMChain({ prompt: refinePrompt, llm, verbose });
+    const refineLLMChain = new llm_chain_js_1.LLMChain({
+        prompt: refinePrompt,
+        llm: refineLLM ?? llm,
+        verbose,
+    });
     const chain = new combine_docs_chain_js_1.RefineDocumentsChain({
         llmChain,
         refineLLMChain,

package/dist/chains/question_answering/load.d.ts

@@ -35,6 +35,7 @@ export interface MapReduceQAChainParams {
     returnIntermediateSteps?: MapReduceDocumentsChainInput["returnIntermediateSteps"];
     combineMapPrompt?: BasePromptTemplate;
     combinePrompt?: BasePromptTemplate;
+    combineLLM?: BaseLanguageModel;
     verbose?: boolean;
 }
 /**
@@ -51,6 +52,7 @@ export declare function loadQAMapReduceChain(llm: BaseLanguageModel, params?: Ma
 export interface RefineQAChainParams {
     questionPrompt?: BasePromptTemplate;
     refinePrompt?: BasePromptTemplate;
+    refineLLM?: BaseLanguageModel;
     verbose?: boolean;
 }
 /**

package/dist/chains/question_answering/load.js

@@ -37,9 +37,13 @@ export function loadQAStuffChain(llm, params = {}) {
  * @returns A MapReduceQAChain instance.
  */
 export function loadQAMapReduceChain(llm, params = {}) {
-    const { combineMapPrompt = COMBINE_QA_PROMPT_SELECTOR.getPrompt(llm), combinePrompt = COMBINE_PROMPT_SELECTOR.getPrompt(llm), verbose, returnIntermediateSteps, } = params;
+    const { combineMapPrompt = COMBINE_QA_PROMPT_SELECTOR.getPrompt(llm), combinePrompt = COMBINE_PROMPT_SELECTOR.getPrompt(llm), verbose, combineLLM, returnIntermediateSteps, } = params;
     const llmChain = new LLMChain({ prompt: combineMapPrompt, llm, verbose });
-    const combineLLMChain = new LLMChain({ prompt: combinePrompt, llm, verbose });
+    const combineLLMChain = new LLMChain({
+        prompt: combinePrompt,
+        llm: combineLLM ?? llm,
+        verbose,
+    });
     const combineDocumentChain = new StuffDocumentsChain({
         llmChain: combineLLMChain,
         documentVariableName: "summaries",
@@ -61,9 +65,13 @@ export function loadQAMapReduceChain(llm, params = {}) {
  * @returns A RefineQAChain instance.
  */
 export function loadQARefineChain(llm, params = {}) {
-    const { questionPrompt = QUESTION_PROMPT_SELECTOR.getPrompt(llm), refinePrompt = REFINE_PROMPT_SELECTOR.getPrompt(llm), verbose, } = params;
+    const { questionPrompt = QUESTION_PROMPT_SELECTOR.getPrompt(llm), refinePrompt = REFINE_PROMPT_SELECTOR.getPrompt(llm), refineLLM, verbose, } = params;
     const llmChain = new LLMChain({ prompt: questionPrompt, llm, verbose });
-    const refineLLMChain = new LLMChain({ prompt: refinePrompt, llm, verbose });
+    const refineLLMChain = new LLMChain({
+        prompt: refinePrompt,
+        llm: refineLLM ?? llm,
+        verbose,
+    });
     const chain = new RefineDocumentsChain({
         llmChain,
         refineLLMChain,

package/dist/chains/summarization/load.cjs

@@ -18,11 +18,11 @@ const loadSummarizationChain = (llm, params = { type: "map_reduce" }) => {
         return chain;
     }
     if (params.type === "map_reduce") {
-        const { combineMapPrompt = stuff_prompts_js_1.DEFAULT_PROMPT, combinePrompt = stuff_prompts_js_1.DEFAULT_PROMPT, returnIntermediateSteps, } = params;
+        const { combineMapPrompt = stuff_prompts_js_1.DEFAULT_PROMPT, combinePrompt = stuff_prompts_js_1.DEFAULT_PROMPT, combineLLM, returnIntermediateSteps, } = params;
         const llmChain = new llm_chain_js_1.LLMChain({ prompt: combineMapPrompt, llm, verbose });
         const combineLLMChain = new llm_chain_js_1.LLMChain({
             prompt: combinePrompt,
-            llm,
+            llm: combineLLM ?? llm,
             verbose,
         });
         const combineDocumentChain = new combine_docs_chain_js_1.StuffDocumentsChain({
@@ -40,9 +40,13 @@ const loadSummarizationChain = (llm, params = { type: "map_reduce" }) => {
         return chain;
     }
     if (params.type === "refine") {
-        const { refinePrompt = refine_prompts_js_1.REFINE_PROMPT, questionPrompt = stuff_prompts_js_1.DEFAULT_PROMPT } = params;
+        const { refinePrompt = refine_prompts_js_1.REFINE_PROMPT, refineLLM, questionPrompt = stuff_prompts_js_1.DEFAULT_PROMPT, } = params;
         const llmChain = new llm_chain_js_1.LLMChain({ prompt: questionPrompt, llm, verbose });
-        const refineLLMChain = new llm_chain_js_1.LLMChain({ prompt: refinePrompt, llm, verbose });
+        const refineLLMChain = new llm_chain_js_1.LLMChain({
+            prompt: refinePrompt,
+            llm: refineLLM ?? llm,
+            verbose,
+        });
         const chain = new combine_docs_chain_js_1.RefineDocumentsChain({
             llmChain,
             refineLLMChain,

package/dist/chains/summarization/load.d.ts

@@ -16,9 +16,11 @@ export type SummarizationChainParams = BaseParams & ({
     type?: "map_reduce";
     combineMapPrompt?: BasePromptTemplate;
     combinePrompt?: BasePromptTemplate;
+    combineLLM?: BaseLanguageModel;
 } & Pick<MapReduceDocumentsChainInput, "returnIntermediateSteps">) | {
     type?: "refine";
     refinePrompt?: BasePromptTemplate;
+    refineLLM?: BaseLanguageModel;
     questionPrompt?: BasePromptTemplate;
 });
 export declare const loadSummarizationChain: (llm: BaseLanguageModel, params?: SummarizationChainParams) => StuffDocumentsChain | MapReduceDocumentsChain | RefineDocumentsChain;

package/dist/chains/summarization/load.js

@@ -15,11 +15,11 @@ export const loadSummarizationChain = (llm, params = { type: "map_reduce" }) =>
         return chain;
     }
     if (params.type === "map_reduce") {
-        const { combineMapPrompt = DEFAULT_PROMPT, combinePrompt = DEFAULT_PROMPT, returnIntermediateSteps, } = params;
+        const { combineMapPrompt = DEFAULT_PROMPT, combinePrompt = DEFAULT_PROMPT, combineLLM, returnIntermediateSteps, } = params;
         const llmChain = new LLMChain({ prompt: combineMapPrompt, llm, verbose });
         const combineLLMChain = new LLMChain({
             prompt: combinePrompt,
-            llm,
+            llm: combineLLM ?? llm,
             verbose,
         });
         const combineDocumentChain = new StuffDocumentsChain({
@@ -37,9 +37,13 @@ export const loadSummarizationChain = (llm, params = { type: "map_reduce" }) =>
         return chain;
     }
     if (params.type === "refine") {
-        const { refinePrompt = REFINE_PROMPT, questionPrompt = DEFAULT_PROMPT } = params;
+        const { refinePrompt = REFINE_PROMPT, refineLLM, questionPrompt = DEFAULT_PROMPT, } = params;
         const llmChain = new LLMChain({ prompt: questionPrompt, llm, verbose });
-        const refineLLMChain = new LLMChain({ prompt: refinePrompt, llm, verbose });
+        const refineLLMChain = new LLMChain({
+            prompt: refinePrompt,
+            llm: refineLLM ?? llm,
+            verbose,
+        });
         const chain = new RefineDocumentsChain({
             llmChain,
             refineLLMChain,

package/dist/llms/bedrock.cjs

@@ -112,6 +112,12 @@ class Bedrock extends base_js_1.LLM {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "endpointUrl", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "codec", {
             enumerable: true,
             configurable: true,
@@ -132,6 +138,7 @@ class Bedrock extends base_js_1.LLM {
         this.temperature = fields?.temperature ?? this.temperature;
         this.maxTokens = fields?.maxTokens ?? this.maxTokens;
         this.fetchFn = fields?.fetchFn ?? fetch;
+        this.endpointUrl = fields?.endpointUrl;
     }
     /** Call out to Bedrock service model.
       Arguments:
@@ -154,7 +161,8 @@ class Bedrock extends base_js_1.LLM {
         const provider = this.model.split(".")[0];
         const service = "bedrock";
         const inputBody = BedrockLLMInputOutputAdapter.prepareInput(provider, prompt, this.maxTokens, this.temperature);
-        const url = new URL(`https://${service}.${this.region}.amazonaws.com/model/${this.model}/invoke-with-response-stream`);
+        const endpointUrl = this.endpointUrl ?? `${service}.${this.region}.amazonaws.com`;
+        const url = new URL(`https://${endpointUrl}/model/${this.model}/invoke-with-response-stream`);
         const request = new protocol_http_1.HttpRequest({
             hostname: url.hostname,
             path: url.pathname,

package/dist/llms/bedrock.d.ts

@@ -29,6 +29,8 @@ export interface BedrockInput {
     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
@@ -46,6 +48,7 @@ export declare class Bedrock extends LLM implements BedrockInput {
     temperature?: number | undefined;
     maxTokens?: number | undefined;
     fetchFn: typeof fetch;
+    endpointUrl?: string;
     codec: EventStreamCodec;
     get lc_secrets(): {
         [key: string]: string;

package/dist/llms/bedrock.js

@@ -109,6 +109,12 @@ export class Bedrock extends LLM {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "endpointUrl", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "codec", {
             enumerable: true,
             configurable: true,
@@ -129,6 +135,7 @@ 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;
     }
     /** Call out to Bedrock service model.
       Arguments:
@@ -151,7 +158,8 @@ export class Bedrock extends LLM {
         const provider = this.model.split(".")[0];
         const service = "bedrock";
         const inputBody = BedrockLLMInputOutputAdapter.prepareInput(provider, prompt, this.maxTokens, this.temperature);
-        const url = new URL(`https://${service}.${this.region}.amazonaws.com/model/${this.model}/invoke-with-response-stream`);
+        const endpointUrl = this.endpointUrl ?? `${service}.${this.region}.amazonaws.com`;
+        const url = new URL(`https://${endpointUrl}/model/${this.model}/invoke-with-response-stream`);
         const request = new HttpRequest({
             hostname: url.hostname,
             path: url.pathname,

package/dist/llms/replicate.cjs

@@ -44,6 +44,12 @@ class Replicate extends base_js_1.LLM {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "promptKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         const apiKey = fields?.apiKey ??
             (0, env_js_1.getEnvironmentVariable)("REPLICATE_API_KEY") ?? // previous environment variable for backwards compatibility
             (0, env_js_1.getEnvironmentVariable)("REPLICATE_API_TOKEN"); // current environment variable, matching the Python library
@@ -53,6 +59,7 @@ class Replicate extends base_js_1.LLM {
         this.apiKey = apiKey;
         this.model = fields.model;
         this.input = fields.input ?? {};
+        this.promptKey = fields.promptKey;
     }
     _llmType() {
         return "replicate";
@@ -64,11 +71,30 @@ class Replicate extends base_js_1.LLM {
             userAgent: "langchain",
             auth: this.apiKey,
         });
+        if (this.promptKey === undefined) {
+            const [modelString, versionString] = this.model.split(":");
+            const version = await replicate.models.versions.get(modelString.split("/")[0], modelString.split("/")[1], versionString);
+            const openapiSchema = version.openapi_schema;
+            const inputProperties = 
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            openapiSchema?.components?.schemas?.Input?.properties;
+            if (inputProperties === undefined) {
+                this.promptKey = "prompt";
+            }
+            else {
+                const sortedInputProperties = Object.entries(inputProperties).sort(([_keyA, valueA], [_keyB, valueB]) => {
+                    const orderA = valueA["x-order"] || 0;
+                    const orderB = valueB["x-order"] || 0;
+                    return orderA - orderB;
+                });
+                this.promptKey = sortedInputProperties[0][0] ?? "prompt";
+            }
+        }
         const output = await this.caller.callWithOptions({ signal: options.signal }, () => replicate.run(this.model, {
-            wait: true,
             input: {
+                // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+                [this.promptKey]: prompt,
                 ...this.input,
-                prompt,
             },
         }));
         if (typeof output === "string") {

package/dist/llms/replicate.d.ts

@@ -10,6 +10,8 @@ export interface ReplicateInput {
         [key: string]: string | number | boolean;
     };
     apiKey?: string;
+    /** The key used to pass prompts to the model. */
+    promptKey?: string;
 }
 /**
  * Class responsible for managing the interaction with the Replicate API.
@@ -26,6 +28,7 @@ export declare class Replicate extends LLM implements ReplicateInput {
     model: ReplicateInput["model"];
     input: ReplicateInput["input"];
     apiKey: string;
+    promptKey?: string;
     constructor(fields: ReplicateInput & BaseLLMParams);
     _llmType(): string;
     /** @ignore */

package/dist/llms/replicate.js

@@ -41,6 +41,12 @@ export class Replicate extends LLM {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "promptKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         const apiKey = fields?.apiKey ??
             getEnvironmentVariable("REPLICATE_API_KEY") ?? // previous environment variable for backwards compatibility
             getEnvironmentVariable("REPLICATE_API_TOKEN"); // current environment variable, matching the Python library
@@ -50,6 +56,7 @@ export class Replicate extends LLM {
         this.apiKey = apiKey;
         this.model = fields.model;
         this.input = fields.input ?? {};
+        this.promptKey = fields.promptKey;
     }
     _llmType() {
         return "replicate";
@@ -61,11 +68,30 @@ export class Replicate extends LLM {
             userAgent: "langchain",
             auth: this.apiKey,
         });
+        if (this.promptKey === undefined) {
+            const [modelString, versionString] = this.model.split(":");
+            const version = await replicate.models.versions.get(modelString.split("/")[0], modelString.split("/")[1], versionString);
+            const openapiSchema = version.openapi_schema;
+            const inputProperties = 
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            openapiSchema?.components?.schemas?.Input?.properties;
+            if (inputProperties === undefined) {
+                this.promptKey = "prompt";
+            }
+            else {
+                const sortedInputProperties = Object.entries(inputProperties).sort(([_keyA, valueA], [_keyB, valueB]) => {
+                    const orderA = valueA["x-order"] || 0;
+                    const orderB = valueB["x-order"] || 0;
+                    return orderA - orderB;
+                });
+                this.promptKey = sortedInputProperties[0][0] ?? "prompt";
+            }
+        }
         const output = await this.caller.callWithOptions({ signal: options.signal }, () => replicate.run(this.model, {
-            wait: true,
             input: {
+                // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+                [this.promptKey]: prompt,
                 ...this.input,
-                prompt,
             },
         }));
         if (typeof output === "string") {

package/dist/prompts/prompt.cjs

@@ -99,6 +99,8 @@ class PromptTemplate extends base_js_1.BaseStringPromptTemplate {
             }
         });
         return new PromptTemplate({
+            // Rely on extracted types
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
             inputVariables: [...names],
             templateFormat,
             template,

package/dist/prompts/prompt.d.ts

@@ -32,7 +32,7 @@ type NonAlphanumeric = " " | "\t" | "\n" | "\r" | '"' | "'" | "{" | "[" | "(" |
  */
 type ExtractTemplateParamsRecursive<T extends string, Result extends string[] = []> = T extends `${string}{${infer Param}}${infer Rest}` ? Param extends `${NonAlphanumeric}${string}` ? ExtractTemplateParamsRecursive<Rest, Result> : ExtractTemplateParamsRecursive<Rest, [...Result, Param]> : Result;
 export type ParamsFromFString<T extends string> = {
-    [Key in ExtractTemplateParamsRecursive<T>[number]]: string;
+    [Key in ExtractTemplateParamsRecursive<T>[number] | (string & Record<never, never>)]: string;
 };
 /**
  * Schema to represent a basic prompt for an LLM.

package/dist/prompts/prompt.js

@@ -96,6 +96,8 @@ export class PromptTemplate extends BaseStringPromptTemplate {
             }
         });
         return new PromptTemplate({
+            // Rely on extracted types
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
             inputVariables: [...names],
             templateFormat,
             template,

package/dist/retrievers/self_query/base.cjs

@@ -84,7 +84,7 @@ class BasicTranslator extends BaseTranslator {
     visitComparison(comparison) {
         return {
             [comparison.attribute]: {
-                [this.formatFunction(comparison.comparator)]: comparison.value,
+                [this.formatFunction(comparison.comparator)]: (0, utils_js_1.castValue)(comparison.value),
             },
         };
     }

package/dist/retrievers/self_query/base.js

@@ -1,5 +1,5 @@
 import { Comparators, Operators, Visitor, } from "../../chains/query_constructor/ir.js";
-import { isFilterEmpty } from "./utils.js";
+import { isFilterEmpty, castValue } from "./utils.js";
 /**
  * Abstract class that provides a blueprint for creating specific
  * translator classes. Defines two abstract methods: formatFunction and
@@ -80,7 +80,7 @@ export class BasicTranslator extends BaseTranslator {
     visitComparison(comparison) {
         return {
             [comparison.attribute]: {
-                [this.formatFunction(comparison.comparator)]: comparison.value,
+                [this.formatFunction(comparison.comparator)]: castValue(comparison.value),
             },
         };
     }

package/dist/retrievers/self_query/functional.cjs

@@ -132,7 +132,7 @@ class FunctionalTranslator extends base_js_1.BaseTranslator {
                     }
                     return false;
                 }
-                return comparatorFunction(documentValue, value);
+                return comparatorFunction(documentValue, (0, utils_js_1.castValue)(value));
             };
         }
         else {

package/dist/retrievers/self_query/functional.js

@@ -1,6 +1,6 @@
 import { Comparators, Operators, } from "../../chains/query_constructor/ir.js";
 import { BaseTranslator } from "./base.js";
-import { isFilterEmpty } from "./utils.js";
+import { castValue, isFilterEmpty } from "./utils.js";
 /**
  * A class that extends `BaseTranslator` to translate structured queries
  * into functional filters.
@@ -129,7 +129,7 @@ export class FunctionalTranslator extends BaseTranslator {
                     }
                     return false;
                 }
-                return comparatorFunction(documentValue, value);
+                return comparatorFunction(documentValue, castValue(value));
             };
         }
         else {

package/dist/retrievers/self_query/utils.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.isString = exports.isFloat = exports.isInt = exports.isFilterEmpty = exports.isObject = void 0;
+exports.castValue = exports.isString = exports.isFloat = exports.isInt = exports.isFilterEmpty = exports.isObject = void 0;
 /**
  * Checks if the provided argument is an object and not an array.
  */
@@ -32,16 +32,32 @@ exports.isFilterEmpty = isFilterEmpty;
  * Checks if the provided value is an integer.
  */
 function isInt(value) {
-    const numberValue = parseFloat(value);
-    return !Number.isNaN(numberValue) && numberValue % 1 === 0;
+    if (typeof value === "number") {
+        return value % 1 === 0;
+    }
+    else if (typeof value === "string") {
+        const numberValue = parseInt(value, 10);
+        return (!Number.isNaN(numberValue) &&
+            numberValue % 1 === 0 &&
+            numberValue.toString() === value);
+    }
+    return false;
 }
 exports.isInt = isInt;
 /**
  * Checks if the provided value is a floating-point number.
  */
 function isFloat(value) {
-    const numberValue = parseFloat(value);
-    return !Number.isNaN(numberValue) && numberValue % 1 !== 0;
+    if (typeof value === "number") {
+        return value % 1 !== 0;
+    }
+    else if (typeof value === "string") {
+        const numberValue = parseFloat(value);
+        return (!Number.isNaN(numberValue) &&
+            numberValue % 1 !== 0 &&
+            numberValue.toString() === value);
+    }
+    return false;
 }
 exports.isFloat = isFloat;
 /**
@@ -49,6 +65,30 @@ exports.isFloat = isFloat;
  * number.
  */
 function isString(value) {
-    return typeof value === "string" && Number.isNaN(parseFloat(value));
+    return (typeof value === "string" &&
+        (Number.isNaN(parseFloat(value)) || parseFloat(value).toString() !== value));
 }
 exports.isString = isString;
+/**
+ * Casts a value that might be string or number to actual string or number.
+ * Since LLM might return back an integer/float as a string, we need to cast
+ * it back to a number, as many vector databases can't handle number as string
+ * values as a comparator.
+ */
+function castValue(input) {
+    let value;
+    if (isString(input)) {
+        value = input;
+    }
+    else if (isInt(input)) {
+        value = parseInt(input, 10);
+    }
+    else if (isFloat(input)) {
+        value = parseFloat(input);
+    }
+    else {
+        throw new Error("Unsupported value type");
+    }
+    return value;
+}
+exports.castValue = castValue;

package/dist/retrievers/self_query/utils.d.ts

@@ -20,3 +20,10 @@ export declare function isFloat(value: unknown): boolean;
  * number.
  */
 export declare function isString(value: unknown): boolean;
+/**
+ * Casts a value that might be string or number to actual string or number.
+ * Since LLM might return back an integer/float as a string, we need to cast
+ * it back to a number, as many vector databases can't handle number as string
+ * values as a comparator.
+ */
+export declare function castValue(input: unknown): string | number;

package/dist/retrievers/self_query/utils.js

@@ -27,20 +27,59 @@ filter) {
  * Checks if the provided value is an integer.
  */
 export function isInt(value) {
-    const numberValue = parseFloat(value);
-    return !Number.isNaN(numberValue) && numberValue % 1 === 0;
+    if (typeof value === "number") {
+        return value % 1 === 0;
+    }
+    else if (typeof value === "string") {
+        const numberValue = parseInt(value, 10);
+        return (!Number.isNaN(numberValue) &&
+            numberValue % 1 === 0 &&
+            numberValue.toString() === value);
+    }
+    return false;
 }
 /**
  * Checks if the provided value is a floating-point number.
  */
 export function isFloat(value) {
-    const numberValue = parseFloat(value);
-    return !Number.isNaN(numberValue) && numberValue % 1 !== 0;
+    if (typeof value === "number") {
+        return value % 1 !== 0;
+    }
+    else if (typeof value === "string") {
+        const numberValue = parseFloat(value);
+        return (!Number.isNaN(numberValue) &&
+            numberValue % 1 !== 0 &&
+            numberValue.toString() === value);
+    }
+    return false;
 }
 /**
  * Checks if the provided value is a string that cannot be parsed into a
  * number.
  */
 export function isString(value) {
-    return typeof value === "string" && Number.isNaN(parseFloat(value));
+    return (typeof value === "string" &&
+        (Number.isNaN(parseFloat(value)) || parseFloat(value).toString() !== value));
+}
+/**
+ * Casts a value that might be string or number to actual string or number.
+ * Since LLM might return back an integer/float as a string, we need to cast
+ * it back to a number, as many vector databases can't handle number as string
+ * values as a comparator.
+ */
+export function castValue(input) {
+    let value;
+    if (isString(input)) {
+        value = input;
+    }
+    else if (isInt(input)) {
+        value = parseInt(input, 10);
+    }
+    else if (isFloat(input)) {
+        value = parseFloat(input);
+    }
+    else {
+        throw new Error("Unsupported value type");
+    }
+    return value;
 }