langchain 0.1.35 → 0.1.37
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/chains/conversational_retrieval_chain.cjs +61 -19
- package/dist/chains/conversational_retrieval_chain.d.ts +61 -19
- package/dist/chains/conversational_retrieval_chain.js +61 -19
- package/dist/chains/llm_chain.cjs +10 -5
- package/dist/chains/llm_chain.d.ts +10 -5
- package/dist/chains/llm_chain.js +10 -5
- package/dist/chains/openai_functions/base.cjs +2 -0
- package/dist/chains/openai_functions/base.d.ts +2 -0
- package/dist/chains/openai_functions/base.js +2 -0
- package/dist/chains/query_constructor/index.cjs +5 -8
- package/dist/chains/query_constructor/index.d.ts +5 -4
- package/dist/chains/query_constructor/index.js +3 -6
- package/dist/chains/query_constructor/ir.cjs +15 -139
- package/dist/chains/query_constructor/ir.d.ts +1 -138
- package/dist/chains/query_constructor/ir.js +1 -132
- package/dist/chains/query_constructor/prompt.cjs +2 -2
- package/dist/chains/query_constructor/prompt.d.ts +1 -1
- package/dist/chains/query_constructor/prompt.js +1 -1
- package/dist/chains/retrieval_qa.cjs +23 -14
- package/dist/chains/retrieval_qa.d.ts +23 -14
- package/dist/chains/retrieval_qa.js +23 -14
- package/dist/document_loaders/fs/unstructured.cjs +1 -1
- package/dist/document_loaders/fs/unstructured.js +1 -1
- package/dist/document_loaders/web/browserbase.cjs +87 -0
- package/dist/document_loaders/web/browserbase.d.ts +49 -0
- package/dist/document_loaders/web/browserbase.js +80 -0
- package/dist/document_loaders/web/firecrawl.cjs +88 -0
- package/dist/document_loaders/web/firecrawl.d.ts +48 -0
- package/dist/document_loaders/web/firecrawl.js +81 -0
- package/dist/document_loaders/web/s3.cjs +2 -2
- package/dist/document_loaders/web/s3.js +2 -2
- package/dist/load/import_constants.cjs +2 -0
- package/dist/load/import_constants.js +2 -0
- package/dist/output_parsers/expression.cjs +1 -1
- package/dist/output_parsers/expression.d.ts +1 -1
- package/dist/output_parsers/expression.js +1 -1
- package/dist/retrievers/self_query/base.cjs +3 -136
- package/dist/retrievers/self_query/base.d.ts +1 -69
- package/dist/retrievers/self_query/base.js +1 -134
- package/dist/retrievers/self_query/chroma.cjs +9 -10
- package/dist/retrievers/self_query/chroma.d.ts +1 -1
- package/dist/retrievers/self_query/chroma.js +1 -2
- package/dist/retrievers/self_query/functional.cjs +2 -195
- package/dist/retrievers/self_query/functional.d.ts +1 -87
- package/dist/retrievers/self_query/functional.js +1 -194
- package/dist/retrievers/self_query/index.cjs +9 -13
- package/dist/retrievers/self_query/index.d.ts +11 -8
- package/dist/retrievers/self_query/index.js +7 -11
- package/dist/retrievers/self_query/pinecone.cjs +9 -10
- package/dist/retrievers/self_query/pinecone.d.ts +1 -1
- package/dist/retrievers/self_query/pinecone.js +1 -2
- package/dist/retrievers/self_query/supabase.cjs +28 -30
- package/dist/retrievers/self_query/supabase.d.ts +1 -2
- package/dist/retrievers/self_query/supabase.js +1 -3
- package/dist/retrievers/self_query/supabase_utils.cjs +2 -2
- package/dist/retrievers/self_query/supabase_utils.d.ts +1 -1
- package/dist/retrievers/self_query/supabase_utils.js +1 -1
- package/dist/retrievers/self_query/vectara.cjs +15 -17
- package/dist/retrievers/self_query/vectara.d.ts +1 -2
- package/dist/retrievers/self_query/vectara.js +1 -3
- package/dist/retrievers/self_query/weaviate.cjs +19 -21
- package/dist/retrievers/self_query/weaviate.d.ts +1 -2
- package/dist/retrievers/self_query/weaviate.js +1 -3
- package/dist/smith/config.d.ts +4 -4
- package/dist/storage/in_memory.cjs +2 -81
- package/dist/storage/in_memory.d.ts +1 -49
- package/dist/storage/in_memory.js +1 -80
- package/dist/text_splitter.cjs +15 -727
- package/dist/text_splitter.d.ts +1 -77
- package/dist/text_splitter.js +1 -720
- package/dist/vectorstores/qdrant.cjs +2 -0
- package/dist/vectorstores/qdrant.js +2 -0
- package/document_loaders/web/browserbase.cjs +1 -0
- package/document_loaders/web/browserbase.d.cts +1 -0
- package/document_loaders/web/browserbase.d.ts +1 -0
- package/document_loaders/web/browserbase.js +1 -0
- package/document_loaders/web/firecrawl.cjs +1 -0
- package/document_loaders/web/firecrawl.d.cts +1 -0
- package/document_loaders/web/firecrawl.d.ts +1 -0
- package/document_loaders/web/firecrawl.js +1 -0
- package/package.json +40 -3
- package/dist/retrievers/self_query/utils.cjs +0 -94
- package/dist/retrievers/self_query/utils.d.ts +0 -29
- package/dist/retrievers/self_query/utils.js +0 -85
|
@@ -2,12 +2,11 @@
|
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
3
|
exports.SelfQueryRetriever = exports.FunctionalTranslator = exports.BasicTranslator = exports.BaseTranslator = void 0;
|
|
4
4
|
const retrievers_1 = require("@langchain/core/retrievers");
|
|
5
|
+
const structured_query_1 = require("@langchain/core/structured_query");
|
|
6
|
+
Object.defineProperty(exports, "BaseTranslator", { enumerable: true, get: function () { return structured_query_1.BaseTranslator; } });
|
|
7
|
+
Object.defineProperty(exports, "BasicTranslator", { enumerable: true, get: function () { return structured_query_1.BasicTranslator; } });
|
|
8
|
+
Object.defineProperty(exports, "FunctionalTranslator", { enumerable: true, get: function () { return structured_query_1.FunctionalTranslator; } });
|
|
5
9
|
const index_js_1 = require("../../chains/query_constructor/index.cjs");
|
|
6
|
-
const functional_js_1 = require("./functional.cjs");
|
|
7
|
-
Object.defineProperty(exports, "FunctionalTranslator", { enumerable: true, get: function () { return functional_js_1.FunctionalTranslator; } });
|
|
8
|
-
const base_js_1 = require("./base.cjs");
|
|
9
|
-
Object.defineProperty(exports, "BaseTranslator", { enumerable: true, get: function () { return base_js_1.BaseTranslator; } });
|
|
10
|
-
Object.defineProperty(exports, "BasicTranslator", { enumerable: true, get: function () { return base_js_1.BasicTranslator; } });
|
|
11
10
|
/**
|
|
12
11
|
* Class for question answering over an index. It retrieves relevant
|
|
13
12
|
* documents based on a query. It extends the BaseRetriever class and
|
|
@@ -41,7 +40,7 @@ class SelfQueryRetriever extends retrievers_1.BaseRetriever {
|
|
|
41
40
|
writable: true,
|
|
42
41
|
value: void 0
|
|
43
42
|
});
|
|
44
|
-
Object.defineProperty(this, "
|
|
43
|
+
Object.defineProperty(this, "queryConstructor", {
|
|
45
44
|
enumerable: true,
|
|
46
45
|
configurable: true,
|
|
47
46
|
writable: true,
|
|
@@ -72,17 +71,14 @@ class SelfQueryRetriever extends retrievers_1.BaseRetriever {
|
|
|
72
71
|
value: { k: 4, forceDefaultFilter: false }
|
|
73
72
|
});
|
|
74
73
|
this.vectorStore = options.vectorStore;
|
|
75
|
-
this.
|
|
74
|
+
this.queryConstructor = options.queryConstructor;
|
|
76
75
|
this.verbose = options.verbose ?? false;
|
|
77
76
|
this.searchParams = options.searchParams ?? this.searchParams;
|
|
78
77
|
this.useOriginalQuery = options.useOriginalQuery ?? this.useOriginalQuery;
|
|
79
78
|
this.structuredQueryTranslator = options.structuredQueryTranslator;
|
|
80
79
|
}
|
|
81
80
|
async _getRelevantDocuments(query, runManager) {
|
|
82
|
-
const
|
|
83
|
-
[this.llmChain.inputKeys[0]]: query,
|
|
84
|
-
}, runManager?.getChild("llm_chain"));
|
|
85
|
-
const generatedStructuredQuery = output;
|
|
81
|
+
const generatedStructuredQuery = await this.queryConstructor.invoke({ query }, runManager?.getChild("query_constructor"));
|
|
86
82
|
const nextArg = this.structuredQueryTranslator.visitStructuredQuery(generatedStructuredQuery);
|
|
87
83
|
const filter = this.structuredQueryTranslator.mergeFilters(this.searchParams?.filter, nextArg.filter, this.searchParams?.mergeFiltersOperator, this.searchParams?.forceDefaultFilter);
|
|
88
84
|
const generatedQuery = generatedStructuredQuery.query;
|
|
@@ -103,7 +99,7 @@ class SelfQueryRetriever extends retrievers_1.BaseRetriever {
|
|
|
103
99
|
*/
|
|
104
100
|
static fromLLM(options) {
|
|
105
101
|
const { structuredQueryTranslator, allowedComparators, allowedOperators, llm, documentContents, attributeInfo, examples, vectorStore, ...rest } = options;
|
|
106
|
-
const
|
|
102
|
+
const queryConstructor = (0, index_js_1.loadQueryConstructorRunnable)({
|
|
107
103
|
llm,
|
|
108
104
|
documentContents,
|
|
109
105
|
attributeInfo,
|
|
@@ -113,7 +109,7 @@ class SelfQueryRetriever extends retrievers_1.BaseRetriever {
|
|
|
113
109
|
});
|
|
114
110
|
return new SelfQueryRetriever({
|
|
115
111
|
...rest,
|
|
116
|
-
|
|
112
|
+
queryConstructor,
|
|
117
113
|
vectorStore,
|
|
118
114
|
structuredQueryTranslator,
|
|
119
115
|
});
|
|
@@ -1,11 +1,10 @@
|
|
|
1
|
+
import { RunnableInterface } from "@langchain/core/runnables";
|
|
1
2
|
import { BaseRetriever, type BaseRetrieverInput } from "@langchain/core/retrievers";
|
|
2
3
|
import { Document } from "@langchain/core/documents";
|
|
3
4
|
import { VectorStore } from "@langchain/core/vectorstores";
|
|
4
|
-
import {
|
|
5
|
-
import {
|
|
6
|
-
import {
|
|
7
|
-
import { FunctionalTranslator } from "./functional.js";
|
|
8
|
-
import { BaseTranslator, BasicTranslator } from "./base.js";
|
|
5
|
+
import { BaseTranslator, BasicTranslator, FunctionalTranslator, StructuredQuery } from "@langchain/core/structured_query";
|
|
6
|
+
import { CallbackManagerForRetrieverRun } from "../../callbacks/manager.js";
|
|
7
|
+
import { QueryConstructorRunnableOptions } from "../../chains/query_constructor/index.js";
|
|
9
8
|
export { BaseTranslator, BasicTranslator, FunctionalTranslator };
|
|
10
9
|
/**
|
|
11
10
|
* Interface for the arguments required to create a SelfQueryRetriever
|
|
@@ -14,7 +13,9 @@ export { BaseTranslator, BasicTranslator, FunctionalTranslator };
|
|
|
14
13
|
export interface SelfQueryRetrieverArgs<T extends VectorStore> extends BaseRetrieverInput {
|
|
15
14
|
vectorStore: T;
|
|
16
15
|
structuredQueryTranslator: BaseTranslator<T>;
|
|
17
|
-
|
|
16
|
+
queryConstructor: RunnableInterface<{
|
|
17
|
+
query: string;
|
|
18
|
+
}, StructuredQuery>;
|
|
18
19
|
verbose?: boolean;
|
|
19
20
|
useOriginalQuery?: boolean;
|
|
20
21
|
searchParams?: {
|
|
@@ -46,7 +47,9 @@ export declare class SelfQueryRetriever<T extends VectorStore> extends BaseRetri
|
|
|
46
47
|
static lc_name(): string;
|
|
47
48
|
get lc_namespace(): string[];
|
|
48
49
|
vectorStore: T;
|
|
49
|
-
|
|
50
|
+
queryConstructor: RunnableInterface<{
|
|
51
|
+
query: string;
|
|
52
|
+
}, StructuredQuery>;
|
|
50
53
|
verbose?: boolean;
|
|
51
54
|
structuredQueryTranslator: BaseTranslator<T>;
|
|
52
55
|
useOriginalQuery: boolean;
|
|
@@ -67,5 +70,5 @@ export declare class SelfQueryRetriever<T extends VectorStore> extends BaseRetri
|
|
|
67
70
|
* @param options The options used to create the SelfQueryRetriever instance. It includes the QueryConstructorChainOptions and all the SelfQueryRetrieverArgs except 'llmChain'.
|
|
68
71
|
* @returns A new instance of SelfQueryRetriever.
|
|
69
72
|
*/
|
|
70
|
-
static fromLLM<T extends VectorStore>(options:
|
|
73
|
+
static fromLLM<T extends VectorStore>(options: QueryConstructorRunnableOptions & Omit<SelfQueryRetrieverArgs<T>, "queryConstructor">): SelfQueryRetriever<T>;
|
|
71
74
|
}
|
|
@@ -1,7 +1,6 @@
|
|
|
1
1
|
import { BaseRetriever, } from "@langchain/core/retrievers";
|
|
2
|
-
import {
|
|
3
|
-
import {
|
|
4
|
-
import { BaseTranslator, BasicTranslator } from "./base.js";
|
|
2
|
+
import { BaseTranslator, BasicTranslator, FunctionalTranslator, } from "@langchain/core/structured_query";
|
|
3
|
+
import { loadQueryConstructorRunnable, } from "../../chains/query_constructor/index.js";
|
|
5
4
|
export { BaseTranslator, BasicTranslator, FunctionalTranslator };
|
|
6
5
|
/**
|
|
7
6
|
* Class for question answering over an index. It retrieves relevant
|
|
@@ -36,7 +35,7 @@ export class SelfQueryRetriever extends BaseRetriever {
|
|
|
36
35
|
writable: true,
|
|
37
36
|
value: void 0
|
|
38
37
|
});
|
|
39
|
-
Object.defineProperty(this, "
|
|
38
|
+
Object.defineProperty(this, "queryConstructor", {
|
|
40
39
|
enumerable: true,
|
|
41
40
|
configurable: true,
|
|
42
41
|
writable: true,
|
|
@@ -67,17 +66,14 @@ export class SelfQueryRetriever extends BaseRetriever {
|
|
|
67
66
|
value: { k: 4, forceDefaultFilter: false }
|
|
68
67
|
});
|
|
69
68
|
this.vectorStore = options.vectorStore;
|
|
70
|
-
this.
|
|
69
|
+
this.queryConstructor = options.queryConstructor;
|
|
71
70
|
this.verbose = options.verbose ?? false;
|
|
72
71
|
this.searchParams = options.searchParams ?? this.searchParams;
|
|
73
72
|
this.useOriginalQuery = options.useOriginalQuery ?? this.useOriginalQuery;
|
|
74
73
|
this.structuredQueryTranslator = options.structuredQueryTranslator;
|
|
75
74
|
}
|
|
76
75
|
async _getRelevantDocuments(query, runManager) {
|
|
77
|
-
const
|
|
78
|
-
[this.llmChain.inputKeys[0]]: query,
|
|
79
|
-
}, runManager?.getChild("llm_chain"));
|
|
80
|
-
const generatedStructuredQuery = output;
|
|
76
|
+
const generatedStructuredQuery = await this.queryConstructor.invoke({ query }, runManager?.getChild("query_constructor"));
|
|
81
77
|
const nextArg = this.structuredQueryTranslator.visitStructuredQuery(generatedStructuredQuery);
|
|
82
78
|
const filter = this.structuredQueryTranslator.mergeFilters(this.searchParams?.filter, nextArg.filter, this.searchParams?.mergeFiltersOperator, this.searchParams?.forceDefaultFilter);
|
|
83
79
|
const generatedQuery = generatedStructuredQuery.query;
|
|
@@ -98,7 +94,7 @@ export class SelfQueryRetriever extends BaseRetriever {
|
|
|
98
94
|
*/
|
|
99
95
|
static fromLLM(options) {
|
|
100
96
|
const { structuredQueryTranslator, allowedComparators, allowedOperators, llm, documentContents, attributeInfo, examples, vectorStore, ...rest } = options;
|
|
101
|
-
const
|
|
97
|
+
const queryConstructor = loadQueryConstructorRunnable({
|
|
102
98
|
llm,
|
|
103
99
|
documentContents,
|
|
104
100
|
attributeInfo,
|
|
@@ -108,7 +104,7 @@ export class SelfQueryRetriever extends BaseRetriever {
|
|
|
108
104
|
});
|
|
109
105
|
return new SelfQueryRetriever({
|
|
110
106
|
...rest,
|
|
111
|
-
|
|
107
|
+
queryConstructor,
|
|
112
108
|
vectorStore,
|
|
113
109
|
structuredQueryTranslator,
|
|
114
110
|
});
|
|
@@ -1,8 +1,7 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
3
|
exports.PineconeTranslator = void 0;
|
|
4
|
-
const
|
|
5
|
-
const base_js_1 = require("./base.cjs");
|
|
4
|
+
const structured_query_1 = require("@langchain/core/structured_query");
|
|
6
5
|
/**
|
|
7
6
|
* Specialized translator class that extends the BasicTranslator. It is
|
|
8
7
|
* designed to work with PineconeStore, a type of vector store in
|
|
@@ -24,17 +23,17 @@ const base_js_1 = require("./base.cjs");
|
|
|
24
23
|
* );
|
|
25
24
|
* ```
|
|
26
25
|
*/
|
|
27
|
-
class PineconeTranslator extends
|
|
26
|
+
class PineconeTranslator extends structured_query_1.BasicTranslator {
|
|
28
27
|
constructor() {
|
|
29
28
|
super({
|
|
30
|
-
allowedOperators: [
|
|
29
|
+
allowedOperators: [structured_query_1.Operators.and, structured_query_1.Operators.or],
|
|
31
30
|
allowedComparators: [
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
|
|
36
|
-
|
|
37
|
-
|
|
31
|
+
structured_query_1.Comparators.eq,
|
|
32
|
+
structured_query_1.Comparators.ne,
|
|
33
|
+
structured_query_1.Comparators.gt,
|
|
34
|
+
structured_query_1.Comparators.gte,
|
|
35
|
+
structured_query_1.Comparators.lt,
|
|
36
|
+
structured_query_1.Comparators.lte,
|
|
38
37
|
],
|
|
39
38
|
});
|
|
40
39
|
}
|
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import type { VectorStoreInterface } from "@langchain/core/vectorstores";
|
|
2
|
-
import { BasicTranslator } from "
|
|
2
|
+
import { BasicTranslator } from "@langchain/core/structured_query";
|
|
3
3
|
/**
|
|
4
4
|
* Specialized translator class that extends the BasicTranslator. It is
|
|
5
5
|
* designed to work with PineconeStore, a type of vector store in
|
|
@@ -1,5 +1,4 @@
|
|
|
1
|
-
import { Comparators, Operators } from "
|
|
2
|
-
import { BasicTranslator } from "./base.js";
|
|
1
|
+
import { BasicTranslator, Comparators, Operators, } from "@langchain/core/structured_query";
|
|
3
2
|
/**
|
|
4
3
|
* Specialized translator class that extends the BasicTranslator. It is
|
|
5
4
|
* designed to work with PineconeStore, a type of vector store in
|
|
@@ -1,9 +1,7 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
3
|
exports.SupabaseTranslator = void 0;
|
|
4
|
-
const
|
|
5
|
-
const base_js_1 = require("./base.cjs");
|
|
6
|
-
const utils_js_1 = require("./utils.cjs");
|
|
4
|
+
const structured_query_1 = require("@langchain/core/structured_query");
|
|
7
5
|
const supabase_utils_js_1 = require("./supabase_utils.cjs");
|
|
8
6
|
/**
|
|
9
7
|
* A specialized translator designed to work with Supabase, extending the
|
|
@@ -24,26 +22,26 @@ const supabase_utils_js_1 = require("./supabase_utils.cjs");
|
|
|
24
22
|
* );
|
|
25
23
|
* ```
|
|
26
24
|
*/
|
|
27
|
-
class SupabaseTranslator extends
|
|
25
|
+
class SupabaseTranslator extends structured_query_1.BaseTranslator {
|
|
28
26
|
constructor() {
|
|
29
27
|
super(...arguments);
|
|
30
28
|
Object.defineProperty(this, "allowedOperators", {
|
|
31
29
|
enumerable: true,
|
|
32
30
|
configurable: true,
|
|
33
31
|
writable: true,
|
|
34
|
-
value: [
|
|
32
|
+
value: [structured_query_1.Operators.and, structured_query_1.Operators.or]
|
|
35
33
|
});
|
|
36
34
|
Object.defineProperty(this, "allowedComparators", {
|
|
37
35
|
enumerable: true,
|
|
38
36
|
configurable: true,
|
|
39
37
|
writable: true,
|
|
40
38
|
value: [
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
46
|
-
|
|
39
|
+
structured_query_1.Comparators.eq,
|
|
40
|
+
structured_query_1.Comparators.ne,
|
|
41
|
+
structured_query_1.Comparators.gt,
|
|
42
|
+
structured_query_1.Comparators.gte,
|
|
43
|
+
structured_query_1.Comparators.lt,
|
|
44
|
+
structured_query_1.Comparators.lte,
|
|
47
45
|
]
|
|
48
46
|
});
|
|
49
47
|
}
|
|
@@ -59,22 +57,22 @@ class SupabaseTranslator extends base_js_1.BaseTranslator {
|
|
|
59
57
|
*/
|
|
60
58
|
getComparatorFunction(comparator) {
|
|
61
59
|
switch (comparator) {
|
|
62
|
-
case
|
|
60
|
+
case structured_query_1.Comparators.eq: {
|
|
63
61
|
return (attr, value) => (rpc) => rpc.eq(this.buildColumnName(attr, value), value);
|
|
64
62
|
}
|
|
65
|
-
case
|
|
63
|
+
case structured_query_1.Comparators.ne: {
|
|
66
64
|
return (attr, value) => (rpc) => rpc.neq(this.buildColumnName(attr, value), value);
|
|
67
65
|
}
|
|
68
|
-
case
|
|
66
|
+
case structured_query_1.Comparators.gt: {
|
|
69
67
|
return (attr, value) => (rpc) => rpc.gt(this.buildColumnName(attr, value), value);
|
|
70
68
|
}
|
|
71
|
-
case
|
|
69
|
+
case structured_query_1.Comparators.gte: {
|
|
72
70
|
return (attr, value) => (rpc) => rpc.gte(this.buildColumnName(attr, value), value);
|
|
73
71
|
}
|
|
74
|
-
case
|
|
72
|
+
case structured_query_1.Comparators.lt: {
|
|
75
73
|
return (attr, value) => (rpc) => rpc.lt(this.buildColumnName(attr, value), value);
|
|
76
74
|
}
|
|
77
|
-
case
|
|
75
|
+
case structured_query_1.Comparators.lte: {
|
|
78
76
|
return (attr, value) => (rpc) => rpc.lte(this.buildColumnName(attr, value), value);
|
|
79
77
|
}
|
|
80
78
|
default: {
|
|
@@ -92,13 +90,13 @@ class SupabaseTranslator extends base_js_1.BaseTranslator {
|
|
|
92
90
|
*/
|
|
93
91
|
buildColumnName(attr, value, includeType = true) {
|
|
94
92
|
let column = "";
|
|
95
|
-
if ((0,
|
|
93
|
+
if ((0, structured_query_1.isString)(value)) {
|
|
96
94
|
column = `metadata->>${attr}`;
|
|
97
95
|
}
|
|
98
|
-
else if ((0,
|
|
96
|
+
else if ((0, structured_query_1.isInt)(value)) {
|
|
99
97
|
column = `metadata->${attr}${includeType ? "::int" : ""}`;
|
|
100
98
|
}
|
|
101
|
-
else if ((0,
|
|
99
|
+
else if ((0, structured_query_1.isFloat)(value)) {
|
|
102
100
|
column = `metadata->${attr}${includeType ? "::float" : ""}`;
|
|
103
101
|
}
|
|
104
102
|
else {
|
|
@@ -141,7 +139,7 @@ class SupabaseTranslator extends base_js_1.BaseTranslator {
|
|
|
141
139
|
visitOperation(operation) {
|
|
142
140
|
const { operator, args } = operation;
|
|
143
141
|
if (this.allowedOperators.includes(operator)) {
|
|
144
|
-
if (operator ===
|
|
142
|
+
if (operator === structured_query_1.Operators.and) {
|
|
145
143
|
if (!args) {
|
|
146
144
|
return (rpc) => rpc;
|
|
147
145
|
}
|
|
@@ -151,7 +149,7 @@ class SupabaseTranslator extends base_js_1.BaseTranslator {
|
|
|
151
149
|
}, rpc);
|
|
152
150
|
return filter;
|
|
153
151
|
}
|
|
154
|
-
else if (operator ===
|
|
152
|
+
else if (operator === structured_query_1.Operators.or) {
|
|
155
153
|
return (rpc) => rpc.or(this.visitOperationAsString(operation));
|
|
156
154
|
}
|
|
157
155
|
else {
|
|
@@ -173,7 +171,7 @@ class SupabaseTranslator extends base_js_1.BaseTranslator {
|
|
|
173
171
|
let { value } = comparison;
|
|
174
172
|
const { comparator: _comparator, attribute } = comparison;
|
|
175
173
|
let comparator = _comparator;
|
|
176
|
-
if (comparator ===
|
|
174
|
+
if (comparator === structured_query_1.Comparators.ne) {
|
|
177
175
|
comparator = "neq";
|
|
178
176
|
}
|
|
179
177
|
if (Array.isArray(value)) {
|
|
@@ -227,27 +225,27 @@ class SupabaseTranslator extends base_js_1.BaseTranslator {
|
|
|
227
225
|
* @returns The merged filter.
|
|
228
226
|
*/
|
|
229
227
|
mergeFilters(defaultFilter, generatedFilter, mergeType = "and") {
|
|
230
|
-
if ((0,
|
|
228
|
+
if ((0, structured_query_1.isFilterEmpty)(defaultFilter) && (0, structured_query_1.isFilterEmpty)(generatedFilter)) {
|
|
231
229
|
return undefined;
|
|
232
230
|
}
|
|
233
|
-
if ((0,
|
|
234
|
-
if ((0,
|
|
231
|
+
if ((0, structured_query_1.isFilterEmpty)(defaultFilter) || mergeType === "replace") {
|
|
232
|
+
if ((0, structured_query_1.isFilterEmpty)(generatedFilter)) {
|
|
235
233
|
return undefined;
|
|
236
234
|
}
|
|
237
235
|
return generatedFilter;
|
|
238
236
|
}
|
|
239
|
-
if ((0,
|
|
237
|
+
if ((0, structured_query_1.isFilterEmpty)(generatedFilter)) {
|
|
240
238
|
if (mergeType === "and") {
|
|
241
239
|
return undefined;
|
|
242
240
|
}
|
|
243
241
|
return defaultFilter;
|
|
244
242
|
}
|
|
245
243
|
let myDefaultFilter = defaultFilter;
|
|
246
|
-
if ((0,
|
|
244
|
+
if ((0, structured_query_1.isObject)(defaultFilter)) {
|
|
247
245
|
const { filter } = this.visitStructuredQuery((0, supabase_utils_js_1.convertObjectFilterToStructuredQuery)(defaultFilter));
|
|
248
246
|
// just in case the built filter is empty somehow
|
|
249
|
-
if ((0,
|
|
250
|
-
if ((0,
|
|
247
|
+
if ((0, structured_query_1.isFilterEmpty)(filter)) {
|
|
248
|
+
if ((0, structured_query_1.isFilterEmpty)(generatedFilter)) {
|
|
251
249
|
return undefined;
|
|
252
250
|
}
|
|
253
251
|
return generatedFilter;
|
|
@@ -1,6 +1,5 @@
|
|
|
1
1
|
import type { SupabaseFilterRPCCall, SupabaseMetadata, SupabaseVectorStore } from "@langchain/community/vectorstores/supabase";
|
|
2
|
-
import { Comparator, Comparison, Operation, Operator, StructuredQuery } from "
|
|
3
|
-
import { BaseTranslator } from "./base.js";
|
|
2
|
+
import { BaseTranslator, Comparator, Comparison, Operation, Operator, StructuredQuery } from "@langchain/core/structured_query";
|
|
4
3
|
/**
|
|
5
4
|
* Represents the possible values that can be used in a comparison in a
|
|
6
5
|
* structured query. It can be a string or a number.
|
|
@@ -1,6 +1,4 @@
|
|
|
1
|
-
import { Comparators, Operators, } from "
|
|
2
|
-
import { BaseTranslator } from "./base.js";
|
|
3
|
-
import { isFilterEmpty, isFloat, isInt, isObject, isString } from "./utils.js";
|
|
1
|
+
import { isFilterEmpty, isFloat, isInt, isObject, isString, BaseTranslator, Comparators, Operators, } from "@langchain/core/structured_query";
|
|
4
2
|
import { ProxyParamsDuplicator, convertObjectFilterToStructuredQuery, } from "./supabase_utils.js";
|
|
5
3
|
/**
|
|
6
4
|
* A specialized translator designed to work with Supabase, extending the
|
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
3
|
exports.convertObjectFilterToStructuredQuery = exports.ProxyParamsDuplicator = void 0;
|
|
4
|
-
const
|
|
4
|
+
const structured_query_1 = require("@langchain/core/structured_query");
|
|
5
5
|
/**
|
|
6
6
|
* Utility class used to duplicate parameters for a proxy object,
|
|
7
7
|
* specifically designed to work with `SupabaseFilter` objects. It
|
|
@@ -250,6 +250,6 @@ exports.ProxyParamsDuplicator = ProxyParamsDuplicator;
|
|
|
250
250
|
* `Operation` and `Comparison` classes to build the query.
|
|
251
251
|
*/
|
|
252
252
|
function convertObjectFilterToStructuredQuery(objFilter) {
|
|
253
|
-
return new
|
|
253
|
+
return new structured_query_1.StructuredQuery("", new structured_query_1.Operation(structured_query_1.Operators.and, Object.entries(objFilter).map(([column, value]) => new structured_query_1.Comparison(structured_query_1.Comparators.eq, column, value))));
|
|
254
254
|
}
|
|
255
255
|
exports.convertObjectFilterToStructuredQuery = convertObjectFilterToStructuredQuery;
|
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import type { SupabaseFilter, SupabaseFilterRPCCall, SupabaseMetadata } from "@langchain/community/vectorstores/supabase";
|
|
2
|
-
import { StructuredQuery } from "
|
|
2
|
+
import { StructuredQuery } from "@langchain/core/structured_query";
|
|
3
3
|
/**
|
|
4
4
|
* Utility class used to duplicate parameters for a proxy object,
|
|
5
5
|
* specifically designed to work with `SupabaseFilter` objects. It
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import { Comparators, Comparison, Operation, Operators, StructuredQuery, } from "
|
|
1
|
+
import { Comparators, Comparison, Operation, Operators, StructuredQuery, } from "@langchain/core/structured_query";
|
|
2
2
|
/**
|
|
3
3
|
* Utility class used to duplicate parameters for a proxy object,
|
|
4
4
|
* specifically designed to work with `SupabaseFilter` objects. It
|
|
@@ -1,9 +1,7 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
3
|
exports.VectaraTranslator = void 0;
|
|
4
|
-
const
|
|
5
|
-
const base_js_1 = require("./base.cjs");
|
|
6
|
-
const utils_js_1 = require("./utils.cjs");
|
|
4
|
+
const structured_query_1 = require("@langchain/core/structured_query");
|
|
7
5
|
function processValue(value) {
|
|
8
6
|
/** Convert a value to a string and add single quotes if it is a string. */
|
|
9
7
|
if (typeof value === "string") {
|
|
@@ -13,37 +11,37 @@ function processValue(value) {
|
|
|
13
11
|
return String(value);
|
|
14
12
|
}
|
|
15
13
|
}
|
|
16
|
-
class VectaraTranslator extends
|
|
14
|
+
class VectaraTranslator extends structured_query_1.BaseTranslator {
|
|
17
15
|
constructor() {
|
|
18
16
|
super(...arguments);
|
|
19
17
|
Object.defineProperty(this, "allowedOperators", {
|
|
20
18
|
enumerable: true,
|
|
21
19
|
configurable: true,
|
|
22
20
|
writable: true,
|
|
23
|
-
value: [
|
|
21
|
+
value: [structured_query_1.Operators.and, structured_query_1.Operators.or]
|
|
24
22
|
});
|
|
25
23
|
Object.defineProperty(this, "allowedComparators", {
|
|
26
24
|
enumerable: true,
|
|
27
25
|
configurable: true,
|
|
28
26
|
writable: true,
|
|
29
27
|
value: [
|
|
30
|
-
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
|
|
28
|
+
structured_query_1.Comparators.eq,
|
|
29
|
+
structured_query_1.Comparators.ne,
|
|
30
|
+
structured_query_1.Comparators.lt,
|
|
31
|
+
structured_query_1.Comparators.lte,
|
|
32
|
+
structured_query_1.Comparators.gt,
|
|
33
|
+
structured_query_1.Comparators.gte,
|
|
36
34
|
]
|
|
37
35
|
});
|
|
38
36
|
}
|
|
39
37
|
formatFunction(func) {
|
|
40
|
-
if (func in
|
|
38
|
+
if (func in structured_query_1.Comparators) {
|
|
41
39
|
if (this.allowedComparators.length > 0 &&
|
|
42
40
|
this.allowedComparators.indexOf(func) === -1) {
|
|
43
41
|
throw new Error(`Comparator ${func} not allowed. Allowed operators: ${this.allowedComparators.join(", ")}`);
|
|
44
42
|
}
|
|
45
43
|
}
|
|
46
|
-
else if (func in
|
|
44
|
+
else if (func in structured_query_1.Operators) {
|
|
47
45
|
if (this.allowedOperators.length > 0 &&
|
|
48
46
|
this.allowedOperators.indexOf(func) === -1) {
|
|
49
47
|
throw new Error(`Operator ${func} not allowed. Allowed operators: ${this.allowedOperators.join(", ")}`);
|
|
@@ -102,16 +100,16 @@ class VectaraTranslator extends base_js_1.BaseTranslator {
|
|
|
102
100
|
return nextArg;
|
|
103
101
|
}
|
|
104
102
|
mergeFilters(defaultFilter, generatedFilter, mergeType = "and", forceDefaultFilter = false) {
|
|
105
|
-
if ((0,
|
|
103
|
+
if ((0, structured_query_1.isFilterEmpty)(defaultFilter) && (0, structured_query_1.isFilterEmpty)(generatedFilter)) {
|
|
106
104
|
return undefined;
|
|
107
105
|
}
|
|
108
|
-
if ((0,
|
|
109
|
-
if ((0,
|
|
106
|
+
if ((0, structured_query_1.isFilterEmpty)(defaultFilter) || mergeType === "replace") {
|
|
107
|
+
if ((0, structured_query_1.isFilterEmpty)(generatedFilter)) {
|
|
110
108
|
return undefined;
|
|
111
109
|
}
|
|
112
110
|
return generatedFilter;
|
|
113
111
|
}
|
|
114
|
-
if ((0,
|
|
112
|
+
if ((0, structured_query_1.isFilterEmpty)(generatedFilter)) {
|
|
115
113
|
if (forceDefaultFilter) {
|
|
116
114
|
return defaultFilter;
|
|
117
115
|
}
|
|
@@ -1,6 +1,5 @@
|
|
|
1
1
|
import { VectaraFilter, VectaraStore } from "@langchain/community/vectorstores/vectara";
|
|
2
|
-
import { Comparator, Comparison, Operation, Operator, StructuredQuery } from "
|
|
3
|
-
import { BaseTranslator } from "./base.js";
|
|
2
|
+
import { BaseTranslator, Comparator, Comparison, Operation, Operator, StructuredQuery } from "@langchain/core/structured_query";
|
|
4
3
|
export type VectaraVisitorResult = VectaraOperationResult | VectaraComparisonResult | VectaraVisitorStructuredQueryResult;
|
|
5
4
|
export type VectaraOperationResult = String;
|
|
6
5
|
export type VectaraComparisonResult = String;
|
|
@@ -1,6 +1,4 @@
|
|
|
1
|
-
import { Comparators, Operators, } from "
|
|
2
|
-
import { BaseTranslator } from "./base.js";
|
|
3
|
-
import { isFilterEmpty } from "./utils.js";
|
|
1
|
+
import { BaseTranslator, isFilterEmpty, Comparators, Operators, } from "@langchain/core/structured_query";
|
|
4
2
|
function processValue(value) {
|
|
5
3
|
/** Convert a value to a string and add single quotes if it is a string. */
|
|
6
4
|
if (typeof value === "string") {
|
|
@@ -1,9 +1,7 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
3
|
exports.WeaviateTranslator = void 0;
|
|
4
|
-
const
|
|
5
|
-
const base_js_1 = require("./base.cjs");
|
|
6
|
-
const utils_js_1 = require("./utils.cjs");
|
|
4
|
+
const structured_query_1 = require("@langchain/core/structured_query");
|
|
7
5
|
/**
|
|
8
6
|
* A class that translates or converts data into a format that can be used
|
|
9
7
|
* with Weaviate, a vector search engine. It extends the `BaseTranslator`
|
|
@@ -23,26 +21,26 @@ const utils_js_1 = require("./utils.cjs");
|
|
|
23
21
|
* );
|
|
24
22
|
* ```
|
|
25
23
|
*/
|
|
26
|
-
class WeaviateTranslator extends
|
|
24
|
+
class WeaviateTranslator extends structured_query_1.BaseTranslator {
|
|
27
25
|
constructor() {
|
|
28
26
|
super(...arguments);
|
|
29
27
|
Object.defineProperty(this, "allowedOperators", {
|
|
30
28
|
enumerable: true,
|
|
31
29
|
configurable: true,
|
|
32
30
|
writable: true,
|
|
33
|
-
value: [
|
|
31
|
+
value: [structured_query_1.Operators.and, structured_query_1.Operators.or]
|
|
34
32
|
});
|
|
35
33
|
Object.defineProperty(this, "allowedComparators", {
|
|
36
34
|
enumerable: true,
|
|
37
35
|
configurable: true,
|
|
38
36
|
writable: true,
|
|
39
37
|
value: [
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
38
|
+
structured_query_1.Comparators.eq,
|
|
39
|
+
structured_query_1.Comparators.ne,
|
|
40
|
+
structured_query_1.Comparators.lt,
|
|
41
|
+
structured_query_1.Comparators.lte,
|
|
42
|
+
structured_query_1.Comparators.gt,
|
|
43
|
+
structured_query_1.Comparators.gte,
|
|
46
44
|
]
|
|
47
45
|
});
|
|
48
46
|
}
|
|
@@ -54,13 +52,13 @@ class WeaviateTranslator extends base_js_1.BaseTranslator {
|
|
|
54
52
|
* @returns A string representation of the function.
|
|
55
53
|
*/
|
|
56
54
|
formatFunction(func) {
|
|
57
|
-
if (func in
|
|
55
|
+
if (func in structured_query_1.Comparators) {
|
|
58
56
|
if (this.allowedComparators.length > 0 &&
|
|
59
57
|
this.allowedComparators.indexOf(func) === -1) {
|
|
60
58
|
throw new Error(`Comparator ${func} not allowed. Allowed operators: ${this.allowedComparators.join(", ")}`);
|
|
61
59
|
}
|
|
62
60
|
}
|
|
63
|
-
else if (func in
|
|
61
|
+
else if (func in structured_query_1.Operators) {
|
|
64
62
|
if (this.allowedOperators.length > 0 &&
|
|
65
63
|
this.allowedOperators.indexOf(func) === -1) {
|
|
66
64
|
throw new Error(`Operator ${func} not allowed. Allowed operators: ${this.allowedOperators.join(", ")}`);
|
|
@@ -102,21 +100,21 @@ class WeaviateTranslator extends base_js_1.BaseTranslator {
|
|
|
102
100
|
* @returns A WeaviateComparisonResult.
|
|
103
101
|
*/
|
|
104
102
|
visitComparison(comparison) {
|
|
105
|
-
if ((0,
|
|
103
|
+
if ((0, structured_query_1.isString)(comparison.value)) {
|
|
106
104
|
return {
|
|
107
105
|
path: [comparison.attribute],
|
|
108
106
|
operator: this.formatFunction(comparison.comparator),
|
|
109
107
|
valueText: comparison.value,
|
|
110
108
|
};
|
|
111
109
|
}
|
|
112
|
-
if ((0,
|
|
110
|
+
if ((0, structured_query_1.isInt)(comparison.value)) {
|
|
113
111
|
return {
|
|
114
112
|
path: [comparison.attribute],
|
|
115
113
|
operator: this.formatFunction(comparison.comparator),
|
|
116
114
|
valueInt: parseInt(comparison.value, 10),
|
|
117
115
|
};
|
|
118
116
|
}
|
|
119
|
-
if ((0,
|
|
117
|
+
if ((0, structured_query_1.isFloat)(comparison.value)) {
|
|
120
118
|
return {
|
|
121
119
|
path: [comparison.attribute],
|
|
122
120
|
operator: this.formatFunction(comparison.comparator),
|
|
@@ -152,17 +150,17 @@ class WeaviateTranslator extends base_js_1.BaseTranslator {
|
|
|
152
150
|
* @returns A merged WeaviateFilter, or undefined if both filters are empty.
|
|
153
151
|
*/
|
|
154
152
|
mergeFilters(defaultFilter, generatedFilter, mergeType = "and") {
|
|
155
|
-
if ((0,
|
|
156
|
-
(0,
|
|
153
|
+
if ((0, structured_query_1.isFilterEmpty)(defaultFilter?.where) &&
|
|
154
|
+
(0, structured_query_1.isFilterEmpty)(generatedFilter?.where)) {
|
|
157
155
|
return undefined;
|
|
158
156
|
}
|
|
159
|
-
if ((0,
|
|
160
|
-
if ((0,
|
|
157
|
+
if ((0, structured_query_1.isFilterEmpty)(defaultFilter?.where) || mergeType === "replace") {
|
|
158
|
+
if ((0, structured_query_1.isFilterEmpty)(generatedFilter?.where)) {
|
|
161
159
|
return undefined;
|
|
162
160
|
}
|
|
163
161
|
return generatedFilter;
|
|
164
162
|
}
|
|
165
|
-
if ((0,
|
|
163
|
+
if ((0, structured_query_1.isFilterEmpty)(generatedFilter?.where)) {
|
|
166
164
|
if (mergeType === "and") {
|
|
167
165
|
return undefined;
|
|
168
166
|
}
|
|
@@ -1,6 +1,5 @@
|
|
|
1
1
|
import { WeaviateFilter, WeaviateStore } from "@langchain/community/vectorstores/weaviate";
|
|
2
|
-
import { Comparator, Comparison, Operation, Operator, StructuredQuery } from "
|
|
3
|
-
import { BaseTranslator } from "./base.js";
|
|
2
|
+
import { BaseTranslator, Comparator, Comparison, Operation, Operator, StructuredQuery } from "@langchain/core/structured_query";
|
|
4
3
|
type WeaviateOperatorValues = {
|
|
5
4
|
valueText: string;
|
|
6
5
|
valueInt: number;
|