@kyuda/n8n-nodes-databricks 1.0.2

package/dist/nodes/Databricks/resources/vectorSearch/parameters.d.ts

@@ -0,0 +1,2 @@
+import type { INodeProperties } from 'n8n-workflow';
+export declare const vectorSearchParameters: INodeProperties[];

package/dist/nodes/Databricks/resources/vectorSearch/parameters.js

@@ -0,0 +1,224 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.vectorSearchParameters = void 0;
+exports.vectorSearchParameters = [
+    {
+        displayName: 'Index Name',
+        name: 'indexName',
+        type: 'string',
+        required: true,
+        displayOptions: {
+            show: {
+                resource: ['vectorSearch'],
+                operation: ['getIndex', 'queryIndex'],
+            },
+        },
+        default: '',
+        description: 'Name of the vector search index',
+    },
+    {
+        displayName: 'Endpoint Name',
+        name: 'endpointName',
+        type: 'string',
+        required: true,
+        displayOptions: {
+            show: {
+                resource: ['vectorSearch'],
+                operation: ['listIndexes'],
+            },
+        },
+        default: '',
+        description: 'Name of the vector search endpoint',
+    },
+    {
+        displayName: 'Query Type',
+        name: 'queryType',
+        type: 'options',
+        required: true,
+        displayOptions: {
+            show: {
+                resource: ['vectorSearch'],
+                operation: ['queryIndex'],
+            },
+        },
+        options: [
+            {
+                name: 'Text Query',
+                value: 'text',
+                description: 'Query using text (automatically converted to vectors)',
+            },
+            {
+                name: 'Vector Query',
+                value: 'vector',
+                description: 'Query using pre-computed vector embeddings',
+            },
+        ],
+        default: 'text',
+        description: 'Type of query to perform',
+    },
+    {
+        displayName: 'Query Text',
+        name: 'queryText',
+        type: 'string',
+        required: true,
+        displayOptions: {
+            show: {
+                resource: ['vectorSearch'],
+                operation: ['queryIndex'],
+                queryType: ['text'],
+            },
+        },
+        default: '',
+        description: 'Text to search for (will be automatically converted to embeddings)',
+        placeholder: 'What is machine learning?',
+    },
+    {
+        displayName: 'Query Vector',
+        name: 'queryVector',
+        type: 'json',
+        required: true,
+        displayOptions: {
+            show: {
+                resource: ['vectorSearch'],
+                operation: ['queryIndex'],
+                queryType: ['vector'],
+            },
+        },
+        default: '[]',
+        description: 'Vector embeddings to search for similar vectors (array of numbers)',
+        placeholder: '[0.1, 0.2, 0.3, ...]',
+    },
+    {
+        displayName: 'Search Mode',
+        name: 'searchMode',
+        type: 'options',
+        displayOptions: {
+            show: {
+                resource: ['vectorSearch'],
+                operation: ['queryIndex'],
+            },
+        },
+        options: [
+            {
+                name: 'Hybrid',
+                value: 'HYBRID',
+                description: 'Combines semantic (vector) and keyword search for best results',
+            },
+            {
+                name: 'ANN (Approximate Nearest Neighbor)',
+                value: 'ANN',
+                description: 'Pure vector similarity search',
+            },
+        ],
+        default: 'HYBRID',
+        description: 'Search algorithm to use',
+    },
+    {
+        displayName: 'Columns to Return',
+        name: 'columns',
+        type: 'string',
+        required: true,
+        displayOptions: {
+            show: {
+                resource: ['vectorSearch'],
+                operation: ['queryIndex'],
+            },
+        },
+        default: '',
+        description: 'Comma-separated list of column names to return in results (e.g., "content,url,title")',
+        placeholder: 'content, url',
+    },
+    {
+        displayName: 'Number of Results',
+        name: 'numResults',
+        type: 'number',
+        displayOptions: {
+            show: {
+                resource: ['vectorSearch'],
+                operation: ['queryIndex'],
+            },
+        },
+        default: 10,
+        description: 'Maximum number of results to return',
+    },
+    {
+        displayName: 'Enable Reranking',
+        name: 'enableReranking',
+        type: 'boolean',
+        default: false,
+        displayOptions: {
+            show: {
+                resource: ['vectorSearch'],
+                operation: ['queryIndex'],
+            },
+        },
+        description: 'Whether to rerank results using a reranker model for improved relevance',
+    },
+    {
+        displayName: 'Reranker Model',
+        name: 'rerankerModel',
+        type: 'string',
+        required: true,
+        default: 'databricks_reranker',
+        displayOptions: {
+            show: {
+                resource: ['vectorSearch'],
+                operation: ['queryIndex'],
+                enableReranking: [true],
+            },
+        },
+        description: 'Name of the reranker model to use',
+    },
+    {
+        displayName: 'Columns to Rerank',
+        name: 'columnsToRerank',
+        type: 'string',
+        required: true,
+        default: '',
+        placeholder: 'content',
+        displayOptions: {
+            show: {
+                resource: ['vectorSearch'],
+                operation: ['queryIndex'],
+                enableReranking: [true],
+            },
+        },
+        description: 'Comma-separated list of columns to use for reranking (e.g., "content,title")',
+    },
+    {
+        displayName: 'Options',
+        name: 'options',
+        type: 'collection',
+        placeholder: 'Add Option',
+        default: {},
+        displayOptions: {
+            show: {
+                resource: ['vectorSearch'],
+                operation: ['queryIndex'],
+            },
+        },
+        options: [
+            {
+                displayName: 'Filter Expression',
+                name: 'filterExpression',
+                type: 'string',
+                default: '',
+                description: 'SQL-like filter expression to apply to the results (e.g., "category = \'docs\' AND published = true")',
+                placeholder: 'category = "documentation"',
+            },
+            {
+                displayName: 'Score Threshold',
+                name: 'scoreThreshold',
+                type: 'number',
+                default: 0,
+                description: 'Minimum relevance score threshold for results. Must be ≥ 0 and ≤ 1.',
+                typeOptions: {
+                    minValue: 0,
+                    maxValue: 1,
+                    numberPrecision: 2,
+                },
+            },
+        ],
+    },
+];
+//# sourceMappingURL=parameters.js.map

package/dist/nodes/Databricks/resources/vectorSearch/parameters.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"parameters.js","sourceRoot":"","sources":["../../../../../nodes/Databricks/resources/vectorSearch/parameters.ts"],"names":[],"mappings":";;;AAEa,QAAA,sBAAsB,GAAsB;IACrD;QACI,WAAW,EAAE,YAAY;QACzB,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,QAAQ;QACd,QAAQ,EAAE,IAAI;QACd,cAAc,EAAE;YACZ,IAAI,EAAE;gBACF,QAAQ,EAAE,CAAC,cAAc,CAAC;gBAC1B,SAAS,EAAE,CAAC,UAAU,EAAE,YAAY,CAAC;aACxC;SACJ;QACD,OAAO,EAAE,EAAE;QACX,WAAW,EAAE,iCAAiC;KACjD;IACD;QACI,WAAW,EAAE,eAAe;QAC5B,IAAI,EAAE,cAAc;QACpB,IAAI,EAAE,QAAQ;QACd,QAAQ,EAAE,IAAI;QACd,cAAc,EAAE;YACZ,IAAI,EAAE;gBACF,QAAQ,EAAE,CAAC,cAAc,CAAC;gBAC1B,SAAS,EAAE,CAAC,aAAa,CAAC;aAC7B;SACJ;QACD,OAAO,EAAE,EAAE;QACX,WAAW,EAAE,oCAAoC;KACpD;IACD;QACI,WAAW,EAAE,YAAY;QACzB,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,SAAS;QACf,QAAQ,EAAE,IAAI;QACd,cAAc,EAAE;YACZ,IAAI,EAAE;gBACF,QAAQ,EAAE,CAAC,cAAc,CAAC;gBAC1B,SAAS,EAAE,CAAC,YAAY,CAAC;aAC5B;SACJ;QACD,OAAO,EAAE;YACL;gBACI,IAAI,EAAE,YAAY;gBAClB,KAAK,EAAE,MAAM;gBACb,WAAW,EAAE,uDAAuD;aACvE;YACD;gBACI,IAAI,EAAE,cAAc;gBACpB,KAAK,EAAE,QAAQ;gBACf,WAAW,EAAE,4CAA4C;aAC5D;SACJ;QACD,OAAO,EAAE,MAAM;QACf,WAAW,EAAE,0BAA0B;KAC1C;IACD;QACI,WAAW,EAAE,YAAY;QACzB,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,QAAQ;QACd,QAAQ,EAAE,IAAI;QACd,cAAc,EAAE;YACZ,IAAI,EAAE;gBACF,QAAQ,EAAE,CAAC,cAAc,CAAC;gBAC1B,SAAS,EAAE,CAAC,YAAY,CAAC;gBACzB,SAAS,EAAE,CAAC,MAAM,CAAC;aACtB;SACJ;QACD,OAAO,EAAE,EAAE;QACX,WAAW,EAAE,oEAAoE;QACjF,WAAW,EAAE,2BAA2B;KAC3C;IACD;QACI,WAAW,EAAE,cAAc;QAC3B,IAAI,EAAE,aAAa;QACnB,IAAI,EAAE,MAAM;QACZ,QAAQ,EAAE,IAAI;QACd,cAAc,EAAE;YACZ,IAAI,EAAE;gBACF,QAAQ,EAAE,CAAC,cAAc,CAAC;gBAC1B,SAAS,EAAE,CAAC,YAAY,CAAC;gBACzB,SAAS,EAAE,CAAC,QAAQ,CAAC;aACxB;SACJ;QACD,OAAO,EAAE,IAAI;QACb,WAAW,EAAE,oEAAoE;QACjF,WAAW,EAAE,sBAAsB;KACtC;IACD;QACI,WAAW,EAAE,aAAa;QAC1B,IAAI,EAAE,YAAY;QAClB,IAAI,EAAE,SAAS;QACf,cAAc,EAAE;YACZ,IAAI,EAAE;gBACF,QAAQ,EAAE,CAAC,cAAc,CAAC;gBAC1B,SAAS,EAAE,CAAC,YAAY,CAAC;aAC5B;SACJ;QACD,OAAO,EAAE;YACL;gBACI,IAAI,EAAE,QAAQ;gBACd,KAAK,EAAE,QAAQ;gBACf,WAAW,EAAE,gEAAgE;aAChF;YACD;gBACI,IAAI,EAAE,oCAAoC;gBAC1C,KAAK,EAAE,KAAK;gBACZ,WAAW,EAAE,+BAA+B;aAC/C;SACJ;QACD,OAAO,EAAE,QAAQ;QACjB,WAAW,EAAE,yBAAyB;KACzC;IACD;QACI,WAAW,EAAE,mBAAmB;QAChC,IAAI,EAAE,SAAS;QACf,IAAI,EAAE,QAAQ;QACd,QAAQ,EAAE,IAAI;QACd,cAAc,EAAE;YACZ,IAAI,EAAE;gBACF,QAAQ,EAAE,CAAC,cAAc,CAAC;gBAC1B,SAAS,EAAE,CAAC,YAAY,CAAC;aAC5B;SACJ;QACD,OAAO,EAAE,EAAE;QACX,WAAW,EAAE,uFAAuF;QACpG,WAAW,EAAE,cAAc;KAC9B;IACD;QACI,WAAW,EAAE,mBAAmB;QAChC,IAAI,EAAE,YAAY;QAClB,IAAI,EAAE,QAAQ;QACd,cAAc,EAAE;YACZ,IAAI,EAAE;gBACF,QAAQ,EAAE,CAAC,cAAc,CAAC;gBAC1B,SAAS,EAAE,CAAC,YAAY,CAAC;aAC5B;SACJ;QACD,OAAO,EAAE,EAAE;QACX,WAAW,EAAE,qCAAqC;KACrD;IACD;QACI,WAAW,EAAE,kBAAkB;QAC/B,IAAI,EAAE,iBAAiB;QACvB,IAAI,EAAE,SAAS;QACf,OAAO,EAAE,KAAK;QACd,cAAc,EAAE;YACZ,IAAI,EAAE;gBACF,QAAQ,EAAE,CAAC,cAAc,CAAC;gBAC1B,SAAS,EAAE,CAAC,YAAY,CAAC;aAC5B;SACJ;QACD,WAAW,EAAE,yEAAyE;KACzF;IACD;QACI,WAAW,EAAE,gBAAgB;QAC7B,IAAI,EAAE,eAAe;QACrB,IAAI,EAAE,QAAQ;QACd,QAAQ,EAAE,IAAI;QACd,OAAO,EAAE,qBAAqB;QAC9B,cAAc,EAAE;YACZ,IAAI,EAAE;gBACF,QAAQ,EAAE,CAAC,cAAc,CAAC;gBAC1B,SAAS,EAAE,CAAC,YAAY,CAAC;gBACzB,eAAe,EAAE,CAAC,IAAI,CAAC;aAC1B;SACJ;QACD,WAAW,EAAE,mCAAmC;KACnD;IACD;QACI,WAAW,EAAE,mBAAmB;QAChC,IAAI,EAAE,iBAAiB;QACvB,IAAI,EAAE,QAAQ;QACd,QAAQ,EAAE,IAAI;QACd,OAAO,EAAE,EAAE;QACX,WAAW,EAAE,SAAS;QACtB,cAAc,EAAE;YACZ,IAAI,EAAE;gBACF,QAAQ,EAAE,CAAC,cAAc,CAAC;gBAC1B,SAAS,EAAE,CAAC,YAAY,CAAC;gBACzB,eAAe,EAAE,CAAC,IAAI,CAAC;aAC1B;SACJ;QACD,WAAW,EAAE,8EAA8E;KAC9F;IACD;QACI,WAAW,EAAE,SAAS;QACtB,IAAI,EAAE,SAAS;QACf,IAAI,EAAE,YAAY;QAClB,WAAW,EAAE,YAAY;QACzB,OAAO,EAAE,EAAE;QACX,cAAc,EAAE;YACZ,IAAI,EAAE;gBACF,QAAQ,EAAE,CAAC,cAAc,CAAC;gBAC1B,SAAS,EAAE,CAAC,YAAY,CAAC;aAC5B;SACJ;QACD,OAAO,EAAE;YACL;gBACI,WAAW,EAAE,mBAAmB;gBAChC,IAAI,EAAE,kBAAkB;gBACxB,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,EAAE;gBACX,WAAW,EAAE,uGAAuG;gBACpH,WAAW,EAAE,4BAA4B;aAC5C;YACD;gBACI,WAAW,EAAE,iBAAiB;gBAC9B,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,CAAC;gBACV,WAAW,EAAE,qEAAqE;gBAClF,WAAW,EAAE;oBACT,QAAQ,EAAE,CAAC;oBACX,QAAQ,EAAE,CAAC;oBACX,eAAe,EAAE,CAAC;iBACrB;aACJ;SACJ;KACJ;CACJ,CAAC"}

package/dist/nodes/agents/DatabricksAiAgent/CallbackHandler.d.ts

@@ -0,0 +1,66 @@
+import type { AgentAction, AgentFinish } from "@langchain/core/agents";
+import { BaseCallbackHandler } from "@langchain/core/callbacks/base";
+import type { Document } from "@langchain/core/documents";
+import type { Serialized } from "@langchain/core/load/serializable";
+import { BaseMessage, type BaseMessageFields } from "@langchain/core/messages";
+import type { LLMResult } from "@langchain/core/outputs";
+import type { ChainValues } from "@langchain/core/utils/types";
+export type LlmMessage = {
+    role: string;
+    content: BaseMessageFields["content"];
+    additional_kwargs?: BaseMessageFields["additional_kwargs"];
+};
+export type AnonymousLlmMessage = {
+    content: BaseMessageFields["content"];
+    additional_kwargs?: BaseMessageFields["additional_kwargs"];
+};
+type ErrorLike = Error | {
+    message: string;
+    stack?: string;
+    [key: string]: unknown;
+};
+type LLMTokenIndex = import("@langchain/core/dist/callbacks/base").NewTokenIndices;
+type ConstructorParams = {
+    userId?: string;
+    sessionId?: string;
+    tags?: string[];
+    version?: string;
+    traceMetadata?: Record<string, unknown>;
+};
+export declare class CallbackHandler extends BaseCallbackHandler {
+    name: string;
+    private readonly runMap;
+    private readonly maxMapSize;
+    private _lastTraceId;
+    get lastTraceId(): string | null;
+    constructor(params?: ConstructorParams);
+    get logger(): import("log4js/types/log4js").Logger;
+    private logError;
+    private ensureMapSize;
+    cleanup(): void;
+    handleLLMNewToken(token: string, _idx: LLMTokenIndex, runId: string, _parentRunId?: string, _tags?: string[], _fields?: Record<string, unknown>): Promise<void>;
+    handleChainStart(chain: Serialized, inputs: ChainValues, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, runType?: string, name?: string): Promise<void>;
+    handleAgentAction(action: AgentAction, runId: string, parentRunId?: string): Promise<void>;
+    handleAgentEnd?(action: AgentFinish, runId: string, parentRunId?: string): Promise<void>;
+    handleChainError(err: ErrorLike, runId: string, parentRunId?: string | undefined): Promise<void>;
+    handleGenerationStart(llm: Serialized, messages: Record<string, unknown> | Record<string, unknown>[] | string[], runId: string, parentRunId?: string | undefined, extraParams?: Record<string, unknown> | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, name?: string): Promise<void>;
+    handleChatModelStart(llm: Serialized, messages: BaseMessage[][], runId: string, parentRunId?: string | undefined, extraParams?: Record<string, unknown> | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, name?: string): Promise<void>;
+    handleChainEnd(outputs: ChainValues, runId: string, parentRunId?: string | undefined): Promise<void>;
+    handleLLMStart(llm: Serialized, prompts: string[], runId: string, parentRunId?: string | undefined, extraParams?: Record<string, unknown> | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, name?: string): Promise<void>;
+    handleToolStart(tool: Serialized, input: string, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, name?: string): Promise<void>;
+    handleRetrieverStart(retriever: Serialized, query: string, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, name?: string): Promise<void>;
+    handleRetrieverEnd(documents: Document<Record<string, any>>[], runId: string, parentRunId?: string | undefined): Promise<void>;
+    handleRetrieverError(err: ErrorLike, runId: string, parentRunId?: string | undefined): Promise<void>;
+    handleToolEnd(output: string, runId: string, parentRunId?: string | undefined): Promise<void>;
+    handleToolError(err: ErrorLike, runId: string, parentRunId?: string | undefined): Promise<void>;
+    handleLLMEnd(output: LLMResult, runId: string, parentRunId?: string | undefined): Promise<void>;
+    handleLLMError(err: ErrorLike, runId: string, parentRunId?: string | undefined): Promise<void>;
+    private startAndRegisterOtelSpan;
+    private handleOtelSpanEnd;
+    private parseAzureRefusalError;
+    private joinTagsAndMetaData;
+    private extractUsageMetadata;
+    private extractModelNameFromMetadata;
+    private extractChatMessageContent;
+}
+export {};