@mastra/duckdb 0.0.0

package/dist/vector/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/vector/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,KAAK,EACV,UAAU,EACV,WAAW,EACX,iBAAiB,EACjB,iBAAiB,EACjB,kBAAkB,EAClB,mBAAmB,EACnB,iBAAiB,EACjB,kBAAkB,EAClB,kBAAkB,EAClB,mBAAmB,EACpB,MAAM,qBAAqB,CAAC;AAE7B,OAAO,KAAK,EAAE,kBAAkB,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAEzE;;;;;;;;;;;GAWG;AACH,qBAAa,YAAa,SAAQ,YAAY,CAAC,kBAAkB,CAAC;IAChE,OAAO,CAAC,MAAM,CAAqB;IACnC,OAAO,CAAC,QAAQ,CAA+B;IAC/C,OAAO,CAAC,WAAW,CAAkB;IACrC,OAAO,CAAC,WAAW,CAA8B;gBAErC,MAAM,EAAE,kBAAkB;IAUtC;;OAEG;YACW,UAAU;IAgDxB;;OAEG;YACW,aAAa;IAQ3B;;OAEG;YACW,QAAQ;IA2BtB;;OAEG;YACW,YAAY;IAoB1B;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAQxB;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAarB,KAAK,CAAC,MAAM,EAAE,iBAAiB,CAAC,kBAAkB,CAAC,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAqE5E,MAAM,CAAC,MAAM,EAAE,kBAAkB,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IA8BrD,WAAW,CAAC,MAAM,EAAE,iBAAiB,GAAG,OAAO,CAAC,IAAI,CAAC;IAuCrD,WAAW,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAehC,aAAa,CAAC,MAAM,EAAE,mBAAmB,GAAG,OAAO,CAAC,UAAU,CAAC;IAsC/D,WAAW,CAAC,MAAM,EAAE,iBAAiB,GAAG,OAAO,CAAC,IAAI,CAAC;IAUrD,YAAY,CAAC,MAAM,EAAE,kBAAkB,CAAC,kBAAkB,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAkD3E,YAAY,CAAC,MAAM,EAAE,kBAAkB,GAAG,OAAO,CAAC,IAAI,CAAC;IAWvD,aAAa,CAAC,MAAM,EAAE,mBAAmB,CAAC,kBAAkB,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAmCnF;;;OAGG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAS7B"}

package/dist/vector/types.d.ts

@@ -0,0 +1,118 @@
+import type { VectorFilter } from '@mastra/core/vector/filter';
+/**
+ * Configuration options for DuckDBVector.
+ */
+export interface DuckDBVectorConfig {
+    /**
+     * Unique identifier for this vector store instance.
+     */
+    id: string;
+    /**
+     * Path to the DuckDB database file.
+     * Use ':memory:' for an in-memory database.
+     * @default ':memory:'
+     */
+    path?: string;
+    /**
+     * Default dimension for vector embeddings.
+     * @default 1536
+     */
+    dimensions?: number;
+    /**
+     * Default distance metric for similarity search.
+     * @default 'cosine'
+     */
+    metric?: 'cosine' | 'euclidean' | 'dotproduct';
+    /**
+     * Memory limit for DuckDB operations.
+     * @example '512MB', '2GB'
+     */
+    memoryLimit?: string;
+    /**
+     * Number of threads for DuckDB operations.
+     * @default Number of CPU cores
+     */
+    threads?: number;
+    /**
+     * Maximum number of connections in the pool.
+     * @default 5
+     */
+    maxPoolSize?: number;
+}
+/**
+ * HNSW index configuration options.
+ */
+export interface HNSWConfig {
+    /**
+     * Maximum number of connections per layer.
+     * Higher values improve recall but increase memory usage.
+     * @default 16
+     */
+    M?: number;
+    /**
+     * Size of the dynamic candidate list during construction.
+     * Higher values improve quality but slow down indexing.
+     * @default 200
+     */
+    efConstruction?: number;
+    /**
+     * Size of the dynamic candidate list during search.
+     * Higher values improve recall but slow down queries.
+     * @default 40
+     */
+    efSearch?: number;
+}
+/**
+ * Filter type for DuckDB vector queries.
+ * Supports MongoDB-style query operators.
+ */
+export type DuckDBVectorFilter = VectorFilter;
+/**
+ * Query options specific to DuckDB vector store.
+ */
+export interface DuckDBQueryOptions {
+    /**
+     * Maximum number of results to return.
+     * @default 10
+     */
+    topK?: number;
+    /**
+     * Minimum similarity score threshold (0-1 for cosine).
+     */
+    minScore?: number;
+    /**
+     * Whether to include the vector in results.
+     * @default false
+     */
+    includeVector?: boolean;
+    /**
+     * Text search query for hybrid search.
+     */
+    textQuery?: string;
+    /**
+     * Weight for vector similarity in hybrid search (0-1).
+     * @default 0.7
+     */
+    vectorWeight?: number;
+}
+/**
+ * Options for importing data from Parquet files.
+ */
+export interface ParquetImportOptions {
+    /**
+     * Column name containing the vector embeddings.
+     * @default 'embedding'
+     */
+    vectorColumn?: string;
+    /**
+     * Column name containing the vector IDs.
+     * @default 'id'
+     */
+    idColumn?: string;
+    /**
+     * Column names to include as metadata.
+     * If not specified, all non-vector columns are included.
+     */
+    metadataColumns?: string[];
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/vector/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/vector/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAE/D;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC;;OAEG;IACH,EAAE,EAAE,MAAM,CAAC;IAEX;;;;OAIG;IACH,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd;;;OAGG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IAEpB;;;OAGG;IACH,MAAM,CAAC,EAAE,QAAQ,GAAG,WAAW,GAAG,YAAY,CAAC;IAE/C;;;OAGG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB;;;OAGG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;IAEjB;;;OAGG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED;;GAEG;AACH,MAAM,WAAW,UAAU;IACzB;;;;OAIG;IACH,CAAC,CAAC,EAAE,MAAM,CAAC;IAEX;;;;OAIG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IAExB;;;;OAIG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED;;;GAGG;AACH,MAAM,MAAM,kBAAkB,GAAG,YAAY,CAAC;AAE9C;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC;;;OAGG;IACH,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;IAElB;;;OAGG;IACH,aAAa,CAAC,EAAE,OAAO,CAAC;IAExB;;OAEG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IAEnB;;;OAGG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC;;;OAGG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC;IAEtB;;;OAGG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;IAElB;;;OAGG;IACH,eAAe,CAAC,EAAE,MAAM,EAAE,CAAC;CAC5B"}

package/package.json

@@ -0,0 +1,66 @@
+{
+  "name": "@mastra/duckdb",
+  "version": "0.0.0",
+  "description": "DuckDB vector store provider for Mastra - embedded high-performance vector storage with HNSW indexing",
+  "type": "module",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "exports": {
+    ".": {
+      "import": {
+        "types": "./dist/index.d.ts",
+        "default": "./dist/index.js"
+      },
+      "require": {
+        "types": "./dist/index.d.ts",
+        "default": "./dist/index.cjs"
+      }
+    },
+    "./package.json": "./package.json"
+  },
+  "scripts": {
+    "build": "tsup --silent --config tsup.config.ts",
+    "build:watch": "tsup --watch --silent --config tsup.config.ts",
+    "test": "vitest run",
+    "lint": "eslint .",
+    "typecheck": "tsc --noEmit"
+  },
+  "license": "Apache-2.0",
+  "dependencies": {
+    "@duckdb/node-api": "1.4.2-r.1"
+  },
+  "devDependencies": {
+    "@internal/lint": "workspace:*",
+    "@internal/storage-test-utils": "workspace:*",
+    "@internal/types-builder": "workspace:*",
+    "@mastra/core": "workspace:*",
+    "@microsoft/api-extractor": "^7.52.8",
+    "@types/node": "22.13.17",
+    "@vitest/coverage-v8": "catalog:",
+    "@vitest/ui": "catalog:",
+    "eslint": "^9.37.0",
+    "rimraf": "^5.0.7",
+    "tsup": "^8.5.0",
+    "typescript": "^5.8.3",
+    "vitest": "catalog:"
+  },
+  "peerDependencies": {
+    "@mastra/core": ">=1.0.0-0 <2.0.0-0"
+  },
+  "files": [
+    "dist",
+    "CHANGELOG.md"
+  ],
+  "homepage": "https://mastra.ai",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/mastra-ai/mastra.git",
+    "directory": "stores/duckdb"
+  },
+  "bugs": {
+    "url": "https://github.com/mastra-ai/mastra/issues"
+  },
+  "engines": {
+    "node": ">=22.13.0"
+  }
+}