@mhalder/qdrant-mcp-server 1.3.1 → 1.5.0

package/build/prompts/template.test.js

@@ -0,0 +1,163 @@
+import { describe, expect, it } from "vitest";
+import { renderTemplate, validateArguments } from "./template.js";
+describe("renderTemplate", () => {
+    it("should render template with provided arguments", () => {
+        const template = "Search {{collection}} for {{query}}";
+        const args = { collection: "papers", query: "machine learning" };
+        const result = renderTemplate(template, args);
+        expect(result.text).toBe("Search papers for machine learning");
+    });
+    it("should handle multiple occurrences of the same variable", () => {
+        const template = "{{name}} said hello. {{name}} said goodbye.";
+        const args = { name: "Alice" };
+        const result = renderTemplate(template, args);
+        expect(result.text).toBe("Alice said hello. Alice said goodbye.");
+    });
+    it("should use default values for missing arguments", () => {
+        const template = "Return {{limit}} results";
+        const args = {};
+        const definitions = [
+            { name: "limit", description: "Number of results", required: false, default: "5" },
+        ];
+        const result = renderTemplate(template, args, definitions);
+        expect(result.text).toBe("Return 5 results");
+    });
+    it("should prefer provided arguments over defaults", () => {
+        const template = "Return {{limit}} results";
+        const args = { limit: "10" };
+        const definitions = [
+            { name: "limit", description: "Number of results", required: false, default: "5" },
+        ];
+        const result = renderTemplate(template, args, definitions);
+        expect(result.text).toBe("Return 10 results");
+    });
+    it("should keep placeholder if no value and no default", () => {
+        const template = "Search {{collection}} for {{query}}";
+        const args = { collection: "papers" };
+        const result = renderTemplate(template, args);
+        expect(result.text).toBe("Search papers for {{query}}");
+    });
+    it("should handle templates with no placeholders", () => {
+        const template = "This is a plain text template";
+        const args = { foo: "bar" };
+        const result = renderTemplate(template, args);
+        expect(result.text).toBe("This is a plain text template");
+    });
+    it("should handle empty template", () => {
+        const template = "";
+        const args = { foo: "bar" };
+        const result = renderTemplate(template, args);
+        expect(result.text).toBe("");
+    });
+    it("should handle empty args", () => {
+        const template = "Search {{collection}}";
+        const result = renderTemplate(template);
+        expect(result.text).toBe("Search {{collection}}");
+    });
+    it("should handle complex template with multiple variables and defaults", () => {
+        const template = "Search the '{{collection}}' collection for documents similar to: '{{query}}'. Return the top {{limit}} most relevant results.";
+        const args = { collection: "papers", query: "neural networks" };
+        const definitions = [
+            { name: "collection", description: "Collection name", required: true },
+            { name: "query", description: "Search query", required: true },
+            { name: "limit", description: "Number of results", required: false, default: "5" },
+        ];
+        const result = renderTemplate(template, args, definitions);
+        expect(result.text).toBe("Search the 'papers' collection for documents similar to: 'neural networks'. Return the top 5 most relevant results.");
+    });
+    it("should handle special characters in argument values", () => {
+        const template = "Query: {{query}}";
+        const args = { query: "What is $100 + $200?" };
+        const result = renderTemplate(template, args);
+        expect(result.text).toBe("Query: What is $100 + $200?");
+    });
+    it("should handle numeric-like values as strings", () => {
+        const template = "ID: {{id}}, Count: {{count}}";
+        const args = { id: "123", count: "456" };
+        const result = renderTemplate(template, args);
+        expect(result.text).toBe("ID: 123, Count: 456");
+    });
+    it("should handle whitespace in variable names", () => {
+        const template = "Value: {{key}}";
+        const args = { key: "  spaced  " };
+        const result = renderTemplate(template, args);
+        expect(result.text).toBe("Value:   spaced  ");
+    });
+    it("should not match invalid placeholder patterns", () => {
+        const template = "{{ invalid }} {missing} {{also invalid}} {{valid}}";
+        const args = { valid: "yes" };
+        const result = renderTemplate(template, args);
+        // Only {{valid}} should be replaced, others don't match \w+ pattern
+        expect(result.text).toBe("{{ invalid }} {missing} {{also invalid}} yes");
+    });
+});
+describe("validateArguments", () => {
+    it("should pass validation when all required arguments are provided", () => {
+        const args = { collection: "papers", query: "test" };
+        const definitions = [
+            { name: "collection", description: "Collection name", required: true },
+            { name: "query", description: "Search query", required: true },
+        ];
+        expect(() => validateArguments(args, definitions)).not.toThrow();
+    });
+    it("should throw error when required argument is missing", () => {
+        const args = { collection: "papers" };
+        const definitions = [
+            { name: "collection", description: "Collection name", required: true },
+            { name: "query", description: "Search query", required: true },
+        ];
+        expect(() => validateArguments(args, definitions)).toThrow("Missing required arguments: query");
+    });
+    it("should throw error when required argument is empty string", () => {
+        const args = { collection: "papers", query: "" };
+        const definitions = [
+            { name: "collection", description: "Collection name", required: true },
+            { name: "query", description: "Search query", required: true },
+        ];
+        expect(() => validateArguments(args, definitions)).toThrow("Missing required arguments: query");
+    });
+    it("should throw error listing multiple missing arguments", () => {
+        const args = {};
+        const definitions = [
+            { name: "collection", description: "Collection name", required: true },
+            { name: "query", description: "Search query", required: true },
+            { name: "filter", description: "Filter", required: true },
+        ];
+        expect(() => validateArguments(args, definitions)).toThrow("Missing required arguments: collection, query, filter");
+    });
+    it("should not throw for missing optional arguments", () => {
+        const args = { collection: "papers", query: "test" };
+        const definitions = [
+            { name: "collection", description: "Collection name", required: true },
+            { name: "query", description: "Search query", required: true },
+            { name: "limit", description: "Limit", required: false, default: "5" },
+        ];
+        expect(() => validateArguments(args, definitions)).not.toThrow();
+    });
+    it("should pass validation with empty definitions", () => {
+        const args = { foo: "bar" };
+        const definitions = [];
+        expect(() => validateArguments(args, definitions)).not.toThrow();
+    });
+    it("should pass validation with no definitions", () => {
+        const args = { foo: "bar" };
+        expect(() => validateArguments(args, [])).not.toThrow();
+    });
+    it("should not require arguments that are not defined", () => {
+        const args = { collection: "papers", extra: "data" };
+        const definitions = [
+            { name: "collection", description: "Collection name", required: true },
+        ];
+        expect(() => validateArguments(args, definitions)).not.toThrow();
+    });
+    it("should handle mixed required and optional arguments", () => {
+        const args = { name: "test" };
+        const definitions = [
+            { name: "name", description: "Name", required: true },
+            { name: "limit", description: "Limit", required: false },
+            { name: "filter", description: "Filter", required: false },
+        ];
+        expect(() => validateArguments(args, definitions)).not.toThrow();
+    });
+});
+//# sourceMappingURL=template.test.js.map

package/build/prompts/template.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"template.test.js","sourceRoot":"","sources":["../../src/prompts/template.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,eAAe,CAAC;AAGlE,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,QAAQ,GAAG,qCAAqC,CAAC;QACvD,MAAM,IAAI,GAAG,EAAE,UAAU,EAAE,QAAQ,EAAE,KAAK,EAAE,kBAAkB,EAAE,CAAC;QACjE,MAAM,MAAM,GAAG,cAAc,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAE9C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,oCAAoC,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,MAAM,QAAQ,GAAG,6CAA6C,CAAC;QAC/D,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;QAC/B,MAAM,MAAM,GAAG,cAAc,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAE9C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,uCAAuC,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,MAAM,QAAQ,GAAG,0BAA0B,CAAC;QAC5C,MAAM,IAAI,GAAG,EAAE,CAAC;QAChB,MAAM,WAAW,GAAqB;YACpC,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,mBAAmB,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,EAAE;SACnF,CAAC;QACF,MAAM,MAAM,GAAG,cAAc,CAAC,QAAQ,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC;QAE3D,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,QAAQ,GAAG,0BAA0B,CAAC;QAC5C,MAAM,IAAI,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;QAC7B,MAAM,WAAW,GAAqB;YACpC,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,mBAAmB,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,EAAE;SACnF,CAAC;QACF,MAAM,MAAM,GAAG,cAAc,CAAC,QAAQ,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC;QAE3D,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,QAAQ,GAAG,qCAAqC,CAAC;QACvD,MAAM,IAAI,GAAG,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,cAAc,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAE9C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,QAAQ,GAAG,+BAA+B,CAAC;QACjD,MAAM,IAAI,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC;QAC5B,MAAM,MAAM,GAAG,cAAc,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAE9C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACtC,MAAM,QAAQ,GAAG,EAAE,CAAC;QACpB,MAAM,IAAI,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC;QAC5B,MAAM,MAAM,GAAG,cAAc,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAE9C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;QAClC,MAAM,QAAQ,GAAG,uBAAuB,CAAC;QACzC,MAAM,MAAM,GAAG,cAAc,CAAC,QAAQ,CAAC,CAAC;QAExC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;QAC7E,MAAM,QAAQ,GACZ,+HAA+H,CAAC;QAClI,MAAM,IAAI,GAAG,EAAE,UAAU,EAAE,QAAQ,EAAE,KAAK,EAAE,iBAAiB,EAAE,CAAC;QAChE,MAAM,WAAW,GAAqB;YACpC,EAAE,IAAI,EAAE,YAAY,EAAE,WAAW,EAAE,iBAAiB,EAAE,QAAQ,EAAE,IAAI,EAAE;YACtE,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,QAAQ,EAAE,IAAI,EAAE;YAC9D,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,mBAAmB,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,EAAE;SACnF,CAAC;QACF,MAAM,MAAM,GAAG,cAAc,CAAC,QAAQ,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC;QAE3D,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CACtB,qHAAqH,CACtH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,QAAQ,GAAG,kBAAkB,CAAC;QACpC,MAAM,IAAI,GAAG,EAAE,KAAK,EAAE,sBAAsB,EAAE,CAAC;QAC/C,MAAM,MAAM,GAAG,cAAc,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAE9C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,QAAQ,GAAG,8BAA8B,CAAC;QAChD,MAAM,IAAI,GAAG,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;QACzC,MAAM,MAAM,GAAG,cAAc,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAE9C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,QAAQ,GAAG,gBAAgB,CAAC;QAClC,MAAM,IAAI,GAAG,EAAE,GAAG,EAAE,YAAY,EAAE,CAAC;QACnC,MAAM,MAAM,GAAG,cAAc,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAE9C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,QAAQ,GAAG,oDAAoD,CAAC;QACtE,MAAM,IAAI,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAG,cAAc,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAE9C,oEAAoE;QACpE,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,8CAA8C,CAAC,CAAC;IAC3E,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;QACzE,MAAM,IAAI,GAAG,EAAE,UAAU,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;QACrD,MAAM,WAAW,GAAqB;YACpC,EAAE,IAAI,EAAE,YAAY,EAAE,WAAW,EAAE,iBAAiB,EAAE,QAAQ,EAAE,IAAI,EAAE;YACtE,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,QAAQ,EAAE,IAAI,EAAE;SAC/D,CAAC;QAEF,MAAM,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,IAAI,GAAG,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC;QACtC,MAAM,WAAW,GAAqB;YACpC,EAAE,IAAI,EAAE,YAAY,EAAE,WAAW,EAAE,iBAAiB,EAAE,QAAQ,EAAE,IAAI,EAAE;YACtE,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,QAAQ,EAAE,IAAI,EAAE;SAC/D,CAAC;QAEF,MAAM,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,mCAAmC,CAAC,CAAC;IAClG,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;QACnE,MAAM,IAAI,GAAG,EAAE,UAAU,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QACjD,MAAM,WAAW,GAAqB;YACpC,EAAE,IAAI,EAAE,YAAY,EAAE,WAAW,EAAE,iBAAiB,EAAE,QAAQ,EAAE,IAAI,EAAE;YACtE,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,QAAQ,EAAE,IAAI,EAAE;SAC/D,CAAC;QAEF,MAAM,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,mCAAmC,CAAC,CAAC;IAClG,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,MAAM,IAAI,GAAG,EAAE,CAAC;QAChB,MAAM,WAAW,GAAqB;YACpC,EAAE,IAAI,EAAE,YAAY,EAAE,WAAW,EAAE,iBAAiB,EAAE,QAAQ,EAAE,IAAI,EAAE;YACtE,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,QAAQ,EAAE,IAAI,EAAE;YAC9D,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE;SAC1D,CAAC;QAEF,MAAM,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC,CAAC,OAAO,CACxD,uDAAuD,CACxD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,MAAM,IAAI,GAAG,EAAE,UAAU,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;QACrD,MAAM,WAAW,GAAqB;YACpC,EAAE,IAAI,EAAE,YAAY,EAAE,WAAW,EAAE,iBAAiB,EAAE,QAAQ,EAAE,IAAI,EAAE;YACtE,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,QAAQ,EAAE,IAAI,EAAE;YAC9D,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,EAAE;SACvE,CAAC;QAEF,MAAM,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,IAAI,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC;QAC5B,MAAM,WAAW,GAAqB,EAAE,CAAC;QAEzC,MAAM,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,IAAI,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC;QAE5B,MAAM,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,MAAM,IAAI,GAAG,EAAE,UAAU,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;QACrD,MAAM,WAAW,GAAqB;YACpC,EAAE,IAAI,EAAE,YAAY,EAAE,WAAW,EAAE,iBAAiB,EAAE,QAAQ,EAAE,IAAI,EAAE;SACvE,CAAC;QAEF,MAAM,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;QAC9B,MAAM,WAAW,GAAqB;YACpC,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE;YACrD,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE;YACxD,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE;SAC3D,CAAC;QAEF,MAAM,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACnE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/build/prompts/types.d.ts

@@ -0,0 +1,34 @@
+/**
+ * Configuration types for MCP prompts
+ */
+/**
+ * Argument definition for a prompt
+ */
+export interface PromptArgument {
+    name: string;
+    description: string;
+    required: boolean;
+    default?: string;
+}
+/**
+ * Individual prompt definition
+ */
+export interface PromptDefinition {
+    name: string;
+    description: string;
+    arguments: PromptArgument[];
+    template: string;
+}
+/**
+ * Root configuration structure
+ */
+export interface PromptsConfig {
+    prompts: PromptDefinition[];
+}
+/**
+ * Parsed template with resolved arguments
+ */
+export interface RenderedPrompt {
+    text: string;
+}
+//# sourceMappingURL=types.d.ts.map

package/build/prompts/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/prompts/types.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,OAAO,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,cAAc,EAAE,CAAC;IAC5B,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B,OAAO,EAAE,gBAAgB,EAAE,CAAC;CAC7B;AAED;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC;CACd"}

package/build/prompts/types.js

@@ -0,0 +1,5 @@
+/**
+ * Configuration types for MCP prompts
+ */
+export {};
+//# sourceMappingURL=types.js.map

package/build/prompts/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/prompts/types.ts"],"names":[],"mappings":"AAAA;;GAEG"}

package/examples/code-search/README.md

@@ -0,0 +1,271 @@
+# Code Search Example
+
+This example demonstrates how to use the code vectorization features to index and search codebases semantically.
+
+## Prerequisites
+
+- Qdrant and Ollama running (see main README)
+- Qdrant MCP server configured
+
+## Example 1: Index a TypeScript Project
+
+```bash
+# Index your project
+/mcp__qdrant__index_codebase /path/to/your/typescript/project
+
+# Expected output:
+# Scanning files...
+# Found 247 files
+# Chunking code...
+# Generating embeddings...
+# Storing chunks...
+# ✓ Indexed 247 files (1,823 chunks) in 45.2s
+```
+
+## Example 2: Search for Authentication Code
+
+```bash
+# Natural language search
+/mcp__qdrant__search_code /path/to/your/project "how does user authentication work?"
+
+# Results will include:
+# - File path and line numbers (e.g., src/auth/middleware.ts:15-42)
+# - Code snippets
+# - Relevance scores
+# - Language information
+```
+
+## Example 3: Search with Filters
+
+### Filter by File Type
+
+```bash
+# Only search TypeScript files
+/mcp__qdrant__search_code /path/to/your/project "error handling" --fileTypes .ts,.tsx
+```
+
+### Filter by Path Pattern
+
+```bash
+# Only search in API routes
+/mcp__qdrant__search_code /path/to/your/project "request validation" --pathPattern src/api/**
+```
+
+## Example 4: Incremental Updates
+
+After making changes to your codebase:
+
+```bash
+# Re-index only changed files
+/mcp__qdrant__reindex_changes /path/to/your/project
+
+# Expected output:
+# Detecting changes...
+# Found: +3 added, ~5 modified, -1 deleted
+# Updating index...
+# ✓ Updated in 8.3s
+# Chunks: +47 added, -23 deleted
+```
+
+## Example 5: Check Index Status
+
+```bash
+# Get information about an indexed codebase
+/mcp__qdrant__get_index_status /path/to/your/project
+
+# Output:
+# {
+#   "isIndexed": true,
+#   "collectionName": "code_a3f8d2e1",
+#   "chunksCount": 1823,
+#   "filesCount": 247,
+#   "lastUpdated": "2025-01-30T10:15:00Z",
+#   "languages": ["typescript", "javascript", "json"]
+# }
+```
+
+## Example 6: Multi-Language Project
+
+For projects with multiple languages:
+
+```bash
+# Index a full-stack project
+/mcp__qdrant__index_codebase /path/to/fullstack/project
+
+# The indexer automatically detects:
+# - Frontend: TypeScript, React, Vue
+# - Backend: Python, Go, Java
+# - Config: JSON, YAML, TOML
+# - Scripts: Bash, Shell
+
+# Search across all languages
+/mcp__qdrant__search_code /path/to/fullstack/project "database connection pooling"
+```
+
+## Example 7: Custom Ignore Patterns
+
+Create a `.contextignore` file in your project root:
+
+```gitignore
+# .contextignore
+**/test/**
+**/*.test.ts
+**/*.spec.ts
+**/fixtures/**
+**/mocks/**
+**/__tests__/**
+**/coverage/**
+*.generated.ts
+```
+
+Then index normally - these patterns will be automatically applied.
+
+## Example 8: Force Re-index
+
+If you need to completely re-index (e.g., after changing chunking settings):
+
+```bash
+# Force full re-index
+/mcp__qdrant__index_codebase /path/to/your/project --forceReindex true
+```
+
+## Example 9: Custom File Extensions
+
+Index non-standard file types:
+
+```bash
+# Add custom extensions
+/mcp__qdrant__index_codebase /path/to/your/project --extensions .proto,.graphql,.prisma
+```
+
+## Use Cases
+
+### 1. New Developer Onboarding
+
+**Scenario**: New developer needs to understand authentication flow
+
+```bash
+# Index the codebase once
+/mcp__qdrant__index_codebase /workspace/company-app
+
+# Then ask questions:
+/mcp__qdrant__search_code /workspace/company-app "authentication middleware"
+/mcp__qdrant__search_code /workspace/company-app "JWT token validation"
+/mcp__qdrant__search_code /workspace/company-app "password hashing"
+```
+
+### 2. Bug Investigation
+
+**Scenario**: Production bug in payment processing
+
+```bash
+# Search for payment-related code
+/mcp__qdrant__search_code /workspace/app "payment processing error handling"
+
+# Narrow down to specific service
+/mcp__qdrant__search_code /workspace/app "stripe payment" --pathPattern src/services/payment/**
+
+# Find related tests
+/mcp__qdrant__search_code /workspace/app "payment processing tests" --fileTypes .test.ts
+```
+
+### 3. Code Review
+
+**Scenario**: Understanding changes in a PR
+
+```bash
+# Index the feature branch
+git checkout feature/new-api
+/mcp__qdrant__index_codebase /workspace/app
+
+# Search for specific implementations
+/mcp__qdrant__search_code /workspace/app "new API endpoints"
+/mcp__qdrant__search_code /workspace/app "validation logic"
+```
+
+### 4. Documentation Writing
+
+**Scenario**: Writing API documentation
+
+```bash
+# Find all API endpoints
+/mcp__qdrant__search_code /workspace/app "REST API endpoints"
+/mcp__qdrant__search_code /workspace/app "GraphQL resolvers"
+
+# Find request/response schemas
+/mcp__qdrant__search_code /workspace/app "API schema definitions"
+```
+
+## Performance Tips
+
+1. **Use incremental updates**: After initial indexing, always use `reindex_changes` instead of full re-index
+2. **Filter aggressively**: Use `fileTypes` and `pathPattern` to narrow search scope
+3. **Batch indexing**: For very large codebases, consider indexing subdirectories separately
+4. **Local embeddings**: Use Ollama for fastest indexing and complete privacy
+5. **Monitor progress**: Check `get_index_status` to see indexing statistics
+
+## Troubleshooting
+
+### Slow Indexing
+
+```bash
+# Check current settings
+echo $CODE_BATCH_SIZE  # Default: 100
+
+# Increase batch size (if you have good internet/API limits)
+export CODE_BATCH_SIZE=200
+
+# Or use local Ollama (fastest)
+export EMBEDDING_PROVIDER=ollama
+```
+
+### Search Returns No Results
+
+```bash
+# Check if codebase is indexed
+/mcp__qdrant__get_index_status /path/to/your/project
+
+# If not indexed:
+/mcp__qdrant__index_codebase /path/to/your/project
+
+# Try broader queries
+# Instead of: "getUserById"
+# Try: "user retrieval functions"
+```
+
+### Files Not Being Indexed
+
+Check ignore patterns:
+
+```bash
+# View .gitignore
+cat /path/to/your/project/.gitignore
+
+# Create .contextignore to override
+echo "!tests/**" > /path/to/your/project/.contextignore
+```
+
+## Advanced Configuration
+
+Set environment variables for customization:
+
+```bash
+# Larger chunks for more context
+export CODE_CHUNK_SIZE=5000
+export CODE_CHUNK_OVERLAP=500
+
+# More results by default
+export CODE_DEFAULT_LIMIT=10
+
+# Custom file types
+export CODE_CUSTOM_EXTENSIONS=".prisma,.proto,.graphql"
+
+# Additional ignore patterns
+export CODE_CUSTOM_IGNORE="**/*.generated.ts,**/dist/**"
+```
+
+## Next Steps
+
+- Explore [hybrid search](../filters/) for better accuracy
+- Set up [custom prompts](../../prompts.example.json) for code search workflows
+- Integrate with your AI assistant for interactive code exploration

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mhalder/qdrant-mcp-server",
-  "version": "1.3.1",
+  "version": "1.5.0",
   "description": "MCP server for semantic search using local Qdrant and Ollama (default) with support for OpenAI, Cohere, and Voyage AI",
   "type": "module",
   "bin": {
@@ -32,13 +32,25 @@
   "publishConfig": {
     "access": "public"
   },
+  "overrides": {
+    "tree-sitter": "^0.25.0"
+  },
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.0.4",
     "@qdrant/js-client-rest": "^1.12.0",
     "bottleneck": "^2.19.5",
     "cohere-ai": "^7.19.0",
     "express": "^5.1.0",
+    "ignore": "^7.0.5",
     "openai": "^4.77.3",
+    "tree-sitter": "^0.25.0",
+    "tree-sitter-bash": "^0.25.0",
+    "tree-sitter-go": "^0.25.0",
+    "tree-sitter-java": "^0.23.5",
+    "tree-sitter-javascript": "^0.25.0",
+    "tree-sitter-python": "^0.25.0",
+    "tree-sitter-rust": "^0.24.0",
+    "tree-sitter-typescript": "^0.23.2",
     "zod": "^3.24.1"
   },
   "devDependencies": {

package/prompts.example.json

@@ -0,0 +1,96 @@
+{
+  "prompts": [
+    {
+      "name": "setup_rag_collection",
+      "description": "Create a new collection optimized for RAG workflows with hybrid search",
+      "arguments": [
+        {
+          "name": "name",
+          "description": "Name for the new collection",
+          "required": true
+        }
+      ],
+      "template": "Create a new Qdrant collection named '{{name}}' optimized for RAG (Retrieval-Augmented Generation) workflows:\n\n1. Create the collection with hybrid search enabled (enableHybrid: true) to combine semantic and keyword search for better retrieval accuracy\n2. Use Cosine distance metric (optimal for most embedding models)\n3. After creation, show the collection info and explain the configuration\n4. Provide guidance on adding documents with proper metadata structure for RAG use cases"
+    },
+    {
+      "name": "analyze_and_optimize",
+      "description": "Analyze a collection and provide optimization recommendations",
+      "arguments": [
+        {
+          "name": "collection",
+          "description": "Name of the collection to analyze",
+          "required": true
+        }
+      ],
+      "template": "Perform a comprehensive analysis of the '{{collection}}' collection:\n\n1. Get the collection info (dimensions, document count, distance metric, hybrid search status)\n2. Analyze the configuration and provide insights:\n   - Is the distance metric appropriate for the use case?\n   - Should hybrid search be enabled for better results?\n   - Is the document count sufficient for meaningful similarity search?\n3. Provide specific, actionable optimization recommendations\n4. If the collection seems suboptimal, suggest a migration strategy"
+    },
+    {
+      "name": "compare_search_strategies",
+      "description": "Compare semantic vs hybrid search to determine best approach",
+      "arguments": [
+        {
+          "name": "collection",
+          "description": "Collection name (must have hybrid search enabled)",
+          "required": true
+        },
+        {
+          "name": "query",
+          "description": "Test query",
+          "required": true
+        }
+      ],
+      "template": "Compare search strategies for '{{query}}' in the '{{collection}}' collection:\n\n1. Perform semantic_search and show top 5 results with scores\n2. Perform hybrid_search and show top 5 results with scores\n3. Analyze the differences:\n   - Which results appear in both? Which are unique to each method?\n   - How do the relevance scores compare?\n   - Which method seems more accurate for this query type?\n4. Recommend which search strategy to use based on:\n   - Query characteristics (keyword-heavy vs conceptual)\n   - Result quality and relevance\n   - Use case requirements"
+    },
+    {
+      "name": "migrate_to_hybrid",
+      "description": "Migrate an existing collection to use hybrid search",
+      "arguments": [
+        {
+          "name": "source",
+          "description": "Name of the existing collection to migrate",
+          "required": true
+        },
+        {
+          "name": "target",
+          "description": "Name for the new hybrid-enabled collection",
+          "required": true
+        }
+      ],
+      "template": "Migrate the '{{source}}' collection to a new hybrid search-enabled collection '{{target}}':\n\n1. First, analyze the source collection to understand its current configuration\n2. Create the target collection with hybrid search enabled, matching the source's distance metric\n3. Guide the user through retrieving documents from the source (note: you'll need to implement a retrieval mechanism or have the user provide the documents)\n4. Explain how to add the documents to the new hybrid collection\n5. Provide verification steps to ensure the migration was successful\n6. Ask if the user wants to delete the old collection after verification"
+    },
+    {
+      "name": "debug_search_quality",
+      "description": "Debug poor search results and suggest improvements",
+      "arguments": [
+        {
+          "name": "collection",
+          "description": "Collection with poor search quality",
+          "required": true
+        },
+        {
+          "name": "problematic_query",
+          "description": "A query that returns poor results",
+          "required": true
+        }
+      ],
+      "template": "Debug search quality issues in '{{collection}}' for query '{{problematic_query}}':\n\n1. Get collection info to understand the configuration\n2. Perform a search with the problematic query and analyze the results:\n   - Are the scores too low (< 0.5)? This suggests poor embedding quality or mismatched query/document domains\n   - Are irrelevant results ranking highly? This suggests metadata or hybrid search could help\n3. Test if hybrid search is available and compare results\n4. Provide specific recommendations:\n   - Should documents be re-embedded with a different model?\n   - Would metadata filtering help narrow results?\n   - Is the collection too small or missing relevant documents?\n   - Should the distance metric be changed?\n5. Suggest concrete next steps to improve search quality"
+    },
+    {
+      "name": "build_knowledge_base",
+      "description": "Guide for building a structured knowledge base with metadata",
+      "arguments": [
+        {
+          "name": "collection",
+          "description": "Collection name for the knowledge base",
+          "required": true
+        },
+        {
+          "name": "domain",
+          "description": "Domain or topic of the knowledge base",
+          "required": true
+        }
+      ],
+      "template": "Build a structured knowledge base in '{{collection}}' for the {{domain}} domain:\n\n1. Create (or verify) the collection with hybrid search enabled for optimal retrieval\n2. Design a metadata schema appropriate for {{domain}}, such as:\n   - category/topic fields for filtering\n   - timestamps for temporal filtering\n   - source/author for provenance\n   - confidence/quality scores\n3. Provide guidance on document structure:\n   - Optimal chunk sizes (typically 200-500 tokens)\n   - How to maintain context across chunks\n   - Metadata to include with each document\n4. Show example documents with proper formatting\n5. Explain search strategies for this knowledge base:\n   - When to use metadata filters\n   - When hybrid search adds value\n   - How to handle multi-step queries"
+    }
+  ]
+}

package/src/code/chunker/base.ts

@@ -0,0 +1,22 @@
+/**
+ * Base interface for code chunkers
+ */
+
+import type { CodeChunk } from "../types.js";
+
+export interface CodeChunker {
+  /**
+   * Split code into semantic chunks
+   */
+  chunk(code: string, filePath: string, language: string): Promise<CodeChunk[]>;
+
+  /**
+   * Check if language is supported by this chunker
+   */
+  supportsLanguage(language: string): boolean;
+
+  /**
+   * Get chunking strategy name
+   */
+  getStrategyName(): string;
+}