@n8n/n8n-nodes-langchain 2.43.2 → 2.44.0
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/credentials/AzureEntraCognitiveServicesOAuth2Api.credentials.js +4 -4
- package/dist/credentials/AzureEntraCognitiveServicesOAuth2Api.credentials.js.map +1 -1
- package/dist/credentials/AzureOpenAiApi.credentials.js +4 -4
- package/dist/credentials/AzureOpenAiApi.credentials.js.map +1 -1
- package/dist/credentials/GoogleVertexAiApi.credentials.d.ts +15 -0
- package/dist/credentials/GoogleVertexAiApi.credentials.js +83 -0
- package/dist/credentials/GoogleVertexAiApi.credentials.js.map +1 -0
- package/dist/credentials/icons/google.svg +1 -0
- package/dist/known/credentials.json +12 -0
- package/dist/known/nodes.json +4 -0
- package/dist/node-definitions/.nodes-hash +1 -1
- package/dist/node-definitions/nodes/n8n-nodes-langchain/embeddingsAzureOpenAi/index.ts +1 -1
- package/dist/node-definitions/nodes/n8n-nodes-langchain/embeddingsAzureOpenAi/v1.schema.js +2 -1
- package/dist/node-definitions/nodes/n8n-nodes-langchain/embeddingsAzureOpenAi/v1.ts +4 -2
- package/dist/node-definitions/nodes/n8n-nodes-langchain/embeddingsGoogleVertex/v1.schema.js +3 -2
- package/dist/node-definitions/nodes/n8n-nodes-langchain/embeddingsGoogleVertex/v1.ts +3 -0
- package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatAzureOpenAi/index.ts +1 -1
- package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatAzureOpenAi/v1.schema.js +2 -1
- package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatAzureOpenAi/v1.ts +11 -2
- package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatAzureOpenAi/v11.schema.js +3 -2
- package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatAzureOpenAi/v11.ts +15 -5
- package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatGoogleGemini/v1.ts +1 -1
- package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatGoogleGemini/v11.ts +1 -1
- package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatGoogleGemini/v12.ts +1 -1
- package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatGoogleVertex/v1.schema.js +3 -2
- package/dist/node-definitions/nodes/n8n-nodes-langchain/lmChatGoogleVertex/v1.ts +3 -0
- package/dist/nodes/agents/Agent/agents/ToolsAgent/V2/execute.d.ts +6 -1
- package/dist/nodes/agents/Agent/agents/ToolsAgent/V2/execute.js +23 -7
- package/dist/nodes/agents/Agent/agents/ToolsAgent/V2/execute.js.map +1 -1
- package/dist/nodes/embeddings/EmbeddingsAzureOpenAi/EmbeddingsAzureOpenAi.node.js +102 -16
- package/dist/nodes/embeddings/EmbeddingsAzureOpenAi/EmbeddingsAzureOpenAi.node.js.map +1 -1
- package/dist/nodes/embeddings/EmbeddingsDatabricks/EmbeddingsDatabricks.node.js +1 -0
- package/dist/nodes/embeddings/EmbeddingsDatabricks/EmbeddingsDatabricks.node.js.map +1 -1
- package/dist/nodes/embeddings/EmbeddingsGoogleVertex/EmbeddingsGoogleVertex.node.d.ts +5 -6
- package/dist/nodes/embeddings/EmbeddingsGoogleVertex/EmbeddingsGoogleVertex.node.js +9 -41
- package/dist/nodes/embeddings/EmbeddingsGoogleVertex/EmbeddingsGoogleVertex.node.js.map +1 -1
- package/dist/nodes/llms/LMChatAnthropic/LmChatAnthropic.node.js +1 -15
- package/dist/nodes/llms/LMChatAnthropic/LmChatAnthropic.node.js.map +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/LmChatAzureOpenAi.node.js +59 -8
- package/dist/nodes/llms/LmChatAzureOpenAi/LmChatAzureOpenAi.node.js.map +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/credentials/api-key.js +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/credentials/api-key.js.map +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/credentials/oauth2.js +1 -2
- package/dist/nodes/llms/LmChatAzureOpenAi/credentials/oauth2.js.map +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/credentials/requireFoundryEndpoint.js +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/credentials/requireFoundryEndpoint.js.map +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/credentials/requireTenantId.js +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/credentials/requireTenantId.js.map +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/error-handling.d.ts +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/error-handling.js +12 -6
- package/dist/nodes/llms/LmChatAzureOpenAi/error-handling.js.map +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/methods/searchModels.js +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/methods/searchModels.js.map +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/properties.js +29 -4
- package/dist/nodes/llms/LmChatAzureOpenAi/properties.js.map +1 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/types.d.ts +1 -0
- package/dist/nodes/llms/LmChatAzureOpenAi/types.js +2 -1
- package/dist/nodes/llms/LmChatAzureOpenAi/types.js.map +1 -1
- package/dist/nodes/llms/LmChatGoogleGemini/LmChatGoogleGemini.node.js +2 -1
- package/dist/nodes/llms/LmChatGoogleGemini/LmChatGoogleGemini.node.js.map +1 -1
- package/dist/nodes/llms/LmChatGoogleVertex/LmChatGoogleVertex.node.d.ts +5 -6
- package/dist/nodes/llms/LmChatGoogleVertex/LmChatGoogleVertex.node.js +9 -41
- package/dist/nodes/llms/LmChatGoogleVertex/LmChatGoogleVertex.node.js.map +1 -1
- package/dist/nodes/vector_store/VectorStoreDatabricks/DatabricksVectorStore.d.ts +1 -0
- package/dist/nodes/vector_store/VectorStoreDatabricks/DatabricksVectorStore.js +19 -5
- package/dist/nodes/vector_store/VectorStoreDatabricks/DatabricksVectorStore.js.map +1 -1
- package/dist/nodes/vector_store/VectorStoreDatabricks/VectorStoreDatabricks.node.js +4 -1
- package/dist/nodes/vector_store/VectorStoreDatabricks/VectorStoreDatabricks.node.js.map +1 -1
- package/dist/nodes/vendors/Databricks/DatabricksAiAgent.node.d.ts +5 -0
- package/dist/nodes/vendors/Databricks/DatabricksAiAgent.node.js +170 -0
- package/dist/nodes/vendors/Databricks/DatabricksAiAgent.node.js.map +1 -0
- package/dist/nodes/vendors/Databricks/databricks.dark.svg +19 -0
- package/dist/nodes/vendors/Databricks/databricks.svg +19 -0
- package/dist/nodes/vendors/Databricks/mlflow/span-collector.d.ts +48 -0
- package/dist/nodes/vendors/Databricks/mlflow/span-collector.js +378 -0
- package/dist/nodes/vendors/Databricks/mlflow/span-collector.js.map +1 -0
- package/dist/nodes/vendors/Databricks/mlflow/trace-writer.d.ts +41 -0
- package/dist/nodes/vendors/Databricks/mlflow/trace-writer.js +129 -0
- package/dist/nodes/vendors/Databricks/mlflow/trace-writer.js.map +1 -0
- package/dist/nodes/vendors/Databricks/mlflow/transport.d.ts +7 -0
- package/dist/nodes/vendors/Databricks/mlflow/transport.js +72 -0
- package/dist/nodes/vendors/Databricks/mlflow/transport.js.map +1 -0
- package/dist/nodes/vendors/Databricks/mlflow/types.d.ts +54 -0
- package/dist/nodes/vendors/Databricks/mlflow/types.js +23 -0
- package/dist/nodes/vendors/Databricks/mlflow/types.js.map +1 -0
- package/dist/typecheck.tsbuildinfo +1 -1
- package/dist/types/credentials.json +3 -2
- package/dist/types/nodes.json +6 -5
- package/dist/utils/google-vertex.d.ts +14 -0
- package/dist/utils/google-vertex.js +130 -0
- package/dist/utils/google-vertex.js.map +1 -0
- package/package.json +14 -12
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"LmChatGoogleVertex.node.js","sourceRoot":"","sources":["../../../../nodes/llms/LmChatGoogleVertex/LmChatGoogleVertex.node.ts"],"names":[],"mappings":";;;
|
|
1
|
+
{"version":3,"file":"LmChatGoogleVertex.node.js","sourceRoot":"","sources":["../../../../nodes/llms/LmChatGoogleVertex/LmChatGoogleVertex.node.ts"],"names":[],"mappings":";;;AACA,gEAAkF;AAClF,oDAI2B;AAC3B,+CAWsB;AAEtB,oEAAkE;AAElE,qDAAkF;AAClF,4EAA2E;AAC3E,sEAA0F;AAE1F,wDAM8B;AAE9B,SAAS,sBAAsB,CAAC,KAAgB;IAC/C,IAAI,KAAK,CAAC,WAAW,EAAE,QAAQ,CAAC,iDAAiD,CAAC,EAAE,CAAC;QACpF,OAAO,0KAA0K,CAAC;IACnL,CAAC;IAED,OAAO,KAAK,CAAC,WAAW,IAAI,eAAe,CAAC;AAC7C,CAAC;AAED;IAAA;QACC,gBAAW,GAAyB;YACnC,WAAW,EAAE,0BAA0B;YAEvC,IAAI,EAAE,oBAAoB;YAC1B,IAAI,EAAE,iBAAiB;YACvB,KAAK,EAAE,CAAC,WAAW,CAAC;YACpB,OAAO,EAAE,CAAC;YACV,WAAW,EAAE,0BAA0B;YACvC,QAAQ,EAAE;gBACT,IAAI,EAAE,0BAA0B;aAChC;YACD,KAAK,EAAE;gBACN,UAAU,EAAE,CAAC,IAAI,CAAC;gBAClB,aAAa,EAAE;oBACd,EAAE,EAAE,CAAC,iBAAiB,EAAE,YAAY,CAAC;oBACrC,iBAAiB,EAAE,CAAC,2BAA2B,CAAC;iBAChD;gBACD,SAAS,EAAE;oBACV,oBAAoB,EAAE;wBACrB;4BACC,GAAG,EAAE,0GAA0G;yBAC/G;qBACD;iBACD;aACD;YAED,MAAM,EAAE,EAAE;YAEV,OAAO,EAAE,CAAC,kCAAmB,CAAC,eAAe,CAAC;YAC9C,WAAW,EAAE,CAAC,OAAO,CAAC;YACtB,WAAW,EAAE,uCAAuB;YACpC,UAAU,EAAE;gBACX,0CAA0B;gBAC1B,IAAA,2CAA4B,EAAC,CAAC,kCAAmB,CAAC,OAAO,EAAE,kCAAmB,CAAC,OAAO,CAAC,CAAC;gBACxF;oBACC,WAAW,EAAE,YAAY;oBACzB,IAAI,EAAE,WAAW;oBACjB,cAAc,EAAE,EAAE,IAAI,EAAE,EAAE,cAAc,EAAE,CAAC,WAAW,CAAC,EAAE,EAAE;oBAC3D,IAAI,EAAE,iBAAiB;oBACvB,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE;oBACpC,QAAQ,EAAE,IAAI;oBACd,WAAW,EAAE,8CAA8C;oBAC3D,KAAK,EAAE;wBACN;4BACC,WAAW,EAAE,WAAW;4BACxB,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,MAAM;4BACZ,WAAW,EAAE;gCACZ,gBAAgB,EAAE,iBAAiB;6BACnC;yBACD;wBACD;4BACC,WAAW,EAAE,IAAI;4BACjB,IAAI,EAAE,IAAI;4BACV,IAAI,EAAE,QAAQ;yBACd;qBACD;iBACD;gBACD;oBACC,WAAW,EAAE,YAAY;oBACzB,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,QAAQ;oBACd,WAAW,EACV,4IAA4I;oBAC7I,OAAO,EAAE,kBAAkB;oBAC3B,WAAW,EAAE;wBACZ,YAAY,EACX,2KAA2K;4BAC3K,0CAAoB;qBACrB;iBACD;gBACD,qCAAmB;gBACnB,IAAA,yCAAoB,EAAC,EAAE,sBAAsB,EAAE,IAAI,EAAE,CAAC;aACtD;SACD,CAAC;QAEF,YAAO,GAAG;YACT,cAAc,EAAE,EAAE,4BAA4B,EAA5B,4CAA4B,EAAE;YAChD,UAAU,EAAE;gBACX,KAAK,CAAC,eAAe,CAEpB,MAAe,EACf,eAAwB;oBAExB,OAAO,MAAM,IAAA,oCAAoB,EAChC,MAAM,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,EACtC,MAAM,EACN,eAAe,CACf,CAAC;gBACH,CAAC;aACD;SACD,CAAC;IA2GH,CAAC;IAzGA,KAAK,CAAC,UAAU,CAA6B,SAAiB;QAC7D,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE,QAAQ,EAAE,GAAG,MAAM,IAAA,8CAA8B,EAChF,IAAI,EACJ,SAAS,CACT,CAAC;QACF,MAAM,QAAQ,GAAG,IAAA,mCAAiB,EAAC,QAAQ,CAAC,CAAC;QAE7C,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,WAAW,EAAE,SAAS,CAAW,CAAC;QAE1E,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,SAAS,EAAE,SAAS,EAAE;YAC3D,eAAe,EAAE,IAAI;YACrB,WAAW,EAAE,GAAG;YAChB,IAAI,EAAE,EAAE;YACR,IAAI,EAAE,GAAG;SACT,CAAC,CAAC;QAGH,IAAA,qCAAsB,EACrB,OAAO,EACP;YACC,eAAe,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE;YACpD,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE;YAChD,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE;YACzC,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE;YACzC,cAAc,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE;SACnD,EACD,IAAI,CAAC,OAAO,EAAE,CACd,CAAC;QAEF,MAAM,cAAc,GAAG,IAAI,CAAC,gBAAgB,CAC3C,+BAA+B,EAC/B,SAAS,EACT,IAAI,CACuB,CAAC;QAE7B,IAAI,CAAC;YACJ,MAAM,WAAW,GAAsB;gBACtC,WAAW,EAAE;oBACZ,SAAS;oBACT,WAAW;iBACX;gBACD,QAAQ;gBACR,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACjC,KAAK,EAAE,SAAS;gBAChB,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,WAAW,EAAE,OAAO,CAAC,WAAW;gBAChC,eAAe,EAAE,OAAO,CAAC,eAAe;gBACxC,cAAc;gBACd,SAAS,EAAE,CAAC,IAAI,4BAAa,CAAC,IAAI,EAAE,EAAE,sBAAsB,EAAE,CAAC,CAAC;gBAEhE,eAAe,EAAE,IAAA,6CAA8B,EAAC,IAAI,EAAE,CAAC,KAAU,EAAE,EAAE;oBAEpE,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;oBAC/C,MAAM,WAAW,GAAG,IAAA,oCAAmB,EAAC,MAAM,EAAE;wBAC/C,SAAS;qBACT,CAAC,CAAC;oBAEH,IAAI,WAAW,EAAE,CAAC;wBACjB,MAAM,IAAI,iCAAkB,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,KAAmB,EAAE,WAAW,CAAC,CAAC;oBAChF,CAAC;oBAED,IAAI,MAAM,KAAK,GAAG,EAAE,CAAC;wBAGpB,MAAM,IAAI,iCAAkB,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,KAAmB,EAAE;4BACjE,OAAO,EAAE,4CAA4C;4BACrD,WAAW,EACV,IAAA,0CAAyB,EAAC,KAA6B,CAAC;gCAExD,CAAC,OAAO,KAAK,EAAE,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAE,KAAK,CAAC,OAAkB,CAAC,CAAC,CAAC,SAAS,CAAC;yBAC7E,CAAC,CAAC;oBACJ,CAAC;oBAED,MAAM,KAAK,CAAC;gBACb,CAAC,CAAC;aACF,CAAC;YAGF,IAAI,OAAO,CAAC,cAAc,KAAK,SAAS,EAAE,CAAC;gBAC1C,WAAW,CAAC,cAAc,GAAG,OAAO,CAAC,cAAc,CAAC;YACrD,CAAC;YAED,MAAM,KAAK,GAAG,IAAI,8BAAY,CAAC,WAAW,CAAC,CAAC;YAE5C,OAAO;gBACN,QAAQ,EAAE,KAAK;aACf,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YAGZ,IAAI,CAAC,EAAE,OAAO,EAAE,UAAU,CAAC,+BAA+B,CAAC,EAAE,CAAC;gBAC7D,MAAM,IAAI,iCAAkB,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,CAAe,EAAE;oBAC7D,OAAO,EAAE,mBAAmB;oBAC5B,WAAW,EAAE,mDAAmD;iBAChE,CAAC,CAAC;YACJ,CAAC;YAGD,MAAM,IAAI,iCAAkB,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,CAAe,EAAE;gBAC7D,OAAO,EAAE,iBAAiB;gBAC1B,WAAW,EAAE,CAAC,CAAC,OAAO;aACtB,CAAC,CAAC;QACJ,CAAC;IACF,CAAC;CACD"}
|
|
@@ -54,6 +54,7 @@ export declare class DatabricksVectorStore extends VectorStore {
|
|
|
54
54
|
addVectors(vectors: number[][], documents: DocumentInterface[], options?: {
|
|
55
55
|
ids?: string[];
|
|
56
56
|
}): Promise<string[]>;
|
|
57
|
+
private assertWritableContentColumn;
|
|
57
58
|
private assertDirectAccess;
|
|
58
59
|
private query;
|
|
59
60
|
}
|
|
@@ -164,16 +164,13 @@ class DatabricksVectorStore extends vectorstores_1.VectorStore {
|
|
|
164
164
|
async addDocuments(documents, options) {
|
|
165
165
|
if (documents.length === 0)
|
|
166
166
|
return [];
|
|
167
|
-
this.
|
|
167
|
+
this.assertWritableContentColumn();
|
|
168
168
|
const vectors = await this.embeddings.embedDocuments(documents.map((doc) => doc.pageContent));
|
|
169
169
|
return await this.addVectors(vectors, documents, options);
|
|
170
170
|
}
|
|
171
171
|
async addVectors(vectors, documents, options) {
|
|
172
|
-
const vectorColumn = this.
|
|
172
|
+
const vectorColumn = this.assertWritableContentColumn();
|
|
173
173
|
const { primaryKey, name } = this.index;
|
|
174
|
-
if (this.contentColumn === primaryKey) {
|
|
175
|
-
throw new n8n_workflow_1.UserError(`Column ${primaryKey} is the primary key of ${name}. Select another content column`);
|
|
176
|
-
}
|
|
177
174
|
const schemaColumns = new Set(this.index.schemaColumns ?? []);
|
|
178
175
|
const ids = documents.map((doc, i) => options?.ids?.[i] ?? doc.id ?? (0, node_crypto_1.randomUUID)());
|
|
179
176
|
const rows = documents.map((doc, i) => ({
|
|
@@ -190,6 +187,23 @@ class DatabricksVectorStore extends vectorstores_1.VectorStore {
|
|
|
190
187
|
}
|
|
191
188
|
return ids;
|
|
192
189
|
}
|
|
190
|
+
assertWritableContentColumn() {
|
|
191
|
+
const vectorColumn = this.assertDirectAccess();
|
|
192
|
+
const { primaryKey, name, schemaColumns } = this.index;
|
|
193
|
+
if (this.contentColumn === primaryKey) {
|
|
194
|
+
throw new n8n_workflow_1.UserError(`Column ${primaryKey} is the primary key of ${name}. Select another Content Column`);
|
|
195
|
+
}
|
|
196
|
+
if (this.contentColumn === vectorColumn) {
|
|
197
|
+
throw new n8n_workflow_1.UserError(`Column ${vectorColumn} holds the embedding vector of ${name}. Select another Content Column`);
|
|
198
|
+
}
|
|
199
|
+
if (schemaColumns !== undefined && !schemaColumns.includes(this.contentColumn)) {
|
|
200
|
+
const selectable = schemaColumns.filter((column) => !this.reserved.has(column));
|
|
201
|
+
throw new n8n_workflow_1.UserError(selectable.length > 0
|
|
202
|
+
? `Index ${name} has no column ${this.contentColumn}. Select a Content Column from: ${selectable.join(', ')}`
|
|
203
|
+
: `Index ${name} has no column ${this.contentColumn}, and declares no other column to select`);
|
|
204
|
+
}
|
|
205
|
+
return vectorColumn;
|
|
206
|
+
}
|
|
193
207
|
assertDirectAccess() {
|
|
194
208
|
const { indexType, embedding, name } = this.index;
|
|
195
209
|
if (indexType !== 'DIRECT_ACCESS' || embedding.kind !== 'self') {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"DatabricksVectorStore.js","sourceRoot":"","sources":["../../../../nodes/vector_store/VectorStoreDatabricks/DatabricksVectorStore.ts"],"names":[],"mappings":";;;;AAAA,6CAAyC;AAEzC,yDAA6E;AAE7E,+DAA2D;AAC3D,oDAAgD;AAChD,+CAA0E;AAsC1E,SAAS,eAAe,CAAC,OAAe;IAEvC,OAAO,OAAO,CAAC,OAAO,CAAC,mBAAmB,EAAE,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;AAChE,CAAC;AAED,MAAM,SAAS,GAAG,6BAA6B,CAAC;AAIhD,SAAS,cAAc,CACtB,KAAa,EACb,IAAY,EACZ,GAA+C;IAE/C,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;QAC5B,MAAM,IAAI,GAAG,CACZ,sBAAsB,IAAI,KAAK,eAAe,CAAC,KAAK,CAAC,yBAAyB,IAAI,EAAE,CACpF,CAAC;IACH,CAAC;AACF,CAAC;AAED,MAAM,cAAc,GAAG,CAAC,KAAc,EAAE,EAAE,CAAC,CAAC,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;AAC3F,MAAM,OAAO,GAAG,CAAC,KAAc,EAAkC,EAAE,CAClE,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,oBAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;AACpD,MAAM,WAAW,GAAG,CAAC,KAAc,EAAwB,EAAE,CAC5D,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,EAAkB,EAAE,CAAC,OAAO,IAAI,KAAK,QAAQ,CAAC;IACtF,CAAC,CAAC,KAAK;IACP,CAAC,CAAC,SAAS,CAAC;AAEd,wBAA+B,GAAY;IAC1C,MAAM,SAAS,GAAG,IAAA,oBAAQ,EAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;IAC7D,IACC,CAAC,IAAA,oBAAQ,EAAC,GAAG,CAAC;QACd,OAAO,GAAG,CAAC,IAAI,KAAK,QAAQ;QAC5B,OAAO,GAAG,CAAC,WAAW,KAAK,QAAQ;QACnC,CAAC,SAAS,KAAK,YAAY,IAAI,SAAS,KAAK,eAAe,CAAC,EAC5D,CAAC;QACF,MAAM,IAAI,+BAAgB,CAAC,yCAAyC,CAAC,CAAC;IACvE,CAAC;IAED,MAAM,SAAS,GAAG,IAAA,oBAAQ,EAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC,CAAC,SAAS,CAAC;IAC9F,MAAM,UAAU,GAAG,IAAA,oBAAQ,EAAC,GAAG,CAAC,wBAAwB,CAAC;QACxD,CAAC,CAAC,GAAG,CAAC,wBAAwB;QAC9B,CAAC,CAAC,SAAS,CAAC;IACb,MAAM,IAAI,GAAG,SAAS,IAAI,UAAU,IAAI,EAAE,CAAC;IAC3C,MAAM,CAAC,YAAY,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;IAC9D,MAAM,CAAC,YAAY,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;IAC9D,MAAM,UAAU,GAAG,cAAc,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;IACtD,MAAM,UAAU,GAAG,cAAc,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;IACtD,MAAM,SAAS,GAAiD,UAAU;QACzE,CAAC,CAAC;YACA,IAAI,EAAE,SAAS;YACf,YAAY,EAAE,UAAU;YACxB,aAAa,EAAE,cAAc,CAAC,YAAY,EAAE,6BAA6B,CAAC;SAC1E;QACF,CAAC,CAAC,UAAU;YACX,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,YAAY,EAAE,UAAU,EAAE;YAC5C,CAAC,CAAC,SAAS,CAAC;IACd,IAAI,CAAC,SAAS;QAAE,MAAM,IAAI,+BAAgB,CAAC,yCAAyC,CAAC,CAAC;IAEtF,IAAI,aAAmC,CAAC;IACxC,IAAI,OAAO,UAAU,EAAE,WAAW,KAAK,QAAQ,EAAE,CAAC;QACjD,IAAI,MAAe,CAAC;QACpB,IAAI,CAAC;YACJ,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;QAC7C,CAAC;QAAC,MAAM,CAAC;YACR,MAAM,IAAI,+BAAgB,CAAC,yCAAyC,CAAC,CAAC;QACvE,CAAC;QACD,aAAa,GAAG,IAAA,oBAAQ,EAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACpE,CAAC;SAAM,CAAC;QAEP,aAAa,GAAG,WAAW,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC;IACzD,CAAC;IAED,OAAO;QACN,IAAI,EAAE,GAAG,CAAC,IAAI;QACd,UAAU,EAAE,GAAG,CAAC,WAAW;QAC3B,SAAS;QACT,SAAS;QACT,aAAa;QACb,WAAW,EAAE,cAAc,CAAC,SAAS,EAAE,YAAY,CAAC;KACpD,CAAC;AACH,CAAC;AAED,KAAK,UAAU,iBAAiB,CAC/B,OAAc,EACd,GAAW,EACX,IAA2C;IAE3C,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,GAAG,EAAE;QACnC,GAAG,IAAI;QACP,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;KAC/C,CAAC,CAAC;IACH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QAElB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACnC,IAAI,OAAO,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC;YACJ,MAAM,MAAM,GAAY,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACzC,IAAI,IAAA,oBAAQ,EAAC,MAAM,CAAC,IAAI,OAAO,MAAM,CAAC,OAAO,KAAK,QAAQ;gBAAE,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QACtF,CAAC;QAAC,MAAM,CAAC,CAAA,CAAC;QACV,MAAM,IAAI,+BAAgB,CACzB,4CAA4C,QAAQ,CAAC,MAAM,MAAM,eAAe,CAAC,OAAO,CAAC,EAAE,CAC3F,CAAC;IACH,CAAC;IACD,OAAO,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;AAC9B,CAAC;AAED,2BAAmC,SAAQ,0BAAW;IAoBrD,MAAM,CAAC,KAAK,CAAC,aAAa,CACzB,OAAc,EACd,IAAY,EACZ,SAAiB;QAEjB,cAAc,CAAC,SAAS,EAAE,OAAO,EAAE,wBAAS,CAAC,CAAC;QAC9C,MAAM,IAAI,GAAG,cAAc,CAC1B,MAAM,iBAAiB,CACtB,OAAO,EACP,GAAG,IAAI,kCAAkC,IAAA,4BAAa,EAAC,SAAS,CAAC,EAAE,CACnE,CACD,CAAC;QAEF,IAAI,IAAI,CAAC,SAAS,KAAK,YAAY,IAAI,CAAC,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YAChF,cAAc,CAAC,IAAI,CAAC,WAAW,EAAE,OAAO,EAAE,+BAAgB,CAAC,CAAC;YAC5D,MAAM,QAAQ,GAAG,MAAM,OAAO,CAC7B,GAAG,IAAI,iCAAiC,IAAA,4BAAa,EAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CACzE,CAAC;YAEF,IAAI,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,KAAK,GAAY,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBAC7C,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,IAAA,oBAAQ,EAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,CAChF,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAClE,CAAC;YACH,CAAC;iBAAM,CAAC;gBACP,MAAM,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC;YAC/B,CAAC;QACF,CAAC;QACD,OAAO,IAAI,CAAC;IACb,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAC7B,UAA+B,EAC/B,MAAmC;QAInC,MAAM,KAAK,GACV,MAAM,CAAC,KAAK;YACZ,CAAC,MAAM,qBAAqB,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;QAC1F,OAAO,IAAI,qBAAqB,CAAC,UAAU,EAAE,EAAE,GAAG,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;IACpE,CAAC;IAED,YACC,UAA+B,EAC/B,MAAoE;QAEpE,KAAK,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;QAC1B,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC;QACzB,MAAM,aAAa,GAClB,MAAM,CAAC,aAAa;YACpB,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QACjF,IAAI,CAAC,aAAa,EAAE,CAAC;YACpB,MAAM,IAAI,wBAAS,CAClB,SAAS,KAAK,CAAC,IAAI,wDAAwD,CAC3E,CAAC;QACH,CAAC;QACD,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAC1B,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;QACxB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,QAAQ,GAAG,IAAI,GAAG,CAAC;YACvB,KAAK,CAAC,UAAU;YAChB,aAAa;YACb,KAAK,CAAC,SAAS,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS;SAC1E,CAAC,CAAC;QACH,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC,eAAe,EAAE,MAAM;YACpD,CAAC,CAAC,MAAM,CAAC,eAAe;YACxB,CAAC,CAAC,CAAC,KAAK,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;QAC9E,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,KAAK,CAAC;QAC3C,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC,MAAM,CAAC;IACpC,CAAC;IAED,gBAAgB;QACf,OAAO,YAAY,CAAC;IACrB,CAAC;IAED,IAAY,SAAS;QACpB,OAAO,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,KAAK,SAAS,CAAC;IAChD,CAAC;IAGD,KAAK,CAAC,yBAAyB,CAC9B,KAAa,EACb,CAAC,GAAG,CAAC,EACL,MAA2B;QAE3B,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,OAAO,MAAM,IAAI,CAAC,KAAK,CACtB,EAAE,WAAW,EAAE,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,SAAS,EAAE,UAAU,EAAE,KAAK,EAAE,EACjE,MAAM,CACN,CAAC;QACH,CAAC;QACD,OAAO,MAAM,IAAI,CAAC,KAAK,CACtB;YACC,WAAW,EAAE,CAAC;YACd,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,YAAY,EAAE,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC;YAErD,GAAG,CAAC,IAAI,CAAC,SAAS,KAAK,QAAQ,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC;SACzD,EACD,MAAM,CACN,CAAC;IACH,CAAC;IAGD,KAAK,CAAC,gBAAgB,CAAC,KAAa,EAAE,CAAC,GAAG,CAAC,EAAE,MAA2B;QACvE,OAAO,CAAC,MAAM,IAAI,CAAC,yBAAyB,CAAC,KAAK,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;IACrF,CAAC;IAED,KAAK,CAAC,+BAA+B,CACpC,KAAe,EACf,CAAS,EACT,MAA2B;QAE3B,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,MAAM,IAAI,wBAAS,CAClB,SAAS,IAAI,CAAC,KAAK,CAAC,IAAI,mEAAmE,CAC3F,CAAC;QACH,CAAC;QAED,OAAO,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,UAAU,EAAE,KAAK,EAAE,YAAY,EAAE,KAAK,EAAE,EAAE,MAAM,CAAC,CAAC;IAC7F,CAAC;IAED,KAAK,CAAC,YAAY,CACjB,SAA8B,EAC9B,OAA4B;QAE5B,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QACtC,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC;QAC9F,OAAO,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;IAC3D,CAAC;IAED,KAAK,CAAC,UAAU,CACf,OAAmB,EACnB,SAA8B,EAC9B,OAA4B;QAE5B,MAAM,YAAY,GAAG,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC/C,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC;QAExC,IAAI,IAAI,CAAC,aAAa,KAAK,UAAU,EAAE,CAAC;YACvC,MAAM,IAAI,wBAAS,CAClB,UAAU,UAAU,0BAA0B,IAAI,iCAAiC,CACnF,CAAC;QACH,CAAC;QACD,MAAM,aAAa,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC;QAC9D,MAAM,GAAG,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,EAAE,IAAI,IAAA,wBAAU,GAAE,CAAC,CAAC;QAGnF,MAAM,IAAI,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YACvC,GAAG,MAAM,CAAC,WAAW,CACpB,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,MAAM,CAClC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,CAC5D,CACD;YACD,CAAC,UAAU,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;YACpB,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE,GAAG,CAAC,WAAW;YACrC,CAAC,YAAY,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC;SAC1B,CAAC,CAAC,CAAC;QAEJ,MAAM,QAAQ,GAAG,MAAM,iBAAiB,CACvC,IAAI,CAAC,KAAK,EACV,GAAG,IAAI,CAAC,IAAI,kCAAkC,IAAA,4BAAa,EAAC,IAAI,CAAC,cAAc,EAC/E,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAC/E,CAAC;QACF,IAAI,CAAC,IAAA,oBAAQ,EAAC,QAAQ,CAAC,IAAI,QAAQ,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YAC1D,MAAM,MAAM,GAAG,IAAA,oBAAQ,EAAC,QAAQ,CAAC,IAAI,IAAA,oBAAQ,EAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;YACtF,MAAM,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC,mBAAmB,CAAC,IAAI,EAAE,CAAC;YAC7D,MAAM,IAAI,+BAAgB,CACzB,uCAAuC,IAAI,0BAA0B,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACxF,CAAC;QACH,CAAC;QACD,OAAO,GAAG,CAAC;IACZ,CAAC;IAGO,kBAAkB;QACzB,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC;QAElD,IAAI,SAAS,KAAK,eAAe,IAAI,SAAS,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAChE,MAAM,IAAI,wBAAS,CAClB,SAAS,IAAI,6EAA6E,CAC1F,CAAC;QACH,CAAC;QACD,OAAO,SAAS,CAAC,YAAY,CAAC;IAC/B,CAAC;IAEO,KAAK,CAAC,KAAK,CAClB,IAA6B,EAC7B,MAAgC;QAEhC,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC;QACxC,MAAM,OAAO,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,UAAU,EAAE,IAAI,CAAC,aAAa,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACxF,MAAM,eAAe,GAAG,MAAM,IAAI,IAAI,CAAC,aAAa,CAAC;QACrD,MAAM,WAAW,GAChB,eAAe,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,MAAM,GAAG,CAAC;YACzD,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC;YACjC,CAAC,CAAC,SAAS,CAAC;QAEd,MAAM,QAAQ,GAAG,MAAM,iBAAiB,CACvC,IAAI,CAAC,KAAK,EACV,GAAG,IAAI,CAAC,IAAI,kCAAkC,IAAA,4BAAa,EAAC,IAAI,CAAC,QAAQ,EACzE;YACC,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACpB,GAAG,IAAI;gBACP,OAAO;gBACP,GAAG,CAAC,WAAW,IAAI,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC;aACjD,CAAC;SACF,CACD,CAAC;QAGF,MAAM,QAAQ,GAAG,IAAA,oBAAQ,EAAC,QAAQ,CAAC,IAAI,IAAA,oBAAQ,EAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;QAC5F,MAAM,MAAM,GAAG,IAAA,oBAAQ,EAAC,QAAQ,CAAC,IAAI,IAAA,oBAAQ,EAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;QACtF,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QAC3F,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC;YAC5C,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,GAAG,EAAoB,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YACzE,CAAC,CAAC,EAAE,CAAC;QACN,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAEjC,MAAM,aAAa,GAAG,CAAC,IAAY,EAAE,EAAE,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE;YAChE,IAAI,EAAE,GAAG,CAAC,EAAE,CAAC;gBACZ,MAAM,IAAI,+BAAgB,CAAC,2CAA2C,IAAI,GAAG,CAAC,CAAC;YAChF,CAAC;YACD,OAAO,EAAE,CAAC;QACX,CAAC,CAAC;QACF,MAAM,SAAS,GAAG,aAAa,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACpD,MAAM,IAAI,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC;QAIvC,MAAM,OAAO,GAAG,aAAa,CAAC,OAAO,EAAE,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC;QACnE,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAC1C,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,aAAa,CAAC,MAAM,CAAC,CAAU,CACpD,CAAC;QAEF,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC;YACxB,IAAI,oBAAQ,CAAC;gBACZ,WAAW,EAAE,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;gBACzC,QAAQ,EAAE,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;gBACjF,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;aACrB,CAAC;YACF,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;SACpB,CAAC,CAAC;IACJ,CAAC;CACD"}
|
|
1
|
+
{"version":3,"file":"DatabricksVectorStore.js","sourceRoot":"","sources":["../../../../nodes/vector_store/VectorStoreDatabricks/DatabricksVectorStore.ts"],"names":[],"mappings":";;;;AAAA,6CAAyC;AAEzC,yDAA6E;AAE7E,+DAA2D;AAC3D,oDAAgD;AAChD,+CAA0E;AAsC1E,SAAS,eAAe,CAAC,OAAe;IAEvC,OAAO,OAAO,CAAC,OAAO,CAAC,mBAAmB,EAAE,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;AAChE,CAAC;AAED,MAAM,SAAS,GAAG,6BAA6B,CAAC;AAIhD,SAAS,cAAc,CACtB,KAAa,EACb,IAAY,EACZ,GAA+C;IAE/C,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;QAC5B,MAAM,IAAI,GAAG,CACZ,sBAAsB,IAAI,KAAK,eAAe,CAAC,KAAK,CAAC,yBAAyB,IAAI,EAAE,CACpF,CAAC;IACH,CAAC;AACF,CAAC;AAED,MAAM,cAAc,GAAG,CAAC,KAAc,EAAE,EAAE,CAAC,CAAC,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;AAC3F,MAAM,OAAO,GAAG,CAAC,KAAc,EAAkC,EAAE,CAClE,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,oBAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;AACpD,MAAM,WAAW,GAAG,CAAC,KAAc,EAAwB,EAAE,CAC5D,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,EAAkB,EAAE,CAAC,OAAO,IAAI,KAAK,QAAQ,CAAC;IACtF,CAAC,CAAC,KAAK;IACP,CAAC,CAAC,SAAS,CAAC;AAEd,wBAA+B,GAAY;IAC1C,MAAM,SAAS,GAAG,IAAA,oBAAQ,EAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;IAC7D,IACC,CAAC,IAAA,oBAAQ,EAAC,GAAG,CAAC;QACd,OAAO,GAAG,CAAC,IAAI,KAAK,QAAQ;QAC5B,OAAO,GAAG,CAAC,WAAW,KAAK,QAAQ;QACnC,CAAC,SAAS,KAAK,YAAY,IAAI,SAAS,KAAK,eAAe,CAAC,EAC5D,CAAC;QACF,MAAM,IAAI,+BAAgB,CAAC,yCAAyC,CAAC,CAAC;IACvE,CAAC;IAED,MAAM,SAAS,GAAG,IAAA,oBAAQ,EAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC,CAAC,SAAS,CAAC;IAC9F,MAAM,UAAU,GAAG,IAAA,oBAAQ,EAAC,GAAG,CAAC,wBAAwB,CAAC;QACxD,CAAC,CAAC,GAAG,CAAC,wBAAwB;QAC9B,CAAC,CAAC,SAAS,CAAC;IACb,MAAM,IAAI,GAAG,SAAS,IAAI,UAAU,IAAI,EAAE,CAAC;IAC3C,MAAM,CAAC,YAAY,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;IAC9D,MAAM,CAAC,YAAY,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;IAC9D,MAAM,UAAU,GAAG,cAAc,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;IACtD,MAAM,UAAU,GAAG,cAAc,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;IACtD,MAAM,SAAS,GAAiD,UAAU;QACzE,CAAC,CAAC;YACA,IAAI,EAAE,SAAS;YACf,YAAY,EAAE,UAAU;YACxB,aAAa,EAAE,cAAc,CAAC,YAAY,EAAE,6BAA6B,CAAC;SAC1E;QACF,CAAC,CAAC,UAAU;YACX,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,YAAY,EAAE,UAAU,EAAE;YAC5C,CAAC,CAAC,SAAS,CAAC;IACd,IAAI,CAAC,SAAS;QAAE,MAAM,IAAI,+BAAgB,CAAC,yCAAyC,CAAC,CAAC;IAEtF,IAAI,aAAmC,CAAC;IACxC,IAAI,OAAO,UAAU,EAAE,WAAW,KAAK,QAAQ,EAAE,CAAC;QACjD,IAAI,MAAe,CAAC;QACpB,IAAI,CAAC;YACJ,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;QAC7C,CAAC;QAAC,MAAM,CAAC;YACR,MAAM,IAAI,+BAAgB,CAAC,yCAAyC,CAAC,CAAC;QACvE,CAAC;QACD,aAAa,GAAG,IAAA,oBAAQ,EAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACpE,CAAC;SAAM,CAAC;QAEP,aAAa,GAAG,WAAW,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC;IACzD,CAAC;IAED,OAAO;QACN,IAAI,EAAE,GAAG,CAAC,IAAI;QACd,UAAU,EAAE,GAAG,CAAC,WAAW;QAC3B,SAAS;QACT,SAAS;QACT,aAAa;QACb,WAAW,EAAE,cAAc,CAAC,SAAS,EAAE,YAAY,CAAC;KACpD,CAAC;AACH,CAAC;AAED,KAAK,UAAU,iBAAiB,CAC/B,OAAc,EACd,GAAW,EACX,IAA2C;IAE3C,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,GAAG,EAAE;QACnC,GAAG,IAAI;QACP,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;KAC/C,CAAC,CAAC;IACH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QAElB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACnC,IAAI,OAAO,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC;YACJ,MAAM,MAAM,GAAY,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACzC,IAAI,IAAA,oBAAQ,EAAC,MAAM,CAAC,IAAI,OAAO,MAAM,CAAC,OAAO,KAAK,QAAQ;gBAAE,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QACtF,CAAC;QAAC,MAAM,CAAC,CAAA,CAAC;QACV,MAAM,IAAI,+BAAgB,CACzB,4CAA4C,QAAQ,CAAC,MAAM,MAAM,eAAe,CAAC,OAAO,CAAC,EAAE,CAC3F,CAAC;IACH,CAAC;IACD,OAAO,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;AAC9B,CAAC;AAED,2BAAmC,SAAQ,0BAAW;IAoBrD,MAAM,CAAC,KAAK,CAAC,aAAa,CACzB,OAAc,EACd,IAAY,EACZ,SAAiB;QAEjB,cAAc,CAAC,SAAS,EAAE,OAAO,EAAE,wBAAS,CAAC,CAAC;QAC9C,MAAM,IAAI,GAAG,cAAc,CAC1B,MAAM,iBAAiB,CACtB,OAAO,EACP,GAAG,IAAI,kCAAkC,IAAA,4BAAa,EAAC,SAAS,CAAC,EAAE,CACnE,CACD,CAAC;QAEF,IAAI,IAAI,CAAC,SAAS,KAAK,YAAY,IAAI,CAAC,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YAChF,cAAc,CAAC,IAAI,CAAC,WAAW,EAAE,OAAO,EAAE,+BAAgB,CAAC,CAAC;YAC5D,MAAM,QAAQ,GAAG,MAAM,OAAO,CAC7B,GAAG,IAAI,iCAAiC,IAAA,4BAAa,EAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CACzE,CAAC;YAEF,IAAI,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,KAAK,GAAY,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBAC7C,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,IAAA,oBAAQ,EAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,CAChF,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAClE,CAAC;YACH,CAAC;iBAAM,CAAC;gBACP,MAAM,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC;YAC/B,CAAC;QACF,CAAC;QACD,OAAO,IAAI,CAAC;IACb,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAC7B,UAA+B,EAC/B,MAAmC;QAInC,MAAM,KAAK,GACV,MAAM,CAAC,KAAK;YACZ,CAAC,MAAM,qBAAqB,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;QAC1F,OAAO,IAAI,qBAAqB,CAAC,UAAU,EAAE,EAAE,GAAG,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;IACpE,CAAC;IAED,YACC,UAA+B,EAC/B,MAAoE;QAEpE,KAAK,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;QAC1B,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC;QACzB,MAAM,aAAa,GAClB,MAAM,CAAC,aAAa;YACpB,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QACjF,IAAI,CAAC,aAAa,EAAE,CAAC;YACpB,MAAM,IAAI,wBAAS,CAClB,SAAS,KAAK,CAAC,IAAI,wDAAwD,CAC3E,CAAC;QACH,CAAC;QACD,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAC1B,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;QACxB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,QAAQ,GAAG,IAAI,GAAG,CAAC;YACvB,KAAK,CAAC,UAAU;YAChB,aAAa;YACb,KAAK,CAAC,SAAS,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS;SAC1E,CAAC,CAAC;QACH,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC,eAAe,EAAE,MAAM;YACpD,CAAC,CAAC,MAAM,CAAC,eAAe;YACxB,CAAC,CAAC,CAAC,KAAK,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;QAC9E,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,KAAK,CAAC;QAC3C,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC,MAAM,CAAC;IACpC,CAAC;IAED,gBAAgB;QACf,OAAO,YAAY,CAAC;IACrB,CAAC;IAED,IAAY,SAAS;QACpB,OAAO,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,KAAK,SAAS,CAAC;IAChD,CAAC;IAGD,KAAK,CAAC,yBAAyB,CAC9B,KAAa,EACb,CAAC,GAAG,CAAC,EACL,MAA2B;QAE3B,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,OAAO,MAAM,IAAI,CAAC,KAAK,CACtB,EAAE,WAAW,EAAE,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,SAAS,EAAE,UAAU,EAAE,KAAK,EAAE,EACjE,MAAM,CACN,CAAC;QACH,CAAC;QACD,OAAO,MAAM,IAAI,CAAC,KAAK,CACtB;YACC,WAAW,EAAE,CAAC;YACd,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,YAAY,EAAE,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC;YAErD,GAAG,CAAC,IAAI,CAAC,SAAS,KAAK,QAAQ,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC;SACzD,EACD,MAAM,CACN,CAAC;IACH,CAAC;IAGD,KAAK,CAAC,gBAAgB,CAAC,KAAa,EAAE,CAAC,GAAG,CAAC,EAAE,MAA2B;QACvE,OAAO,CAAC,MAAM,IAAI,CAAC,yBAAyB,CAAC,KAAK,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;IACrF,CAAC;IAED,KAAK,CAAC,+BAA+B,CACpC,KAAe,EACf,CAAS,EACT,MAA2B;QAE3B,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,MAAM,IAAI,wBAAS,CAClB,SAAS,IAAI,CAAC,KAAK,CAAC,IAAI,mEAAmE,CAC3F,CAAC;QACH,CAAC;QAED,OAAO,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,UAAU,EAAE,KAAK,EAAE,YAAY,EAAE,KAAK,EAAE,EAAE,MAAM,CAAC,CAAC;IAC7F,CAAC;IAED,KAAK,CAAC,YAAY,CACjB,SAA8B,EAC9B,OAA4B;QAE5B,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QACtC,IAAI,CAAC,2BAA2B,EAAE,CAAC;QACnC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC;QAC9F,OAAO,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;IAC3D,CAAC;IAED,KAAK,CAAC,UAAU,CACf,OAAmB,EACnB,SAA8B,EAC9B,OAA4B;QAE5B,MAAM,YAAY,GAAG,IAAI,CAAC,2BAA2B,EAAE,CAAC;QACxD,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC;QACxC,MAAM,aAAa,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC;QAC9D,MAAM,GAAG,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,EAAE,IAAI,IAAA,wBAAU,GAAE,CAAC,CAAC;QAGnF,MAAM,IAAI,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YACvC,GAAG,MAAM,CAAC,WAAW,CACpB,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,MAAM,CAClC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,CAC5D,CACD;YACD,CAAC,UAAU,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;YACpB,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE,GAAG,CAAC,WAAW;YACrC,CAAC,YAAY,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC;SAC1B,CAAC,CAAC,CAAC;QAEJ,MAAM,QAAQ,GAAG,MAAM,iBAAiB,CACvC,IAAI,CAAC,KAAK,EACV,GAAG,IAAI,CAAC,IAAI,kCAAkC,IAAA,4BAAa,EAAC,IAAI,CAAC,cAAc,EAC/E,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAC/E,CAAC;QACF,IAAI,CAAC,IAAA,oBAAQ,EAAC,QAAQ,CAAC,IAAI,QAAQ,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YAC1D,MAAM,MAAM,GAAG,IAAA,oBAAQ,EAAC,QAAQ,CAAC,IAAI,IAAA,oBAAQ,EAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;YACtF,MAAM,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC,mBAAmB,CAAC,IAAI,EAAE,CAAC;YAC7D,MAAM,IAAI,+BAAgB,CACzB,uCAAuC,IAAI,0BAA0B,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACxF,CAAC;QACH,CAAC;QACD,OAAO,GAAG,CAAC;IACZ,CAAC;IAGO,2BAA2B;QAClC,MAAM,YAAY,GAAG,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC/C,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC;QACvD,IAAI,IAAI,CAAC,aAAa,KAAK,UAAU,EAAE,CAAC;YACvC,MAAM,IAAI,wBAAS,CAClB,UAAU,UAAU,0BAA0B,IAAI,iCAAiC,CACnF,CAAC;QACH,CAAC;QACD,IAAI,IAAI,CAAC,aAAa,KAAK,YAAY,EAAE,CAAC;YACzC,MAAM,IAAI,wBAAS,CAClB,UAAU,YAAY,kCAAkC,IAAI,iCAAiC,CAC7F,CAAC;QACH,CAAC;QAGD,IAAI,aAAa,KAAK,SAAS,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC;YAChF,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;YAChF,MAAM,IAAI,wBAAS,CAClB,UAAU,CAAC,MAAM,GAAG,CAAC;gBACpB,CAAC,CAAC,SAAS,IAAI,kBAAkB,IAAI,CAAC,aAAa,mCAAmC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;gBAC7G,CAAC,CAAC,SAAS,IAAI,kBAAkB,IAAI,CAAC,aAAa,0CAA0C,CAC9F,CAAC;QACH,CAAC;QACD,OAAO,YAAY,CAAC;IACrB,CAAC;IAGO,kBAAkB;QACzB,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC;QAElD,IAAI,SAAS,KAAK,eAAe,IAAI,SAAS,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAChE,MAAM,IAAI,wBAAS,CAClB,SAAS,IAAI,6EAA6E,CAC1F,CAAC;QACH,CAAC;QACD,OAAO,SAAS,CAAC,YAAY,CAAC;IAC/B,CAAC;IAEO,KAAK,CAAC,KAAK,CAClB,IAA6B,EAC7B,MAAgC;QAEhC,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC;QACxC,MAAM,OAAO,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,UAAU,EAAE,IAAI,CAAC,aAAa,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACxF,MAAM,eAAe,GAAG,MAAM,IAAI,IAAI,CAAC,aAAa,CAAC;QACrD,MAAM,WAAW,GAChB,eAAe,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,MAAM,GAAG,CAAC;YACzD,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC;YACjC,CAAC,CAAC,SAAS,CAAC;QAEd,MAAM,QAAQ,GAAG,MAAM,iBAAiB,CACvC,IAAI,CAAC,KAAK,EACV,GAAG,IAAI,CAAC,IAAI,kCAAkC,IAAA,4BAAa,EAAC,IAAI,CAAC,QAAQ,EACzE;YACC,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACpB,GAAG,IAAI;gBACP,OAAO;gBACP,GAAG,CAAC,WAAW,IAAI,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC;aACjD,CAAC;SACF,CACD,CAAC;QAGF,MAAM,QAAQ,GAAG,IAAA,oBAAQ,EAAC,QAAQ,CAAC,IAAI,IAAA,oBAAQ,EAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;QAC5F,MAAM,MAAM,GAAG,IAAA,oBAAQ,EAAC,QAAQ,CAAC,IAAI,IAAA,oBAAQ,EAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;QACtF,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QAC3F,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC;YAC5C,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,GAAG,EAAoB,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YACzE,CAAC,CAAC,EAAE,CAAC;QACN,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAEjC,MAAM,aAAa,GAAG,CAAC,IAAY,EAAE,EAAE,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE;YAChE,IAAI,EAAE,GAAG,CAAC,EAAE,CAAC;gBACZ,MAAM,IAAI,+BAAgB,CAAC,2CAA2C,IAAI,GAAG,CAAC,CAAC;YAChF,CAAC;YACD,OAAO,EAAE,CAAC;QACX,CAAC,CAAC;QACF,MAAM,SAAS,GAAG,aAAa,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACpD,MAAM,IAAI,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC;QAIvC,MAAM,OAAO,GAAG,aAAa,CAAC,OAAO,EAAE,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC;QACnE,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAC1C,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,aAAa,CAAC,MAAM,CAAC,CAAU,CACpD,CAAC;QAEF,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC;YACxB,IAAI,oBAAQ,CAAC;gBACZ,WAAW,EAAE,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;gBACzC,QAAQ,EAAE,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;gBACjF,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;aACrB,CAAC;YACF,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;SACpB,CAAC,CAAC;IACJ,CAAC;CACD"}
|
|
@@ -164,13 +164,16 @@ async function createStore(ctx, embeddings, itemIndex, filter) {
|
|
|
164
164
|
});
|
|
165
165
|
}
|
|
166
166
|
async function searchIndexes(filter) {
|
|
167
|
+
const insertOnly = this.getCurrentNodeParameter('mode') === 'insert';
|
|
167
168
|
const credentials = await this.getCredentials(token_provider_1.DATABRICKS_CREDENTIAL_TYPE);
|
|
168
169
|
(0, constants_1.assertHttpsHost)(this, credentials.host);
|
|
169
170
|
const host = credentials.host.replace(/\/$/, '');
|
|
170
171
|
const endpoints = await (0, list_pages_1.listDatabricksPages)(this, `${host}/api/2.0/vector-search/endpoints`, {}, (page) => page.endpoints);
|
|
171
172
|
const results = (await Promise.all(endpoints.map(async (endpoint) => {
|
|
172
173
|
const indexes = await (0, list_pages_1.listDatabricksPages)(this, `${host}/api/2.0/vector-search/indexes`, { endpoint_name: endpoint.name }, (page) => page.vector_indexes);
|
|
173
|
-
return indexes
|
|
174
|
+
return indexes
|
|
175
|
+
.filter((index) => !insertOnly || index.index_type === 'DIRECT_ACCESS')
|
|
176
|
+
.map((index) => ({
|
|
174
177
|
name: index.name,
|
|
175
178
|
value: index.name,
|
|
176
179
|
description: `${index.index_type} - ${endpoint.name}`,
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"VectorStoreDatabricks.node.js","sourceRoot":"","sources":["../../../../nodes/vector_store/VectorStoreDatabricks/VectorStoreDatabricks.node.ts"],"names":[],"mappings":";;;AACA,oDAAsE;AACtE,+CASsB;AAEtB,6DAGsC;AACtC,2DAA8D;AAC9D,6DAAmE;AACnE,qEAG0C;AAC1C,mEAA0F;AAE1F,MAAM,kBAAkB,GAAG,MAAM,CAAC;AAElC,MAAM,kBAAkB,GAAoB;IAC3C,WAAW,EAAE,OAAO;IACpB,IAAI,EAAE,iBAAiB;IACvB,IAAI,EAAE,iBAAiB;IACvB,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE;IACpC,QAAQ,EAAE,IAAI;IACd,KAAK,EAAE;QACN;YACC,WAAW,EAAE,WAAW;YACxB,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,MAAM;YACZ,WAAW,EAAE,oBAAoB;YACjC,WAAW,EAAE;gBACZ,gBAAgB,EAAE,eAAe;gBACjC,UAAU,EAAE,IAAI;aAChB;SACD;QACD;YACC,WAAW,EAAE,MAAM;YACnB,IAAI,EAAE,IAAI;YACV,IAAI,EAAE,QAAQ;YACd,WAAW,EAAE,sBAAsB;YACnC,UAAU,EAAE;gBACX;oBACC,IAAI,EAAE,OAAO;oBACb,UAAU,EAAE;wBACX,KAAK,EAAE,kCAAkC;wBACzC,YAAY,EAAE,yCAAyC;qBACvD;iBACD;aACD;SACD;KACD;CACD,CAAC;AAEF,MAAM,iBAAiB,GAAG;IACzB,iBAAiB,EAAE,iBAAiB;IACpC,oBAAoB,EAAE,CAAC,uBAAuB,CAAC;CAC/C,CAAC;AAEF,MAAM,YAAY,GAAsB;IACvC,kBAAkB;IAClB;QACC,WAAW,EAAE,gBAAgB;QAC7B,IAAI,EAAE,eAAe;QACrB,IAAI,EAAE,SAAS;QACf,OAAO,EAAE,EAAE;QACX,WAAW,EAAE,iBAAiB;QAC9B,WAAW,EACV,qOAAqO;KACtO;IACD;QACC,WAAW,EAAE,SAAS;QACtB,IAAI,EAAE,SAAS;QACf,IAAI,EAAE,YAAY;QAClB,WAAW,EAAE,YAAY;QACzB,OAAO,EAAE,EAAE;QAEX,cAAc,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,CAAC,MAAM,EAAE,UAAU,EAAE,kBAAkB,CAAC,EAAE,EAAE;QAC5E,OAAO,EAAE;YACR;gBACC,WAAW,EAAE,kBAAkB;gBAC/B,IAAI,EAAE,iBAAiB;gBACvB,IAAI,EAAE,cAAc;gBACpB,OAAO,EAAE,EAAE;gBACX,WAAW,EAAE,iBAAiB;gBAC9B,WAAW,EACV,2NAA2N;aAC5N;YACD;gBACC,WAAW,EAAE,aAAa;gBAC1B,IAAI,EAAE,YAAY;gBAClB,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,KAAK;gBACd,OAAO,EAAE;oBACR,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE;oBAChC,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE;iBACnC;gBACD,WAAW,EACV,2IAA2I;aAC5I;YACD;gBACC,WAAW,EAAE,iBAAiB;gBAC9B,IAAI,EAAE,kBAAkB;gBACxB,IAAI,EAAE,MAAM;gBACZ,WAAW,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE;gBACxB,OAAO,EAAE,IAAI;gBACb,YAAY,EAAE,QAAQ;gBACtB,WAAW,EACV,sHAAsH;aACvH;SACD;KACD;CACD,CAAC;AAWF,KAAK,UAAU,OAAO,CAAC,GAA2B;IACjD,MAAM,UAAU,GAAG,MAAM,GAAG,CAAC,cAAc,CAC1C,2CAA0B,CAC1B,CAAC;IACF,IAAA,2BAAe,EAAC,GAAG,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC;IACtC,MAAM,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IAChD,MAAM,YAAY,GAAG,GAAG,CAAC,OAAO,CAAC,qBAAqB,EAAE,CAAC;IAEzD,MAAM,YAAY,GACjB,0BAA0B,IAAI,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,wBAAwB,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IAChF,MAAM,EAAE,KAAK,EAAE,GAAG,IAAA,sCAAyB,EAAC,GAAG,EAAE,UAAU,EAAE;QAC5D,WAAW,EAAE,IAAI;QACjB,YAAY;QACZ,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YAIhC,MAAM,OAAO,GAAG,CAAC,WAAW,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC,CAAC;YAC1D,IAAI,IAAI,EAAE,MAAM;gBAAE,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC5C,IAAI,YAAY;gBAAE,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAC7C,OAAO,MAAM,IAAA,yBAAU,EAAC;gBACvB,KAAK;gBACL,IAAI,EAAE,EAAE,GAAG,IAAI,EAAE,MAAM,EAAE,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;gBACnD,YAAY;aACZ,CAAC,CAAC;QACJ,CAAC;KACD,CAAC,CAAC;IACH,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,GAAG,EAAE,EAAE,CAAC;AAC5C,CAAC;AAID,MAAM,IAAI,GAAG,IAAI,OAAO,EAAwC,CAAC;AAEjE,KAAK,UAAU,eAAe,CAAC,GAA2B;IACzD,IAAI,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IACxB,IAAI,CAAC,GAAG,EAAE,CAAC;QAEV,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YAClC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACjB,MAAM,KAAK,CAAC;QACb,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IACpB,CAAC;IACD,OAAO,MAAM,GAAG,CAAC;AAClB,CAAC;AAED,KAAK,UAAU,WAAW,CACzB,GAA6C,EAC7C,UAAsB,EACtB,SAAiB,EACjB,MAAgC;IAEhC,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,MAAM,eAAe,CAAC,GAAG,CAAC,CAAC;IAC5D,MAAM,IAAI,GAAG,GAAG,CAAC,OAAO,EAAE,CAAC;IAC3B,MAAM,SAAS,GAAG,GAAG,CAAC,gBAAgB,CAAC,iBAAiB,EAAE,SAAS,EAAE,EAAE,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC;IACjG,IAAA,kCAAmB,EAAC,iBAAiB,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;IACxD,MAAM,aAAa,GAAG,GAAG,CAAC,gBAAgB,CAAC,eAAe,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;IAC3E,IAAA,kCAAmB,EAAC,eAAe,EAAE,aAAa,EAAE,IAAI,CAAC,CAAC;IAC1D,MAAM,eAAe,GAAG,GAAG,CAAC,gBAAgB,CAAC,yBAAyB,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;IACvF,IAAA,iCAAkB,EACjB,iBAAiB,EACjB,eAAe,EACf,CAAC,KAAK,EAAmB,EAAE,CAAC,OAAO,KAAK,KAAK,QAAQ,EACrD,IAAI,CACJ,CAAC;IACF,MAAM,UAAU,GAAG,GAAG,CAAC,gBAAgB,CAAC,oBAAoB,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;IAChF,IAAA,kCAAmB,EAAC,oBAAoB,EAAE,UAAU,EAAE,IAAI,CAAC,CAAC;IAC5D,MAAM,SAAS,GAAG,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC;IAE7D,IAAI,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACnC,IAAI,CAAC,KAAK,EAAE,CAAC;QAEZ,KAAK,GAAG,6CAAqB,CAAC,aAAa,CAAC,KAAK,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YACnF,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YAC1B,MAAM,KAAK,CAAC;QACb,CAAC,CAAC,CAAC;QACH,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IAC/B,CAAC;IAED,OAAO,MAAM,6CAAqB,CAAC,iBAAiB,CAAC,UAAU,EAAE;QAChE,KAAK;QACL,IAAI;QACJ,SAAS;QACT,aAAa;QACb,eAAe;QACf,SAAS;QACT,MAAM;QACN,KAAK,EAAE,MAAM,KAAK;KAClB,CAAC,CAAC;AACJ,CAAC;AAQD,KAAK,UAAU,aAAa,CAE3B,MAAe;IAEf,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,CAC5C,2CAA0B,CAC1B,CAAC;IACF,IAAA,2BAAe,EAAC,IAAI,EAAE,WAAW,CAAC,IAAI,CAAC,CAAC;IACxC,MAAM,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IAEjD,MAAM,SAAS,GAAG,MAAM,IAAA,gCAAmB,EAC1C,IAAI,EACJ,GAAG,IAAI,kCAAkC,EACzC,EAAE,EACF,CAAC,IAAc,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAClC,CAAC;IACF,MAAM,OAAO,GAAG,CACf,MAAM,OAAO,CAAC,GAAG,CAChB,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE;QAChC,MAAM,OAAO,GAAG,MAAM,IAAA,gCAAmB,EACxC,IAAI,EACJ,GAAG,IAAI,gCAAgC,EACvC,EAAE,aAAa,EAAE,QAAQ,CAAC,IAAI,EAAE,EAChC,CAAC,IAAc,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CACvC,CAAC;
|
|
1
|
+
{"version":3,"file":"VectorStoreDatabricks.node.js","sourceRoot":"","sources":["../../../../nodes/vector_store/VectorStoreDatabricks/VectorStoreDatabricks.node.ts"],"names":[],"mappings":";;;AACA,oDAAsE;AACtE,+CASsB;AAEtB,6DAGsC;AACtC,2DAA8D;AAC9D,6DAAmE;AACnE,qEAG0C;AAC1C,mEAA0F;AAE1F,MAAM,kBAAkB,GAAG,MAAM,CAAC;AAElC,MAAM,kBAAkB,GAAoB;IAC3C,WAAW,EAAE,OAAO;IACpB,IAAI,EAAE,iBAAiB;IACvB,IAAI,EAAE,iBAAiB;IACvB,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE;IACpC,QAAQ,EAAE,IAAI;IACd,KAAK,EAAE;QACN;YACC,WAAW,EAAE,WAAW;YACxB,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,MAAM;YACZ,WAAW,EAAE,oBAAoB;YACjC,WAAW,EAAE;gBACZ,gBAAgB,EAAE,eAAe;gBACjC,UAAU,EAAE,IAAI;aAChB;SACD;QACD;YACC,WAAW,EAAE,MAAM;YACnB,IAAI,EAAE,IAAI;YACV,IAAI,EAAE,QAAQ;YACd,WAAW,EAAE,sBAAsB;YACnC,UAAU,EAAE;gBACX;oBACC,IAAI,EAAE,OAAO;oBACb,UAAU,EAAE;wBACX,KAAK,EAAE,kCAAkC;wBACzC,YAAY,EAAE,yCAAyC;qBACvD;iBACD;aACD;SACD;KACD;CACD,CAAC;AAEF,MAAM,iBAAiB,GAAG;IACzB,iBAAiB,EAAE,iBAAiB;IACpC,oBAAoB,EAAE,CAAC,uBAAuB,CAAC;CAC/C,CAAC;AAEF,MAAM,YAAY,GAAsB;IACvC,kBAAkB;IAClB;QACC,WAAW,EAAE,gBAAgB;QAC7B,IAAI,EAAE,eAAe;QACrB,IAAI,EAAE,SAAS;QACf,OAAO,EAAE,EAAE;QACX,WAAW,EAAE,iBAAiB;QAC9B,WAAW,EACV,qOAAqO;KACtO;IACD;QACC,WAAW,EAAE,SAAS;QACtB,IAAI,EAAE,SAAS;QACf,IAAI,EAAE,YAAY;QAClB,WAAW,EAAE,YAAY;QACzB,OAAO,EAAE,EAAE;QAEX,cAAc,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,CAAC,MAAM,EAAE,UAAU,EAAE,kBAAkB,CAAC,EAAE,EAAE;QAC5E,OAAO,EAAE;YACR;gBACC,WAAW,EAAE,kBAAkB;gBAC/B,IAAI,EAAE,iBAAiB;gBACvB,IAAI,EAAE,cAAc;gBACpB,OAAO,EAAE,EAAE;gBACX,WAAW,EAAE,iBAAiB;gBAC9B,WAAW,EACV,2NAA2N;aAC5N;YACD;gBACC,WAAW,EAAE,aAAa;gBAC1B,IAAI,EAAE,YAAY;gBAClB,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,KAAK;gBACd,OAAO,EAAE;oBACR,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE;oBAChC,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE;iBACnC;gBACD,WAAW,EACV,2IAA2I;aAC5I;YACD;gBACC,WAAW,EAAE,iBAAiB;gBAC9B,IAAI,EAAE,kBAAkB;gBACxB,IAAI,EAAE,MAAM;gBACZ,WAAW,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE;gBACxB,OAAO,EAAE,IAAI;gBACb,YAAY,EAAE,QAAQ;gBACtB,WAAW,EACV,sHAAsH;aACvH;SACD;KACD;CACD,CAAC;AAWF,KAAK,UAAU,OAAO,CAAC,GAA2B;IACjD,MAAM,UAAU,GAAG,MAAM,GAAG,CAAC,cAAc,CAC1C,2CAA0B,CAC1B,CAAC;IACF,IAAA,2BAAe,EAAC,GAAG,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC;IACtC,MAAM,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IAChD,MAAM,YAAY,GAAG,GAAG,CAAC,OAAO,CAAC,qBAAqB,EAAE,CAAC;IAEzD,MAAM,YAAY,GACjB,0BAA0B,IAAI,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,wBAAwB,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IAChF,MAAM,EAAE,KAAK,EAAE,GAAG,IAAA,sCAAyB,EAAC,GAAG,EAAE,UAAU,EAAE;QAC5D,WAAW,EAAE,IAAI;QACjB,YAAY;QACZ,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YAIhC,MAAM,OAAO,GAAG,CAAC,WAAW,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC,CAAC;YAC1D,IAAI,IAAI,EAAE,MAAM;gBAAE,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC5C,IAAI,YAAY;gBAAE,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAC7C,OAAO,MAAM,IAAA,yBAAU,EAAC;gBACvB,KAAK;gBACL,IAAI,EAAE,EAAE,GAAG,IAAI,EAAE,MAAM,EAAE,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;gBACnD,YAAY;aACZ,CAAC,CAAC;QACJ,CAAC;KACD,CAAC,CAAC;IACH,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,GAAG,EAAE,EAAE,CAAC;AAC5C,CAAC;AAID,MAAM,IAAI,GAAG,IAAI,OAAO,EAAwC,CAAC;AAEjE,KAAK,UAAU,eAAe,CAAC,GAA2B;IACzD,IAAI,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IACxB,IAAI,CAAC,GAAG,EAAE,CAAC;QAEV,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YAClC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACjB,MAAM,KAAK,CAAC;QACb,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IACpB,CAAC;IACD,OAAO,MAAM,GAAG,CAAC;AAClB,CAAC;AAED,KAAK,UAAU,WAAW,CACzB,GAA6C,EAC7C,UAAsB,EACtB,SAAiB,EACjB,MAAgC;IAEhC,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,MAAM,eAAe,CAAC,GAAG,CAAC,CAAC;IAC5D,MAAM,IAAI,GAAG,GAAG,CAAC,OAAO,EAAE,CAAC;IAC3B,MAAM,SAAS,GAAG,GAAG,CAAC,gBAAgB,CAAC,iBAAiB,EAAE,SAAS,EAAE,EAAE,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC;IACjG,IAAA,kCAAmB,EAAC,iBAAiB,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;IACxD,MAAM,aAAa,GAAG,GAAG,CAAC,gBAAgB,CAAC,eAAe,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;IAC3E,IAAA,kCAAmB,EAAC,eAAe,EAAE,aAAa,EAAE,IAAI,CAAC,CAAC;IAC1D,MAAM,eAAe,GAAG,GAAG,CAAC,gBAAgB,CAAC,yBAAyB,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;IACvF,IAAA,iCAAkB,EACjB,iBAAiB,EACjB,eAAe,EACf,CAAC,KAAK,EAAmB,EAAE,CAAC,OAAO,KAAK,KAAK,QAAQ,EACrD,IAAI,CACJ,CAAC;IACF,MAAM,UAAU,GAAG,GAAG,CAAC,gBAAgB,CAAC,oBAAoB,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;IAChF,IAAA,kCAAmB,EAAC,oBAAoB,EAAE,UAAU,EAAE,IAAI,CAAC,CAAC;IAC5D,MAAM,SAAS,GAAG,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC;IAE7D,IAAI,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACnC,IAAI,CAAC,KAAK,EAAE,CAAC;QAEZ,KAAK,GAAG,6CAAqB,CAAC,aAAa,CAAC,KAAK,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YACnF,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YAC1B,MAAM,KAAK,CAAC;QACb,CAAC,CAAC,CAAC;QACH,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IAC/B,CAAC;IAED,OAAO,MAAM,6CAAqB,CAAC,iBAAiB,CAAC,UAAU,EAAE;QAChE,KAAK;QACL,IAAI;QACJ,SAAS;QACT,aAAa;QACb,eAAe;QACf,SAAS;QACT,MAAM;QACN,KAAK,EAAE,MAAM,KAAK;KAClB,CAAC,CAAC;AACJ,CAAC;AAQD,KAAK,UAAU,aAAa,CAE3B,MAAe;IAEf,MAAM,UAAU,GAAG,IAAI,CAAC,uBAAuB,CAAC,MAAM,CAAC,KAAK,QAAQ,CAAC;IACrE,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,CAC5C,2CAA0B,CAC1B,CAAC;IACF,IAAA,2BAAe,EAAC,IAAI,EAAE,WAAW,CAAC,IAAI,CAAC,CAAC;IACxC,MAAM,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IAEjD,MAAM,SAAS,GAAG,MAAM,IAAA,gCAAmB,EAC1C,IAAI,EACJ,GAAG,IAAI,kCAAkC,EACzC,EAAE,EACF,CAAC,IAAc,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAClC,CAAC;IACF,MAAM,OAAO,GAAG,CACf,MAAM,OAAO,CAAC,GAAG,CAChB,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE;QAChC,MAAM,OAAO,GAAG,MAAM,IAAA,gCAAmB,EACxC,IAAI,EACJ,GAAG,IAAI,gCAAgC,EACvC,EAAE,aAAa,EAAE,QAAQ,CAAC,IAAI,EAAE,EAChC,CAAC,IAAc,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CACvC,CAAC;QACF,OAAO,OAAO;aACZ,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,UAAU,IAAI,KAAK,CAAC,UAAU,KAAK,eAAe,CAAC;aACtE,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;YAChB,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,KAAK,EAAE,KAAK,CAAC,IAAI;YACjB,WAAW,EAAE,GAAG,KAAK,CAAC,UAAU,MAAM,QAAQ,CAAC,IAAI,EAAE;SACrD,CAAC,CAAC,CAAC;IACN,CAAC,CAAC,CACF,CACD;SACC,IAAI,EAAE;SACN,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IAE/C,MAAM,WAAW,GAAG,MAAM,EAAE,WAAW,EAAE,CAAC;IAC1C,OAAO;QACN,OAAO,EAAE,WAAW;YACnB,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;YAC7E,CAAC,CAAC,OAAO;KACV,CAAC;AACH,CAAC;AAED,KAAK,UAAU,eAAe;IAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,uBAAuB,CAAC,iBAAiB,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC;IAC1F,IAAI,OAAO,SAAS,KAAK,QAAQ,IAAI,SAAS,KAAK,EAAE;QAAE,OAAO,EAAE,CAAC;IAEjE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,MAAM,eAAe,CAAC,IAAI,CAAC,CAAC;IACpD,MAAM,IAAI,GAAG,MAAM,6CAAqB,CAAC,aAAa,CAAC,KAAK,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;IAC/E,MAAM,EAAE,SAAS,EAAE,GAAG,IAAI,CAAC;IAC3B,MAAM,MAAM,GAAG,SAAS,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS,CAAC;IACjF,MAAM,MAAM,GAAG,SAAS,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS,CAAC;IAC9E,MAAM,OAAO,GAAG,CAAC,IAAI,CAAC,aAAa,IAAI,EAAE,CAAC;SACxC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,KAAK,IAAI,CAAC,UAAU,IAAI,MAAM,KAAK,MAAM,IAAI,MAAM,KAAK,MAAM,CAAC;SACxF,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;IACrD,OAAO,MAAM;QACZ,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,yBAAyB,EAAE,EAAE,GAAG,OAAO,CAAC;QACvF,CAAC,CAAC,OAAO,CAAC;AACZ,CAAC;AAED,2BAAmC,SAAQ,IAAA,oCAAqB,EAAwB;IACvF,IAAI,EAAE;QACL,WAAW,EAAE,yBAAyB;QACtC,IAAI,EAAE,uBAAuB;QAC7B,WAAW,EAAE,iDAAiD;QAC9D,IAAI,EAAE;YACL,KAAK,EAAE,wCAAwC;YAC/C,IAAI,EAAE,6CAA6C;SACnD;QACD,OAAO,EACN,8GAA8G;QAC/G,WAAW,EAAE;YACZ;gBACC,IAAI,EAAE,qBAAqB;gBAC3B,QAAQ,EAAE,IAAI;aACd;SACD;KACD;IACD,MAAM,EAAE,IAAI;IAEZ,YAAY,EAAE,IAAI;IAClB,OAAO,EAAE,EAAE,UAAU,EAAE,EAAE,aAAa,EAAE,EAAE,WAAW,EAAE,EAAE,eAAe,EAAE,EAAE;IAC5E,YAAY;IACZ,KAAK,CAAC,oBAAoB,CAAC,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS;QAChE,OAAO,MAAM,WAAW,CAAC,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;IAClE,CAAC;IACD,KAAK,CAAC,mBAAmB,CAAC,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,SAAS;QAClE,MAAM,KAAK,GAAG,MAAM,WAAW,CAAC,OAAO,EAAE,UAAU,EAAE,SAAS,CAAC,CAAC;QAChE,MAAM,KAAK,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;IACrC,CAAC;CACD,CAAC;CAAG"}
|
|
@@ -0,0 +1,5 @@
|
|
|
1
|
+
import { type IExecuteFunctions, type INodeExecutionData, type INodeType, type INodeTypeDescription } from 'n8n-workflow';
|
|
2
|
+
export declare class DatabricksAiAgent implements INodeType {
|
|
3
|
+
description: INodeTypeDescription;
|
|
4
|
+
execute(this: IExecuteFunctions): Promise<INodeExecutionData[][]>;
|
|
5
|
+
}
|
|
@@ -0,0 +1,170 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.DatabricksAiAgent = void 0;
|
|
4
|
+
const sanitize_error_detail_1 = require("@n8n/utils/redaction/sanitize-error-detail");
|
|
5
|
+
const n8n_workflow_1 = require("n8n-workflow");
|
|
6
|
+
const descriptions_1 = require("../../../utils/descriptions");
|
|
7
|
+
const description_1 = require("../../agents/Agent/agents/ToolsAgent/V2/description");
|
|
8
|
+
const execute_1 = require("../../agents/Agent/agents/ToolsAgent/V2/execute");
|
|
9
|
+
const utils_1 = require("../../agents/Agent/utils");
|
|
10
|
+
const span_collector_1 = require("./mlflow/span-collector");
|
|
11
|
+
const trace_writer_1 = require("./mlflow/trace-writer");
|
|
12
|
+
const transport_1 = require("./mlflow/transport");
|
|
13
|
+
const MAX_LOGGED_ERROR_CHARS = 1_000;
|
|
14
|
+
function sanitizedErrorMessage(error) {
|
|
15
|
+
const message = error instanceof Error ? error.message : String(error);
|
|
16
|
+
return (0, sanitize_error_detail_1.sanitizeErrorDetail)(message, MAX_LOGGED_ERROR_CHARS);
|
|
17
|
+
}
|
|
18
|
+
const exportTracesField = {
|
|
19
|
+
displayName: 'Export Traces to MLflow',
|
|
20
|
+
name: 'exportTraces',
|
|
21
|
+
type: 'boolean',
|
|
22
|
+
default: false,
|
|
23
|
+
noDataExpression: true,
|
|
24
|
+
description: 'Whether to send each run to MLflow Tracing in your Databricks workspace. Needs the Databricks credential above. Traces include prompts, model outputs and tool results.',
|
|
25
|
+
};
|
|
26
|
+
const experimentField = {
|
|
27
|
+
displayName: 'Experiment',
|
|
28
|
+
name: 'experiment',
|
|
29
|
+
type: 'resourceLocator',
|
|
30
|
+
default: { mode: 'name', value: '' },
|
|
31
|
+
displayOptions: { show: { exportTraces: [true] } },
|
|
32
|
+
description: 'Where traces are stored. Leave the name empty to create one for this workflow automatically.',
|
|
33
|
+
modes: [
|
|
34
|
+
{
|
|
35
|
+
displayName: 'By Name',
|
|
36
|
+
name: 'name',
|
|
37
|
+
type: 'string',
|
|
38
|
+
placeholder: '/Shared/my-agent-traces',
|
|
39
|
+
hint: 'Created if it does not exist. Empty means /Shared/n8n-workflows-<workflow ID>.',
|
|
40
|
+
},
|
|
41
|
+
{
|
|
42
|
+
displayName: 'By ID',
|
|
43
|
+
name: 'id',
|
|
44
|
+
type: 'string',
|
|
45
|
+
placeholder: '822142731468161',
|
|
46
|
+
},
|
|
47
|
+
],
|
|
48
|
+
};
|
|
49
|
+
async function prepareTraceExport(ctx) {
|
|
50
|
+
try {
|
|
51
|
+
const credential = await ctx.getCredentials('databricksOAuth2Api');
|
|
52
|
+
const { request, upload } = (0, transport_1.createMlflowTransport)(ctx, credential);
|
|
53
|
+
const experiment = ctx.getNodeParameter('experiment', 0, { mode: 'name', value: '' });
|
|
54
|
+
if (experiment.mode === 'id' && experiment.value) {
|
|
55
|
+
return { request, upload, experimentId: experiment.value };
|
|
56
|
+
}
|
|
57
|
+
const name = experiment.value || (0, trace_writer_1.defaultExperimentName)(ctx.getWorkflow().id ?? 'unknown');
|
|
58
|
+
return { request, upload, experimentId: await (0, trace_writer_1.ensureExperiment)(request, name) };
|
|
59
|
+
}
|
|
60
|
+
catch (error) {
|
|
61
|
+
ctx.logger.warn(`Databricks AI Agent: could not prepare MLflow trace export: ${sanitizedErrorMessage(error)}`);
|
|
62
|
+
return undefined;
|
|
63
|
+
}
|
|
64
|
+
}
|
|
65
|
+
async function flushTrace(ctx, exporter, collector) {
|
|
66
|
+
const trace = collector.finish();
|
|
67
|
+
if (!trace)
|
|
68
|
+
return;
|
|
69
|
+
try {
|
|
70
|
+
await (0, trace_writer_1.writeTrace)(exporter, trace);
|
|
71
|
+
}
|
|
72
|
+
catch (error) {
|
|
73
|
+
ctx.logger.warn(`Databricks AI Agent: MLflow trace export failed: ${sanitizedErrorMessage(error)}`);
|
|
74
|
+
}
|
|
75
|
+
}
|
|
76
|
+
class DatabricksAiAgent {
|
|
77
|
+
constructor() {
|
|
78
|
+
this.description = {
|
|
79
|
+
displayName: 'Databricks AI Agent',
|
|
80
|
+
name: 'databricksAiAgent',
|
|
81
|
+
icon: { light: 'file:databricks.svg', dark: 'file:databricks.dark.svg' },
|
|
82
|
+
group: ['transform'],
|
|
83
|
+
version: [2.2],
|
|
84
|
+
hidden: true,
|
|
85
|
+
description: 'An AI agent that exports its runs to MLflow Tracing in Databricks',
|
|
86
|
+
defaults: {
|
|
87
|
+
name: 'Databricks AI Agent',
|
|
88
|
+
},
|
|
89
|
+
codex: {
|
|
90
|
+
categories: ['AI'],
|
|
91
|
+
subcategories: {
|
|
92
|
+
AI: ['Agents', 'Root Nodes'],
|
|
93
|
+
},
|
|
94
|
+
},
|
|
95
|
+
inputs: `={{
|
|
96
|
+
((hasOutputParser, needsFallback) => {
|
|
97
|
+
${utils_1.getInputs.toString()};
|
|
98
|
+
return getInputs(true, hasOutputParser, needsFallback);
|
|
99
|
+
})($parameter.hasOutputParser === undefined || $parameter.hasOutputParser === true, $parameter.needsFallback !== undefined && $parameter.needsFallback === true)
|
|
100
|
+
}}`,
|
|
101
|
+
outputs: [n8n_workflow_1.NodeConnectionTypes.Main],
|
|
102
|
+
credentials: [
|
|
103
|
+
{
|
|
104
|
+
name: 'databricksOAuth2Api',
|
|
105
|
+
required: false,
|
|
106
|
+
},
|
|
107
|
+
],
|
|
108
|
+
properties: [
|
|
109
|
+
descriptions_1.promptTypeOptions,
|
|
110
|
+
{
|
|
111
|
+
...descriptions_1.textFromPreviousNode,
|
|
112
|
+
displayOptions: { show: { promptType: ['auto'] } },
|
|
113
|
+
},
|
|
114
|
+
{
|
|
115
|
+
...descriptions_1.textInput,
|
|
116
|
+
displayOptions: { show: { promptType: ['define'] } },
|
|
117
|
+
},
|
|
118
|
+
{
|
|
119
|
+
displayName: 'Require Specific Output Format',
|
|
120
|
+
name: 'hasOutputParser',
|
|
121
|
+
type: 'boolean',
|
|
122
|
+
default: false,
|
|
123
|
+
noDataExpression: true,
|
|
124
|
+
},
|
|
125
|
+
{
|
|
126
|
+
displayName: 'Enable Fallback Model',
|
|
127
|
+
name: 'needsFallback',
|
|
128
|
+
type: 'boolean',
|
|
129
|
+
default: false,
|
|
130
|
+
noDataExpression: true,
|
|
131
|
+
},
|
|
132
|
+
exportTracesField,
|
|
133
|
+
experimentField,
|
|
134
|
+
...(0, description_1.getToolsAgentProperties)({ withStreaming: true }),
|
|
135
|
+
],
|
|
136
|
+
};
|
|
137
|
+
}
|
|
138
|
+
async execute() {
|
|
139
|
+
const exportTraces = this.getNodeParameter('exportTraces', 0, false);
|
|
140
|
+
if (!exportTraces) {
|
|
141
|
+
return await execute_1.toolsAgentExecute.call(this);
|
|
142
|
+
}
|
|
143
|
+
if (!this.getNode().credentials?.databricksOAuth2Api) {
|
|
144
|
+
throw new n8n_workflow_1.NodeOperationError(this.getNode(), 'Select a Databricks credential to export traces to MLflow', {
|
|
145
|
+
description: '"Export Traces to MLflow" is on, but this node has no credential selected.',
|
|
146
|
+
});
|
|
147
|
+
}
|
|
148
|
+
const exporter = await prepareTraceExport(this);
|
|
149
|
+
if (!exporter) {
|
|
150
|
+
return await execute_1.toolsAgentExecute.call(this);
|
|
151
|
+
}
|
|
152
|
+
const collectors = new Map();
|
|
153
|
+
return await execute_1.toolsAgentExecute.call(this, {
|
|
154
|
+
createCallbacks: (itemIndex) => {
|
|
155
|
+
const collector = new span_collector_1.MlflowSpanCollector();
|
|
156
|
+
collectors.set(itemIndex, collector);
|
|
157
|
+
return [collector];
|
|
158
|
+
},
|
|
159
|
+
onItemFinished: async (itemIndex) => {
|
|
160
|
+
const collector = collectors.get(itemIndex);
|
|
161
|
+
if (!collector)
|
|
162
|
+
return;
|
|
163
|
+
collectors.delete(itemIndex);
|
|
164
|
+
await flushTrace(this, exporter, collector);
|
|
165
|
+
},
|
|
166
|
+
});
|
|
167
|
+
}
|
|
168
|
+
}
|
|
169
|
+
exports.DatabricksAiAgent = DatabricksAiAgent;
|
|
170
|
+
//# sourceMappingURL=DatabricksAiAgent.node.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"DatabricksAiAgent.node.js","sourceRoot":"","sources":["../../../../nodes/vendors/Databricks/DatabricksAiAgent.node.ts"],"names":[],"mappings":";;;AAAA,sFAAiF;AACjF,+CAQsB;AAGtB,sDAAyF;AAEzF,qFAA8F;AAC9F,6EAAoF;AACpF,oDAAqD;AACrD,4DAA8D;AAE9D,wDAA4F;AAC5F,kDAA2D;AAE3D,MAAM,sBAAsB,GAAG,KAAK,CAAC;AAGrC,SAAS,qBAAqB,CAAC,KAAc;IAC5C,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACvE,OAAO,IAAA,2CAAmB,EAAC,OAAO,EAAE,sBAAsB,CAAC,CAAC;AAC7D,CAAC;AAOD,MAAM,iBAAiB,GAAoB;IAC1C,WAAW,EAAE,yBAAyB;IACtC,IAAI,EAAE,cAAc;IACpB,IAAI,EAAE,SAAS;IACf,OAAO,EAAE,KAAK;IACd,gBAAgB,EAAE,IAAI;IACtB,WAAW,EACV,yKAAyK;CAC1K,CAAC;AAEF,MAAM,eAAe,GAAoB;IACxC,WAAW,EAAE,YAAY;IACzB,IAAI,EAAE,YAAY;IAClB,IAAI,EAAE,iBAAiB;IACvB,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE;IACpC,cAAc,EAAE,EAAE,IAAI,EAAE,EAAE,YAAY,EAAE,CAAC,IAAI,CAAC,EAAE,EAAE;IAClD,WAAW,EACV,8FAA8F;IAC/F,KAAK,EAAE;QACN;YACC,WAAW,EAAE,SAAS;YACtB,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,QAAQ;YACd,WAAW,EAAE,yBAAyB;YACtC,IAAI,EAAE,gFAAgF;SACtF;QACD;YACC,WAAW,EAAE,OAAO;YACpB,IAAI,EAAE,IAAI;YACV,IAAI,EAAE,QAAQ;YACd,WAAW,EAAE,iBAAiB;SAC9B;KACD;CACD,CAAC;AAYF,KAAK,UAAU,kBAAkB,CAAC,GAAsB;IACvD,IAAI,CAAC;QACJ,MAAM,UAAU,GAAG,MAAM,GAAG,CAAC,cAAc,CAA6B,qBAAqB,CAAC,CAAC;QAC/F,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,IAAA,iCAAqB,EAAC,GAAG,EAAE,UAAU,CAAC,CAAC;QAEnE,MAAM,UAAU,GAAG,GAAG,CAAC,gBAAgB,CAAC,YAAY,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE,CAGnF,CAAC;QACF,IAAI,UAAU,CAAC,IAAI,KAAK,IAAI,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;YAClD,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,UAAU,CAAC,KAAK,EAAE,CAAC;QAC5D,CAAC;QAED,MAAM,IAAI,GAAG,UAAU,CAAC,KAAK,IAAI,IAAA,oCAAqB,EAAC,GAAG,CAAC,WAAW,EAAE,CAAC,EAAE,IAAI,SAAS,CAAC,CAAC;QAC1F,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,IAAA,+BAAgB,EAAC,OAAO,EAAE,IAAI,CAAC,EAAE,CAAC;IACjF,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,GAAG,CAAC,MAAM,CAAC,IAAI,CACd,+DAA+D,qBAAqB,CAAC,KAAK,CAAC,EAAE,CAC7F,CAAC;QACF,OAAO,SAAS,CAAC;IAClB,CAAC;AACF,CAAC;AAGD,KAAK,UAAU,UAAU,CACxB,GAAsB,EACtB,QAAuB,EACvB,SAA8B;IAE9B,MAAM,KAAK,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC;IACjC,IAAI,CAAC,KAAK;QAAE,OAAO;IAEnB,IAAI,CAAC;QACJ,MAAM,IAAA,yBAAU,EAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IACnC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,GAAG,CAAC,MAAM,CAAC,IAAI,CACd,oDAAoD,qBAAqB,CAAC,KAAK,CAAC,EAAE,CAClF,CAAC;IACH,CAAC;AACF,CAAC;AAED;IAAA;QACC,gBAAW,GAAyB;YACnC,WAAW,EAAE,qBAAqB;YAClC,IAAI,EAAE,mBAAmB;YACzB,IAAI,EAAE,EAAE,KAAK,EAAE,qBAAqB,EAAE,IAAI,EAAE,0BAA0B,EAAE;YACxE,KAAK,EAAE,CAAC,WAAW,CAAC;YAGpB,OAAO,EAAE,CAAC,GAAG,CAAC;YACd,MAAM,EAAE,IAAI;YACZ,WAAW,EAAE,mEAAmE;YAChF,QAAQ,EAAE;gBACT,IAAI,EAAE,qBAAqB;aAC3B;YACD,KAAK,EAAE;gBACN,UAAU,EAAE,CAAC,IAAI,CAAC;gBAClB,aAAa,EAAE;oBACd,EAAE,EAAE,CAAC,QAAQ,EAAE,YAAY,CAAC;iBAC5B;aACD;YACD,MAAM,EAAE;;MAEJ,iBAAS,CAAC,QAAQ,EAAE;;;KAGrB;YACH,OAAO,EAAE,CAAC,kCAAmB,CAAC,IAAI,CAAC;YAKnC,WAAW,EAAE;gBACZ;oBACC,IAAI,EAAE,qBAAqB;oBAC3B,QAAQ,EAAE,KAAK;iBACf;aACD;YACD,UAAU,EAAE;gBACX,gCAAiB;gBACjB;oBACC,GAAG,mCAAoB;oBACvB,cAAc,EAAE,EAAE,IAAI,EAAE,EAAE,UAAU,EAAE,CAAC,MAAM,CAAC,EAAE,EAAE;iBAClD;gBACD;oBACC,GAAG,wBAAS;oBACZ,cAAc,EAAE,EAAE,IAAI,EAAE,EAAE,UAAU,EAAE,CAAC,QAAQ,CAAC,EAAE,EAAE;iBACpD;gBACD;oBACC,WAAW,EAAE,gCAAgC;oBAC7C,IAAI,EAAE,iBAAiB;oBACvB,IAAI,EAAE,SAAS;oBACf,OAAO,EAAE,KAAK;oBACd,gBAAgB,EAAE,IAAI;iBACtB;gBACD;oBACC,WAAW,EAAE,uBAAuB;oBACpC,IAAI,EAAE,eAAe;oBACrB,IAAI,EAAE,SAAS;oBACf,OAAO,EAAE,KAAK;oBACd,gBAAgB,EAAE,IAAI;iBACtB;gBACD,iBAAiB;gBACjB,eAAe;gBACf,GAAG,IAAA,qCAAuB,EAAC,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;aACnD;SACD,CAAC;IA4CH,CAAC;IA1CA,KAAK,CAAC,OAAO;QACZ,MAAM,YAAY,GAAG,IAAI,CAAC,gBAAgB,CAAC,cAAc,EAAE,CAAC,EAAE,KAAK,CAAY,CAAC;QAChF,IAAI,CAAC,YAAY,EAAE,CAAC;YACnB,OAAO,MAAM,2BAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3C,CAAC;QAID,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,WAAW,EAAE,mBAAmB,EAAE,CAAC;YACtD,MAAM,IAAI,iCAAkB,CAC3B,IAAI,CAAC,OAAO,EAAE,EACd,2DAA2D,EAC3D;gBACC,WAAW,EAAE,4EAA4E;aACzF,CACD,CAAC;QACH,CAAC;QAID,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC,IAAI,CAAC,CAAC;QAChD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACf,OAAO,MAAM,2BAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3C,CAAC;QAGD,MAAM,UAAU,GAAG,IAAI,GAAG,EAA+B,CAAC;QAE1D,OAAO,MAAM,2BAAiB,CAAC,IAAI,CAAC,IAAI,EAAE;YACzC,eAAe,EAAE,CAAC,SAAS,EAAE,EAAE;gBAC9B,MAAM,SAAS,GAAG,IAAI,oCAAmB,EAAE,CAAC;gBAC5C,UAAU,CAAC,GAAG,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;gBACrC,OAAO,CAAC,SAAS,CAAC,CAAC;YACpB,CAAC;YACD,cAAc,EAAE,KAAK,EAAE,SAAS,EAAE,EAAE;gBACnC,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;gBAC5C,IAAI,CAAC,SAAS;oBAAE,OAAO;gBACvB,UAAU,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;gBAC7B,MAAM,UAAU,CAAC,IAAI,EAAE,QAAQ,EAAE,SAAS,CAAC,CAAC;YAC7C,CAAC;SACD,CAAC,CAAC;IACJ,CAAC;CACD"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
<svg version="1.1" id="Layer_1" xmlns:x="ns_extend;" xmlns:i="ns_ai;" xmlns:graph="ns_graphs;" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" x="0px" y="0px" viewBox="0 0 40.1 42" style="enable-background:new 0 0 40.1 42;" xml:space="preserve">
|
|
2
|
+
<style type="text/css">
|
|
3
|
+
.st0{fill:#FF3621;}
|
|
4
|
+
</style>
|
|
5
|
+
<metadata>
|
|
6
|
+
<sfw xmlns="ns_sfw;">
|
|
7
|
+
<slices>
|
|
8
|
+
</slices>
|
|
9
|
+
<sliceSourceBounds bottomLeftOrigin="true" height="42" width="40.1" x="-69.1" y="-10.5">
|
|
10
|
+
</sliceSourceBounds>
|
|
11
|
+
</sfw>
|
|
12
|
+
</metadata>
|
|
13
|
+
<g>
|
|
14
|
+
<path class="st0" d="M40.1,31.1v-7.4l-0.8-0.5L20.1,33.7l-18.2-10l0-4.3l18.2,9.9l20.1-10.9v-7.3l-0.8-0.5L20.1,21.2L2.6,11.6
|
|
15
|
+
L20.1,2l14.1,7.7l1.1-0.6V8.3L20.1,0L0,10.9V12L20.1,23l18.2-10v4.4l-18.2,10L0.8,16.8L0,17.3v7.4l20.1,10.9l18.2-9.9v4.3l-18.2,10
|
|
16
|
+
L0.8,29.5L0,30v1.1L20.1,42L40.1,31.1z">
|
|
17
|
+
</path>
|
|
18
|
+
</g>
|
|
19
|
+
</svg>
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
<svg version="1.1" id="Layer_1" xmlns:x="ns_extend;" xmlns:i="ns_ai;" xmlns:graph="ns_graphs;" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" x="0px" y="0px" viewBox="0 0 40.1 42" style="enable-background:new 0 0 40.1 42;" xml:space="preserve">
|
|
2
|
+
<style type="text/css">
|
|
3
|
+
.st0{fill:#FF3621;}
|
|
4
|
+
</style>
|
|
5
|
+
<metadata>
|
|
6
|
+
<sfw xmlns="ns_sfw;">
|
|
7
|
+
<slices>
|
|
8
|
+
</slices>
|
|
9
|
+
<sliceSourceBounds bottomLeftOrigin="true" height="42" width="40.1" x="-69.1" y="-10.5">
|
|
10
|
+
</sliceSourceBounds>
|
|
11
|
+
</sfw>
|
|
12
|
+
</metadata>
|
|
13
|
+
<g>
|
|
14
|
+
<path class="st0" d="M40.1,31.1v-7.4l-0.8-0.5L20.1,33.7l-18.2-10l0-4.3l18.2,9.9l20.1-10.9v-7.3l-0.8-0.5L20.1,21.2L2.6,11.6
|
|
15
|
+
L20.1,2l14.1,7.7l1.1-0.6V8.3L20.1,0L0,10.9V12L20.1,23l18.2-10v4.4l-18.2,10L0.8,16.8L0,17.3v7.4l20.1,10.9l18.2-9.9v4.3l-18.2,10
|
|
16
|
+
L0.8,29.5L0,30v1.1L20.1,42L40.1,31.1z">
|
|
17
|
+
</path>
|
|
18
|
+
</g>
|
|
19
|
+
</svg>
|
|
@@ -0,0 +1,48 @@
|
|
|
1
|
+
import { BaseCallbackHandler } from '@langchain/core/callbacks/base';
|
|
2
|
+
import type { DocumentInterface } from '@langchain/core/documents';
|
|
3
|
+
import type { Serialized } from '@langchain/core/load/serializable';
|
|
4
|
+
import { BaseMessage } from '@langchain/core/messages';
|
|
5
|
+
import type { LLMResult } from '@langchain/core/outputs';
|
|
6
|
+
import type { ChainValues } from '@langchain/core/utils/types';
|
|
7
|
+
import type { CollectedTrace } from './types';
|
|
8
|
+
export declare class MlflowSpanCollector extends BaseCallbackHandler {
|
|
9
|
+
name: string;
|
|
10
|
+
awaitHandlers: boolean;
|
|
11
|
+
readonly traceId: string;
|
|
12
|
+
private readonly spanTraceId;
|
|
13
|
+
constructor();
|
|
14
|
+
private readonly spans;
|
|
15
|
+
private readonly parentOf;
|
|
16
|
+
private readonly runOrder;
|
|
17
|
+
private readonly openRuns;
|
|
18
|
+
private rootRunId?;
|
|
19
|
+
private startedAtMs?;
|
|
20
|
+
private lastActivityMs;
|
|
21
|
+
private executionId?;
|
|
22
|
+
private workflowId?;
|
|
23
|
+
private nodeName?;
|
|
24
|
+
private requestPreview?;
|
|
25
|
+
private responsePreview?;
|
|
26
|
+
private tokenUsage?;
|
|
27
|
+
private nearestKeptAncestor;
|
|
28
|
+
private captureRunMetadata;
|
|
29
|
+
private startSpan;
|
|
30
|
+
private closeSpan;
|
|
31
|
+
private endSpan;
|
|
32
|
+
private addTokenUsage;
|
|
33
|
+
private failSpan;
|
|
34
|
+
handleChainStart(_chain: Serialized, inputs: ChainValues, runId: string, parentRunId?: string, _tags?: string[], metadata?: Record<string, unknown>, _runType?: string, runName?: string): void;
|
|
35
|
+
handleChainEnd(outputs: ChainValues, runId: string): void;
|
|
36
|
+
handleChainError(error: Error, runId: string): void;
|
|
37
|
+
handleChatModelStart(llm: Serialized, messages: BaseMessage[][], runId: string, parentRunId?: string, _extraParams?: Record<string, unknown>, _tags?: string[], metadata?: Record<string, unknown>, runName?: string): void;
|
|
38
|
+
handleLLMStart(llm: Serialized, prompts: string[], runId: string, parentRunId?: string, _extraParams?: Record<string, unknown>, _tags?: string[], metadata?: Record<string, unknown>, runName?: string): void;
|
|
39
|
+
handleLLMEnd(output: LLMResult, runId: string): void;
|
|
40
|
+
handleLLMError(error: Error, runId: string): void;
|
|
41
|
+
handleToolStart(tool: Serialized, input: string, runId: string, parentRunId?: string, _tags?: string[], metadata?: Record<string, unknown>, runName?: string): void;
|
|
42
|
+
handleToolEnd(output: unknown, runId: string): void;
|
|
43
|
+
handleToolError(error: Error, runId: string): void;
|
|
44
|
+
handleRetrieverStart(retriever: Serialized, query: string, runId: string, parentRunId?: string, _tags?: string[], metadata?: Record<string, unknown>, name?: string): void;
|
|
45
|
+
handleRetrieverEnd(documents: DocumentInterface[], runId: string): void;
|
|
46
|
+
handleRetrieverError(error: Error, runId: string): void;
|
|
47
|
+
finish(): CollectedTrace | undefined;
|
|
48
|
+
}
|