langchain 0.0.76 → 0.0.78

package/dist/agents/structured_chat/outputParser.cjs

@@ -0,0 +1,92 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.StructuredChatOutputParserWithRetries = exports.StructuredChatOutputParser = void 0;
+const types_js_1 = require("../types.cjs");
+const prompt_js_1 = require("./prompt.cjs");
+const fix_js_1 = require("../../output_parsers/fix.cjs");
+const output_parser_js_1 = require("../../schema/output_parser.cjs");
+const index_js_1 = require("../../prompts/index.cjs");
+class StructuredChatOutputParser extends types_js_1.AgentActionOutputParser {
+    constructor(toolNames) {
+        super();
+        Object.defineProperty(this, "toolNames", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: toolNames
+        });
+    }
+    async parse(text) {
+        try {
+            const regex = /```(?:json)?(.*)(```)/gs;
+            const actionMatch = regex.exec(text);
+            if (actionMatch === null) {
+                throw new output_parser_js_1.OutputParserException(`Could not parse an action. The agent action must be within a markdown code block, and "action" must be a provided tool or "Final Answer"`);
+            }
+            const response = JSON.parse(actionMatch[1].trim());
+            const { action, action_input } = response;
+            if (action === "Final Answer") {
+                return { returnValues: { output: action_input }, log: text };
+            }
+            return { tool: action, toolInput: action_input || {}, log: text };
+        }
+        catch (e) {
+            throw new output_parser_js_1.OutputParserException(`Failed to parse. Text: "${text}". Error: ${e}`);
+        }
+    }
+    getFormatInstructions() {
+        return (0, index_js_1.renderTemplate)(prompt_js_1.AGENT_ACTION_FORMAT_INSTRUCTIONS, "f-string", {
+            tool_names: this.toolNames.join(", "),
+        });
+    }
+}
+exports.StructuredChatOutputParser = StructuredChatOutputParser;
+class StructuredChatOutputParserWithRetries extends types_js_1.AgentActionOutputParser {
+    constructor(fields) {
+        super();
+        Object.defineProperty(this, "baseParser", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "outputFixingParser", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "toolNames", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: []
+        });
+        this.toolNames = fields.toolNames ?? this.toolNames;
+        this.baseParser =
+            fields?.baseParser ?? new StructuredChatOutputParser(this.toolNames);
+        this.outputFixingParser = fields?.outputFixingParser;
+    }
+    async parse(text) {
+        if (this.outputFixingParser !== undefined) {
+            return this.outputFixingParser.parse(text);
+        }
+        return this.baseParser.parse(text);
+    }
+    getFormatInstructions() {
+        return (0, index_js_1.renderTemplate)(prompt_js_1.FORMAT_INSTRUCTIONS, "f-string", {
+            tool_names: this.toolNames.join(", "),
+        });
+    }
+    static fromLLM(llm, options) {
+        const baseParser = options.baseParser ??
+            new StructuredChatOutputParser(options.toolNames ?? []);
+        const outputFixingParser = fix_js_1.OutputFixingParser.fromLLM(llm, baseParser);
+        return new StructuredChatOutputParserWithRetries({
+            baseParser,
+            outputFixingParser,
+            toolNames: options.toolNames,
+        });
+    }
+}
+exports.StructuredChatOutputParserWithRetries = StructuredChatOutputParserWithRetries;

package/dist/agents/structured_chat/outputParser.d.ts

@@ -0,0 +1,24 @@
+import { AgentActionOutputParser } from "../types.js";
+import { OutputFixingParser } from "../../output_parsers/fix.js";
+import { BaseLanguageModel } from "../../base_language/index.js";
+import { AgentAction, AgentFinish } from "../../schema/index.js";
+export declare class StructuredChatOutputParser extends AgentActionOutputParser {
+    private toolNames;
+    constructor(toolNames: string[]);
+    parse(text: string): Promise<AgentAction | AgentFinish>;
+    getFormatInstructions(): string;
+}
+export interface StructuredChatOutputParserArgs {
+    baseParser?: StructuredChatOutputParser;
+    outputFixingParser?: OutputFixingParser<AgentAction | AgentFinish>;
+    toolNames?: string[];
+}
+export declare class StructuredChatOutputParserWithRetries extends AgentActionOutputParser {
+    private baseParser;
+    private outputFixingParser?;
+    private toolNames;
+    constructor(fields: StructuredChatOutputParserArgs);
+    parse(text: string): Promise<AgentAction | AgentFinish>;
+    getFormatInstructions(): string;
+    static fromLLM(llm: BaseLanguageModel, options: Omit<StructuredChatOutputParserArgs, "outputFixingParser">): StructuredChatOutputParserWithRetries;
+}

package/dist/agents/structured_chat/outputParser.js

@@ -0,0 +1,87 @@
+import { AgentActionOutputParser } from "../types.js";
+import { AGENT_ACTION_FORMAT_INSTRUCTIONS, FORMAT_INSTRUCTIONS, } from "./prompt.js";
+import { OutputFixingParser } from "../../output_parsers/fix.js";
+import { OutputParserException } from "../../schema/output_parser.js";
+import { renderTemplate } from "../../prompts/index.js";
+export class StructuredChatOutputParser extends AgentActionOutputParser {
+    constructor(toolNames) {
+        super();
+        Object.defineProperty(this, "toolNames", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: toolNames
+        });
+    }
+    async parse(text) {
+        try {
+            const regex = /```(?:json)?(.*)(```)/gs;
+            const actionMatch = regex.exec(text);
+            if (actionMatch === null) {
+                throw new OutputParserException(`Could not parse an action. The agent action must be within a markdown code block, and "action" must be a provided tool or "Final Answer"`);
+            }
+            const response = JSON.parse(actionMatch[1].trim());
+            const { action, action_input } = response;
+            if (action === "Final Answer") {
+                return { returnValues: { output: action_input }, log: text };
+            }
+            return { tool: action, toolInput: action_input || {}, log: text };
+        }
+        catch (e) {
+            throw new OutputParserException(`Failed to parse. Text: "${text}". Error: ${e}`);
+        }
+    }
+    getFormatInstructions() {
+        return renderTemplate(AGENT_ACTION_FORMAT_INSTRUCTIONS, "f-string", {
+            tool_names: this.toolNames.join(", "),
+        });
+    }
+}
+export class StructuredChatOutputParserWithRetries extends AgentActionOutputParser {
+    constructor(fields) {
+        super();
+        Object.defineProperty(this, "baseParser", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "outputFixingParser", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "toolNames", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: []
+        });
+        this.toolNames = fields.toolNames ?? this.toolNames;
+        this.baseParser =
+            fields?.baseParser ?? new StructuredChatOutputParser(this.toolNames);
+        this.outputFixingParser = fields?.outputFixingParser;
+    }
+    async parse(text) {
+        if (this.outputFixingParser !== undefined) {
+            return this.outputFixingParser.parse(text);
+        }
+        return this.baseParser.parse(text);
+    }
+    getFormatInstructions() {
+        return renderTemplate(FORMAT_INSTRUCTIONS, "f-string", {
+            tool_names: this.toolNames.join(", "),
+        });
+    }
+    static fromLLM(llm, options) {
+        const baseParser = options.baseParser ??
+            new StructuredChatOutputParser(options.toolNames ?? []);
+        const outputFixingParser = OutputFixingParser.fromLLM(llm, baseParser);
+        return new StructuredChatOutputParserWithRetries({
+            baseParser,
+            outputFixingParser,
+            toolNames: options.toolNames,
+        });
+    }
+}

package/dist/agents/structured_chat/prompt.cjs

@@ -0,0 +1,62 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SUFFIX = exports.FORMAT_INSTRUCTIONS = exports.AGENT_ACTION_FORMAT_INSTRUCTIONS = exports.PREFIX = void 0;
+exports.PREFIX = `Answer the following questions truthfully and as best you can.`;
+exports.AGENT_ACTION_FORMAT_INSTRUCTIONS = `Output a JSON markdown code snippet containing a valid JSON blob (denoted below by $JSON_BLOB).
+This $JSON_BLOB must have a "action" key (with the name of the tool to use) and an "action_input" key (tool input).
+
+Valid "action" values: "Final Answer" (which you must use when giving your final response to the user), or one of [{tool_names}].
+
+The $JSON_BLOB must be valid, parseable JSON and only contain a SINGLE action. Here is an example of an acceptable output:
+
+\`\`\`json
+{{
+  "action": $TOOL_NAME
+  "action_input": $INPUT
+}}
+\`\`\`
+
+Remember to include the surrounding markdown code snippet delimiters (begin with "\`\`\`" json and close with "\`\`\`")!
+`;
+exports.FORMAT_INSTRUCTIONS = `You have access to the following tools.
+You must format your inputs to these tools to match their "JSON schema" definitions below.
+
+"JSON Schema" is a declarative language that allows you to annotate and validate JSON documents.
+
+For example, the example "JSON Schema" instance {{"properties": {{"foo": {{"description": "a list of test words", "type": "array", "items": {{"type": "string"}}}}}}, "required": ["foo"]}}}}
+would match an object with one required property, "foo". The "type" property specifies "foo" must be an "array", and the "description" property semantically describes it as "a list of test words". The items within "foo" must be strings.
+Thus, the object {{"foo": ["bar", "baz"]}} is a well-formatted instance of this example "JSON Schema". The object {{"properties": {{"foo": ["bar", "baz"]}}}} is not well-formatted.
+
+Here are the JSON Schema instances for the tools you have access to:
+
+{tool_schemas}
+
+The way you use the tools is as follows:
+
+------------------------
+
+${exports.AGENT_ACTION_FORMAT_INSTRUCTIONS}
+
+If you are using a tool, "action_input" must adhere to the tool's input schema, given above.
+
+------------------------
+
+ALWAYS use the following format:
+
+Question: the input question you must answer
+Thought: you should always think about what to do
+Action:
+\`\`\`json
+$JSON_BLOB
+\`\`\`
+Observation: the result of the action
+... (this Thought/Action/Observation can repeat N times)
+Thought: I now know the final answer
+Action:
+\`\`\`json
+{{
+  "action": "Final Answer",
+  "action_input": "Final response to human"
+}}
+\`\`\``;
+exports.SUFFIX = `Begin! Reminder to ALWAYS use the above format, and to use tools if appropriate.`;

package/dist/agents/structured_chat/prompt.d.ts

@@ -0,0 +1,4 @@
+export declare const PREFIX = "Answer the following questions truthfully and as best you can.";
+export declare const AGENT_ACTION_FORMAT_INSTRUCTIONS = "Output a JSON markdown code snippet containing a valid JSON blob (denoted below by $JSON_BLOB).\nThis $JSON_BLOB must have a \"action\" key (with the name of the tool to use) and an \"action_input\" key (tool input).\n\nValid \"action\" values: \"Final Answer\" (which you must use when giving your final response to the user), or one of [{tool_names}].\n\nThe $JSON_BLOB must be valid, parseable JSON and only contain a SINGLE action. Here is an example of an acceptable output:\n\n```json\n{{\n  \"action\": $TOOL_NAME\n  \"action_input\": $INPUT\n}}\n```\n\nRemember to include the surrounding markdown code snippet delimiters (begin with \"```\" json and close with \"```\")!\n";
+export declare const FORMAT_INSTRUCTIONS: string;
+export declare const SUFFIX = "Begin! Reminder to ALWAYS use the above format, and to use tools if appropriate.";

package/dist/agents/structured_chat/prompt.js

@@ -0,0 +1,59 @@
+export const PREFIX = `Answer the following questions truthfully and as best you can.`;
+export const AGENT_ACTION_FORMAT_INSTRUCTIONS = `Output a JSON markdown code snippet containing a valid JSON blob (denoted below by $JSON_BLOB).
+This $JSON_BLOB must have a "action" key (with the name of the tool to use) and an "action_input" key (tool input).
+
+Valid "action" values: "Final Answer" (which you must use when giving your final response to the user), or one of [{tool_names}].
+
+The $JSON_BLOB must be valid, parseable JSON and only contain a SINGLE action. Here is an example of an acceptable output:
+
+\`\`\`json
+{{
+  "action": $TOOL_NAME
+  "action_input": $INPUT
+}}
+\`\`\`
+
+Remember to include the surrounding markdown code snippet delimiters (begin with "\`\`\`" json and close with "\`\`\`")!
+`;
+export const FORMAT_INSTRUCTIONS = `You have access to the following tools.
+You must format your inputs to these tools to match their "JSON schema" definitions below.
+
+"JSON Schema" is a declarative language that allows you to annotate and validate JSON documents.
+
+For example, the example "JSON Schema" instance {{"properties": {{"foo": {{"description": "a list of test words", "type": "array", "items": {{"type": "string"}}}}}}, "required": ["foo"]}}}}
+would match an object with one required property, "foo". The "type" property specifies "foo" must be an "array", and the "description" property semantically describes it as "a list of test words". The items within "foo" must be strings.
+Thus, the object {{"foo": ["bar", "baz"]}} is a well-formatted instance of this example "JSON Schema". The object {{"properties": {{"foo": ["bar", "baz"]}}}} is not well-formatted.
+
+Here are the JSON Schema instances for the tools you have access to:
+
+{tool_schemas}
+
+The way you use the tools is as follows:
+
+------------------------
+
+${AGENT_ACTION_FORMAT_INSTRUCTIONS}
+
+If you are using a tool, "action_input" must adhere to the tool's input schema, given above.
+
+------------------------
+
+ALWAYS use the following format:
+
+Question: the input question you must answer
+Thought: you should always think about what to do
+Action:
+\`\`\`json
+$JSON_BLOB
+\`\`\`
+Observation: the result of the action
+... (this Thought/Action/Observation can repeat N times)
+Thought: I now know the final answer
+Action:
+\`\`\`json
+{{
+  "action": "Final Answer",
+  "action_input": "Final response to human"
+}}
+\`\`\``;
+export const SUFFIX = `Begin! Reminder to ALWAYS use the above format, and to use tools if appropriate.`;

package/dist/callbacks/handlers/tracer_langchain.cjs

@@ -1,10 +1,11 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.LangChainTracer = void 0;
+const async_caller_js_1 = require("../../util/async_caller.cjs");
 const env_js_1 = require("../../util/env.cjs");
 const tracer_js_1 = require("./tracer.cjs");
 class LangChainTracer extends tracer_js_1.BaseTracer {
-    constructor({ exampleId, tenantId, sessionName, sessionExtra, } = {}) {
+    constructor({ exampleId, tenantId, sessionName, sessionExtra, callerParams, } = {}) {
         super();
         Object.defineProperty(this, "name", {
             enumerable: true,
@@ -59,6 +60,12 @@ class LangChainTracer extends tracer_js_1.BaseTracer {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "caller", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         // eslint-disable-next-line no-process-env
         if (typeof process !== "undefined" && process.env?.LANGCHAIN_API_KEY) {
             // eslint-disable-next-line no-process-env
@@ -79,6 +86,7 @@ class LangChainTracer extends tracer_js_1.BaseTracer {
                 "default";
         this.sessionExtra = sessionExtra;
         this.exampleId = exampleId;
+        this.caller = new async_caller_js_1.AsyncCaller(callerParams ?? {});
     }
     async ensureSession() {
         if (this.session) {
@@ -86,7 +94,7 @@ class LangChainTracer extends tracer_js_1.BaseTracer {
         }
         const tenantId = await this.ensureTenantId();
         const endpoint = `${this.endpoint}/sessions?upsert=true`;
-        const res = await fetch(endpoint, {
+        const res = await this.caller.call(fetch, endpoint, {
             method: "POST",
             headers: this.headers,
             body: JSON.stringify({
@@ -108,7 +116,7 @@ class LangChainTracer extends tracer_js_1.BaseTracer {
             return this.tenantId;
         }
         const endpoint = `${this.endpoint}/tenants`;
-        const response = await fetch(endpoint, {
+        const response = await this.caller.call(fetch, endpoint, {
             method: "GET",
             headers: this.headers,
         });
@@ -149,7 +157,7 @@ class LangChainTracer extends tracer_js_1.BaseTracer {
     async persistRun(run) {
         const persistedRun = await this._convertToCreate(run, this.exampleId);
         const endpoint = `${this.endpoint}/runs`;
-        const response = await fetch(endpoint, {
+        const response = await this.caller.call(fetch, endpoint, {
             method: "POST",
             headers: this.headers,
             body: JSON.stringify(persistedRun),

package/dist/callbacks/handlers/tracer_langchain.d.ts

@@ -1,3 +1,4 @@
+import { AsyncCaller, AsyncCallerParams } from "../../util/async_caller.js";
 import { BaseTracer, Run, BaseRun } from "./tracer.js";
 export interface RunCreate extends BaseRun {
     child_runs: this[];
@@ -21,6 +22,7 @@ export interface LangChainTracerFields {
     tenantId?: string;
     sessionName?: string;
     sessionExtra?: Record<string, unknown>;
+    callerParams?: AsyncCallerParams;
 }
 export declare class LangChainTracer extends BaseTracer implements LangChainTracerFields {
     name: string;
@@ -31,7 +33,8 @@ export declare class LangChainTracer extends BaseTracer implements LangChainTrac
     protected session: TracerSession;
     exampleId?: string;
     tenantId?: string;
-    constructor({ exampleId, tenantId, sessionName, sessionExtra, }?: LangChainTracerFields);
+    caller: AsyncCaller;
+    constructor({ exampleId, tenantId, sessionName, sessionExtra, callerParams, }?: LangChainTracerFields);
     protected ensureSession(): Promise<TracerSession>;
     protected ensureTenantId(): Promise<string>;
     private _convertToCreate;

package/dist/callbacks/handlers/tracer_langchain.js

@@ -1,7 +1,8 @@
+import { AsyncCaller } from "../../util/async_caller.js";
 import { getRuntimeEnvironment } from "../../util/env.js";
 import { BaseTracer } from "./tracer.js";
 export class LangChainTracer extends BaseTracer {
-    constructor({ exampleId, tenantId, sessionName, sessionExtra, } = {}) {
+    constructor({ exampleId, tenantId, sessionName, sessionExtra, callerParams, } = {}) {
         super();
         Object.defineProperty(this, "name", {
             enumerable: true,
@@ -56,6 +57,12 @@ export class LangChainTracer extends BaseTracer {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "caller", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         // eslint-disable-next-line no-process-env
         if (typeof process !== "undefined" && process.env?.LANGCHAIN_API_KEY) {
             // eslint-disable-next-line no-process-env
@@ -76,6 +83,7 @@ export class LangChainTracer extends BaseTracer {
                 "default";
         this.sessionExtra = sessionExtra;
         this.exampleId = exampleId;
+        this.caller = new AsyncCaller(callerParams ?? {});
     }
     async ensureSession() {
         if (this.session) {
@@ -83,7 +91,7 @@ export class LangChainTracer extends BaseTracer {
         }
         const tenantId = await this.ensureTenantId();
         const endpoint = `${this.endpoint}/sessions?upsert=true`;
-        const res = await fetch(endpoint, {
+        const res = await this.caller.call(fetch, endpoint, {
             method: "POST",
             headers: this.headers,
             body: JSON.stringify({
@@ -105,7 +113,7 @@ export class LangChainTracer extends BaseTracer {
             return this.tenantId;
         }
         const endpoint = `${this.endpoint}/tenants`;
-        const response = await fetch(endpoint, {
+        const response = await this.caller.call(fetch, endpoint, {
             method: "GET",
             headers: this.headers,
         });
@@ -146,7 +154,7 @@ export class LangChainTracer extends BaseTracer {
     async persistRun(run) {
         const persistedRun = await this._convertToCreate(run, this.exampleId);
         const endpoint = `${this.endpoint}/runs`;
-        const response = await fetch(endpoint, {
+        const response = await this.caller.call(fetch, endpoint, {
             method: "POST",
             headers: this.headers,
             body: JSON.stringify(persistedRun),

package/dist/callbacks/manager.cjs

@@ -334,6 +334,10 @@ class CallbackManager extends BaseCallbackManager {
                 ? localHandlers.map(ensureHandler)
                 : localHandlers?.handlers, false);
         }
+        const verboseEnabled = (typeof process !== "undefined"
+            ? // eslint-disable-next-line no-process-env
+                process.env?.LANGCHAIN_VERBOSE !== undefined
+            : false) || options?.verbose;
         const tracingV2Enabled = typeof process !== "undefined"
             ? // eslint-disable-next-line no-process-env
                 process.env?.LANGCHAIN_TRACING_V2 !== undefined
@@ -343,11 +347,11 @@ class CallbackManager extends BaseCallbackManager {
                 ? // eslint-disable-next-line no-process-env
                     process.env?.LANGCHAIN_TRACING !== undefined
                 : false);
-        if (options?.verbose || tracingEnabled) {
+        if (verboseEnabled || tracingEnabled) {
             if (!callbackManager) {
                 callbackManager = new CallbackManager();
             }
-            if (options?.verbose &&
+            if (verboseEnabled &&
                 !callbackManager.handlers.some((handler) => handler.name === console_js_1.ConsoleCallbackHandler.prototype.name)) {
                 const consoleHandler = new console_js_1.ConsoleCallbackHandler();
                 callbackManager.addHandler(consoleHandler, true);

package/dist/callbacks/manager.js

@@ -327,6 +327,10 @@ export class CallbackManager extends BaseCallbackManager {
                 ? localHandlers.map(ensureHandler)
                 : localHandlers?.handlers, false);
         }
+        const verboseEnabled = (typeof process !== "undefined"
+            ? // eslint-disable-next-line no-process-env
+                process.env?.LANGCHAIN_VERBOSE !== undefined
+            : false) || options?.verbose;
         const tracingV2Enabled = typeof process !== "undefined"
             ? // eslint-disable-next-line no-process-env
                 process.env?.LANGCHAIN_TRACING_V2 !== undefined
@@ -336,11 +340,11 @@ export class CallbackManager extends BaseCallbackManager {
                 ? // eslint-disable-next-line no-process-env
                     process.env?.LANGCHAIN_TRACING !== undefined
                 : false);
-        if (options?.verbose || tracingEnabled) {
+        if (verboseEnabled || tracingEnabled) {
             if (!callbackManager) {
                 callbackManager = new CallbackManager();
             }
-            if (options?.verbose &&
+            if (verboseEnabled &&
                 !callbackManager.handlers.some((handler) => handler.name === ConsoleCallbackHandler.prototype.name)) {
                 const consoleHandler = new ConsoleCallbackHandler();
                 callbackManager.addHandler(consoleHandler, true);

package/dist/chains/query_constructor/index.cjs

@@ -0,0 +1,105 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.loadQueryContstructorChain = exports.formatAttributeInfo = exports.StructuredQueryOutputParser = exports.EXAMPLE_PROMPT = exports.DEFAULT_SUFFIX = exports.DEFAULT_SCHEMA = exports.DEFAULT_PREFIX = exports.DEFAULT_EXAMPLES = exports.QueryTransformer = void 0;
+const zod_1 = require("zod");
+const parser_js_1 = require("./parser.cjs");
+Object.defineProperty(exports, "QueryTransformer", { enumerable: true, get: function () { return parser_js_1.QueryTransformer; } });
+const ir_js_1 = require("./ir.cjs");
+const prompt_js_1 = require("./prompt.cjs");
+Object.defineProperty(exports, "DEFAULT_EXAMPLES", { enumerable: true, get: function () { return prompt_js_1.DEFAULT_EXAMPLES; } });
+Object.defineProperty(exports, "DEFAULT_PREFIX", { enumerable: true, get: function () { return prompt_js_1.DEFAULT_PREFIX; } });
+Object.defineProperty(exports, "DEFAULT_SCHEMA", { enumerable: true, get: function () { return prompt_js_1.DEFAULT_SCHEMA; } });
+Object.defineProperty(exports, "DEFAULT_SUFFIX", { enumerable: true, get: function () { return prompt_js_1.DEFAULT_SUFFIX; } });
+Object.defineProperty(exports, "EXAMPLE_PROMPT", { enumerable: true, get: function () { return prompt_js_1.EXAMPLE_PROMPT; } });
+const template_js_1 = require("../../prompts/template.cjs");
+const llm_chain_js_1 = require("../llm_chain.cjs");
+const few_shot_js_1 = require("../../prompts/few_shot.cjs");
+const structured_js_1 = require("../../output_parsers/structured.cjs");
+const queryInputSchema = /* #__PURE__ */ zod_1.z.object({
+    query: /* #__PURE__ */ zod_1.z
+        .string()
+        .describe("text string to compare to document contents"),
+    filter: /* #__PURE__ */ zod_1.z
+        .string()
+        .optional()
+        .describe("logical condition statement for filtering documents"),
+});
+class StructuredQueryOutputParser extends structured_js_1.AsymmetricStructuredOutputParser {
+    constructor(parserFunction) {
+        super(queryInputSchema);
+        Object.defineProperty(this, "parserFunction", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: parserFunction
+        });
+    }
+    async outputProcessor(input) {
+        return this.parserFunction(input.query, input.filter);
+    }
+    static fromComponents(allowedComparators = [], allowedOperators = []) {
+        const queryTransformer = new parser_js_1.QueryTransformer(allowedComparators, allowedOperators);
+        return new StructuredQueryOutputParser(async (query, filter) => {
+            let myQuery = query;
+            if (myQuery.length === 0) {
+                myQuery = " ";
+            }
+            if (filter === "NO_FILTER" || filter === undefined) {
+                return new ir_js_1.StructuredQuery(query);
+            }
+            else {
+                const parsedFilter = await queryTransformer.parse(filter);
+                return new ir_js_1.StructuredQuery(query, parsedFilter);
+            }
+        });
+    }
+}
+exports.StructuredQueryOutputParser = StructuredQueryOutputParser;
+function formatAttributeInfo(info) {
+    const infoObj = info.reduce((acc, attr) => {
+        acc[attr.name] = {
+            type: attr.type,
+            description: attr.description,
+        };
+        return acc;
+    }, {});
+    return JSON.stringify(infoObj, null, 2)
+        .replaceAll("{", "{{")
+        .replaceAll("}", "}}");
+}
+exports.formatAttributeInfo = formatAttributeInfo;
+const defaultExample = prompt_js_1.DEFAULT_EXAMPLES.map((EXAMPLE) => EXAMPLE);
+function _getPrompt(documentContents, attributeInfo, allowedComparators, allowedOperators, examples = defaultExample) {
+    const myAllowedComparators = allowedComparators ?? Object.values(ir_js_1.Comparators);
+    const myAllowedOperators = allowedOperators ?? Object.values(ir_js_1.Operators);
+    const attributeJSON = formatAttributeInfo(attributeInfo);
+    const schema = (0, template_js_1.interpolateFString)(prompt_js_1.DEFAULT_SCHEMA, {
+        allowed_comparators: myAllowedComparators.join(" | "),
+        allowed_operators: myAllowedOperators.join(" | "),
+    });
+    const prefix = (0, template_js_1.interpolateFString)(prompt_js_1.DEFAULT_PREFIX, {
+        schema,
+    });
+    const suffix = (0, template_js_1.interpolateFString)(prompt_js_1.DEFAULT_SUFFIX, {
+        i: examples.length + 1,
+        content: documentContents,
+        attributes: attributeJSON,
+    });
+    const outputParser = StructuredQueryOutputParser.fromComponents(allowedComparators, allowedOperators);
+    return new few_shot_js_1.FewShotPromptTemplate({
+        examples,
+        examplePrompt: prompt_js_1.EXAMPLE_PROMPT,
+        inputVariables: ["query"],
+        suffix,
+        prefix,
+        outputParser,
+    });
+}
+function loadQueryContstructorChain(opts) {
+    const prompt = _getPrompt(opts.documentContents, opts.attributeInfo, opts.allowedComparators, opts.allowedOperators, opts.examples);
+    return new llm_chain_js_1.LLMChain({
+        llm: opts.llm,
+        prompt,
+    });
+}
+exports.loadQueryContstructorChain = loadQueryContstructorChain;

package/dist/chains/query_constructor/index.d.ts

@@ -0,0 +1,37 @@
+import { z } from "zod";
+import { QueryTransformer, TraverseType } from "./parser.js";
+import { Comparator, Operator, StructuredQuery } from "./ir.js";
+import { InputValues } from "../../schema/index.js";
+import { DEFAULT_EXAMPLES, DEFAULT_PREFIX, DEFAULT_SCHEMA, DEFAULT_SUFFIX, EXAMPLE_PROMPT } from "./prompt.js";
+import { LLMChain } from "../llm_chain.js";
+import { BaseLanguageModel } from "../../base_language/index.js";
+import { AsymmetricStructuredOutputParser } from "../../output_parsers/structured.js";
+import { AttributeInfo } from "../../schema/query_constructor.js";
+export { QueryTransformer, TraverseType };
+export { DEFAULT_EXAMPLES, DEFAULT_PREFIX, DEFAULT_SCHEMA, DEFAULT_SUFFIX, EXAMPLE_PROMPT, };
+declare const queryInputSchema: z.ZodObject<{
+    query: z.ZodString;
+    filter: z.ZodOptional<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+    query: string;
+    filter?: string | undefined;
+}, {
+    query: string;
+    filter?: string | undefined;
+}>;
+export declare class StructuredQueryOutputParser extends AsymmetricStructuredOutputParser<typeof queryInputSchema, StructuredQuery> {
+    private parserFunction;
+    constructor(parserFunction: (query: string, filter?: string) => Promise<StructuredQuery>);
+    outputProcessor(input: z.infer<typeof queryInputSchema>): Promise<StructuredQuery>;
+    static fromComponents(allowedComparators?: Comparator[], allowedOperators?: Operator[]): StructuredQueryOutputParser;
+}
+export declare function formatAttributeInfo(info: AttributeInfo[]): string;
+export type QueryConstructorChainOptions = {
+    llm: BaseLanguageModel;
+    documentContents: string;
+    attributeInfo: AttributeInfo[];
+    examples?: InputValues[];
+    allowedComparators?: Comparator[];
+    allowedOperators?: Operator[];
+};
+export declare function loadQueryContstructorChain(opts: QueryConstructorChainOptions): LLMChain<string>;