@neural-tools/vector-db 0.1.3 → 0.1.5

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.
@@ -0,0 +1,110 @@
1
+ interface VectorDBConfig {
2
+ provider: 'pinecone' | 'qdrant' | 'chromadb' | 'local';
3
+ apiKey?: string;
4
+ endpoint?: string;
5
+ dimension?: number;
6
+ metric?: 'cosine' | 'euclidean' | 'dotproduct';
7
+ namespace?: string;
8
+ }
9
+ interface Vector {
10
+ id: string;
11
+ values: number[];
12
+ metadata?: Record<string, any>;
13
+ }
14
+ interface QueryResult {
15
+ id: string;
16
+ score: number;
17
+ metadata?: Record<string, any>;
18
+ values?: number[];
19
+ }
20
+ interface QueryOptions {
21
+ topK?: number;
22
+ filter?: Record<string, any>;
23
+ includeValues?: boolean;
24
+ includeMetadata?: boolean;
25
+ }
26
+ interface UpsertOptions {
27
+ namespace?: string;
28
+ batch?: boolean;
29
+ }
30
+ declare abstract class VectorStore {
31
+ protected config: VectorDBConfig;
32
+ constructor(config: VectorDBConfig);
33
+ /**
34
+ * Initialize the vector store connection
35
+ */
36
+ abstract connect(): Promise<void>;
37
+ /**
38
+ * Insert or update vectors
39
+ */
40
+ abstract upsert(vectors: Vector[], options?: UpsertOptions): Promise<void>;
41
+ /**
42
+ * Query for similar vectors
43
+ */
44
+ abstract query(vector: number[], options?: QueryOptions): Promise<QueryResult[]>;
45
+ /**
46
+ * Delete vectors by ID
47
+ */
48
+ abstract delete(ids: string[], namespace?: string): Promise<void>;
49
+ /**
50
+ * Get vector by ID
51
+ */
52
+ abstract fetch(ids: string[], namespace?: string): Promise<Vector[]>;
53
+ /**
54
+ * Delete all vectors in a namespace
55
+ */
56
+ abstract deleteNamespace(namespace: string): Promise<void>;
57
+ /**
58
+ * Close the connection
59
+ */
60
+ abstract disconnect(): Promise<void>;
61
+ }
62
+
63
+ /**
64
+ * Simple in-memory vector store for local development and testing
65
+ */
66
+ declare class LocalVectorStore extends VectorStore {
67
+ private vectors;
68
+ connect(): Promise<void>;
69
+ upsert(vectors: Vector[], options?: UpsertOptions): Promise<void>;
70
+ query(vector: number[], options?: QueryOptions): Promise<QueryResult[]>;
71
+ delete(ids: string[]): Promise<void>;
72
+ fetch(ids: string[]): Promise<Vector[]>;
73
+ deleteNamespace(): Promise<void>;
74
+ disconnect(): Promise<void>;
75
+ /**
76
+ * Calculate cosine similarity between two vectors
77
+ */
78
+ private cosineSimilarity;
79
+ /**
80
+ * Check if metadata matches filter
81
+ */
82
+ private matchesFilter;
83
+ /**
84
+ * Get all vectors (for testing/debugging)
85
+ */
86
+ getAll(): Vector[];
87
+ /**
88
+ * Get count of vectors
89
+ */
90
+ count(): number;
91
+ }
92
+
93
+ /**
94
+ * Create a vector store instance based on the configuration
95
+ */
96
+ declare function createVectorStore(config: VectorDBConfig): Promise<VectorStore>;
97
+ /**
98
+ * Helper function to create embeddings from text
99
+ * In production, this would call OpenAI, Anthropic, or other embedding APIs
100
+ */
101
+ declare function createEmbedding(text: string): Promise<number[]>;
102
+ /**
103
+ * Batch text into chunks for embedding
104
+ */
105
+ declare function chunkText(text: string, options?: {
106
+ chunkSize?: number;
107
+ overlap?: number;
108
+ }): string[];
109
+
110
+ export { LocalVectorStore, type QueryOptions, type QueryResult, type UpsertOptions, type Vector, type VectorDBConfig, VectorStore, chunkText, createEmbedding, createVectorStore };
package/dist/index.d.ts CHANGED
@@ -1,19 +1,110 @@
1
- import { VectorDBConfig, VectorStore } from './types';
2
- export * from './types';
3
- export { LocalVectorStore } from './local-store';
1
+ interface VectorDBConfig {
2
+ provider: 'pinecone' | 'qdrant' | 'chromadb' | 'local';
3
+ apiKey?: string;
4
+ endpoint?: string;
5
+ dimension?: number;
6
+ metric?: 'cosine' | 'euclidean' | 'dotproduct';
7
+ namespace?: string;
8
+ }
9
+ interface Vector {
10
+ id: string;
11
+ values: number[];
12
+ metadata?: Record<string, any>;
13
+ }
14
+ interface QueryResult {
15
+ id: string;
16
+ score: number;
17
+ metadata?: Record<string, any>;
18
+ values?: number[];
19
+ }
20
+ interface QueryOptions {
21
+ topK?: number;
22
+ filter?: Record<string, any>;
23
+ includeValues?: boolean;
24
+ includeMetadata?: boolean;
25
+ }
26
+ interface UpsertOptions {
27
+ namespace?: string;
28
+ batch?: boolean;
29
+ }
30
+ declare abstract class VectorStore {
31
+ protected config: VectorDBConfig;
32
+ constructor(config: VectorDBConfig);
33
+ /**
34
+ * Initialize the vector store connection
35
+ */
36
+ abstract connect(): Promise<void>;
37
+ /**
38
+ * Insert or update vectors
39
+ */
40
+ abstract upsert(vectors: Vector[], options?: UpsertOptions): Promise<void>;
41
+ /**
42
+ * Query for similar vectors
43
+ */
44
+ abstract query(vector: number[], options?: QueryOptions): Promise<QueryResult[]>;
45
+ /**
46
+ * Delete vectors by ID
47
+ */
48
+ abstract delete(ids: string[], namespace?: string): Promise<void>;
49
+ /**
50
+ * Get vector by ID
51
+ */
52
+ abstract fetch(ids: string[], namespace?: string): Promise<Vector[]>;
53
+ /**
54
+ * Delete all vectors in a namespace
55
+ */
56
+ abstract deleteNamespace(namespace: string): Promise<void>;
57
+ /**
58
+ * Close the connection
59
+ */
60
+ abstract disconnect(): Promise<void>;
61
+ }
62
+
63
+ /**
64
+ * Simple in-memory vector store for local development and testing
65
+ */
66
+ declare class LocalVectorStore extends VectorStore {
67
+ private vectors;
68
+ connect(): Promise<void>;
69
+ upsert(vectors: Vector[], options?: UpsertOptions): Promise<void>;
70
+ query(vector: number[], options?: QueryOptions): Promise<QueryResult[]>;
71
+ delete(ids: string[]): Promise<void>;
72
+ fetch(ids: string[]): Promise<Vector[]>;
73
+ deleteNamespace(): Promise<void>;
74
+ disconnect(): Promise<void>;
75
+ /**
76
+ * Calculate cosine similarity between two vectors
77
+ */
78
+ private cosineSimilarity;
79
+ /**
80
+ * Check if metadata matches filter
81
+ */
82
+ private matchesFilter;
83
+ /**
84
+ * Get all vectors (for testing/debugging)
85
+ */
86
+ getAll(): Vector[];
87
+ /**
88
+ * Get count of vectors
89
+ */
90
+ count(): number;
91
+ }
92
+
4
93
  /**
5
94
  * Create a vector store instance based on the configuration
6
95
  */
7
- export declare function createVectorStore(config: VectorDBConfig): Promise<VectorStore>;
96
+ declare function createVectorStore(config: VectorDBConfig): Promise<VectorStore>;
8
97
  /**
9
98
  * Helper function to create embeddings from text
10
99
  * In production, this would call OpenAI, Anthropic, or other embedding APIs
11
100
  */
12
- export declare function createEmbedding(text: string): Promise<number[]>;
101
+ declare function createEmbedding(text: string): Promise<number[]>;
13
102
  /**
14
103
  * Batch text into chunks for embedding
15
104
  */
16
- export declare function chunkText(text: string, options?: {
105
+ declare function chunkText(text: string, options?: {
17
106
  chunkSize?: number;
18
107
  overlap?: number;
19
108
  }): string[];
109
+
110
+ export { LocalVectorStore, type QueryOptions, type QueryResult, type UpsertOptions, type Vector, type VectorDBConfig, VectorStore, chunkText, createEmbedding, createVectorStore };
package/dist/index.js CHANGED
@@ -1,87 +1 @@
1
- "use strict";
2
- var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
3
- if (k2 === undefined) k2 = k;
4
- var desc = Object.getOwnPropertyDescriptor(m, k);
5
- if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
6
- desc = { enumerable: true, get: function() { return m[k]; } };
7
- }
8
- Object.defineProperty(o, k2, desc);
9
- }) : (function(o, m, k, k2) {
10
- if (k2 === undefined) k2 = k;
11
- o[k2] = m[k];
12
- }));
13
- var __exportStar = (this && this.__exportStar) || function(m, exports) {
14
- for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
15
- };
16
- Object.defineProperty(exports, "__esModule", { value: true });
17
- exports.LocalVectorStore = void 0;
18
- exports.createVectorStore = createVectorStore;
19
- exports.createEmbedding = createEmbedding;
20
- exports.chunkText = chunkText;
21
- const local_store_1 = require("./local-store");
22
- const core_1 = require("@neural-tools/core");
23
- __exportStar(require("./types"), exports);
24
- var local_store_2 = require("./local-store");
25
- Object.defineProperty(exports, "LocalVectorStore", { enumerable: true, get: function () { return local_store_2.LocalVectorStore; } });
26
- /**
27
- * Create a vector store instance based on the configuration
28
- */
29
- async function createVectorStore(config) {
30
- switch (config.provider) {
31
- case 'local':
32
- return new local_store_1.LocalVectorStore(config);
33
- case 'pinecone':
34
- await (0, core_1.requireFeature)('vector-db', 'Vector Database Integration');
35
- throw new Error('Pinecone integration coming soon. Use local provider for now.');
36
- case 'qdrant':
37
- await (0, core_1.requireFeature)('vector-db', 'Vector Database Integration');
38
- throw new Error('Qdrant integration coming soon. Use local provider for now.');
39
- case 'chromadb':
40
- await (0, core_1.requireFeature)('vector-db', 'Vector Database Integration');
41
- throw new Error('ChromaDB integration coming soon. Use local provider for now.');
42
- default:
43
- throw new Error(`Unknown vector store provider: ${config.provider}`);
44
- }
45
- }
46
- /**
47
- * Helper function to create embeddings from text
48
- * In production, this would call OpenAI, Anthropic, or other embedding APIs
49
- */
50
- async function createEmbedding(text) {
51
- // Placeholder: Simple hash-based embedding for testing
52
- // In production, use proper embedding models
53
- const hash = simpleHash(text);
54
- const dimension = 384; // Common embedding dimension
55
- const embedding = new Array(dimension).fill(0);
56
- // Create a deterministic but distributed embedding from hash
57
- for (let i = 0; i < dimension; i++) {
58
- embedding[i] = Math.sin(hash + i) * Math.cos(hash / (i + 1));
59
- }
60
- // Normalize
61
- const norm = Math.sqrt(embedding.reduce((sum, val) => sum + val * val, 0));
62
- return embedding.map(val => val / norm);
63
- }
64
- function simpleHash(str) {
65
- let hash = 0;
66
- for (let i = 0; i < str.length; i++) {
67
- const char = str.charCodeAt(i);
68
- hash = ((hash << 5) - hash) + char;
69
- hash = hash & hash; // Convert to 32-bit integer
70
- }
71
- return Math.abs(hash);
72
- }
73
- /**
74
- * Batch text into chunks for embedding
75
- */
76
- function chunkText(text, options = {}) {
77
- const chunkSize = options.chunkSize || 500;
78
- const overlap = options.overlap || 50;
79
- const chunks = [];
80
- let start = 0;
81
- while (start < text.length) {
82
- const end = Math.min(start + chunkSize, text.length);
83
- chunks.push(text.slice(start, end));
84
- start += chunkSize - overlap;
85
- }
86
- return chunks;
87
- }
1
+ "use strict";var d=Object.defineProperty;var m=Object.getOwnPropertyDescriptor;var p=Object.getOwnPropertyNames;var f=Object.prototype.hasOwnProperty;var h=(t,e)=>{for(var r in e)d(t,r,{get:e[r],enumerable:!0})},g=(t,e,r,o)=>{if(e&&typeof e=="object"||typeof e=="function")for(let n of p(e))!f.call(t,n)&&n!==r&&d(t,n,{get:()=>e[n],enumerable:!(o=m(e,n))||o.enumerable});return t};var v=t=>g(d({},"__esModule",{value:!0}),t);var P={};h(P,{LocalVectorStore:()=>c,VectorStore:()=>a,chunkText:()=>w,createEmbedding:()=>y,createVectorStore:()=>b});module.exports=v(P);var a=class{config;constructor(e){this.config=e}};var c=class extends a{vectors=new Map;async connect(){}async upsert(e,r){for(let o of e)this.vectors.set(o.id,o)}async query(e,r){let o=r?.topK||10,n=[];for(let[s,i]of this.vectors){if(r?.filter&&!this.matchesFilter(i.metadata,r.filter))continue;let l=this.cosineSimilarity(e,i.values);n.push({id:s,score:l,metadata:r?.includeMetadata!==!1?i.metadata:void 0,values:r?.includeValues?i.values:void 0})}return n.sort((s,i)=>i.score-s.score),n.slice(0,o)}async delete(e){for(let r of e)this.vectors.delete(r)}async fetch(e){let r=[];for(let o of e){let n=this.vectors.get(o);n&&r.push(n)}return r}async deleteNamespace(){this.vectors.clear()}async disconnect(){}cosineSimilarity(e,r){if(e.length!==r.length)throw new Error("Vectors must have same length");let o=0,n=0,s=0;for(let i=0;i<e.length;i++)o+=e[i]*r[i],n+=e[i]*e[i],s+=r[i]*r[i];return o/(Math.sqrt(n)*Math.sqrt(s))}matchesFilter(e,r){if(!e)return!1;for(let[o,n]of Object.entries(r))if(e[o]!==n)return!1;return!0}getAll(){return Array.from(this.vectors.values())}count(){return this.vectors.size}};var u=require("@neural-tools/core");async function b(t){switch(t.provider){case"local":return new c(t);case"pinecone":throw await(0,u.requireFeature)("vector-db","Vector Database Integration"),new Error("Pinecone integration coming soon. Use local provider for now.");case"qdrant":throw await(0,u.requireFeature)("vector-db","Vector Database Integration"),new Error("Qdrant integration coming soon. Use local provider for now.");case"chromadb":throw await(0,u.requireFeature)("vector-db","Vector Database Integration"),new Error("ChromaDB integration coming soon. Use local provider for now.");default:throw new Error(`Unknown vector store provider: ${t.provider}`)}}async function y(t){let e=V(t),r=384,o=new Array(r).fill(0);for(let s=0;s<r;s++)o[s]=Math.sin(e+s)*Math.cos(e/(s+1));let n=Math.sqrt(o.reduce((s,i)=>s+i*i,0));return o.map(s=>s/n)}function V(t){let e=0;for(let r=0;r<t.length;r++){let o=t.charCodeAt(r);e=(e<<5)-e+o,e=e&e}return Math.abs(e)}function w(t,e={}){let r=e.chunkSize||500,o=e.overlap||50,n=[],s=0;for(;s<t.length;){let i=Math.min(s+r,t.length);n.push(t.slice(s,i)),s+=r-o}return n}0&&(module.exports={LocalVectorStore,VectorStore,chunkText,createEmbedding,createVectorStore});
package/dist/index.mjs ADDED
@@ -0,0 +1 @@
1
+ var a=class{config;constructor(e){this.config=e}};var c=class extends a{vectors=new Map;async connect(){}async upsert(e,r){for(let n of e)this.vectors.set(n.id,n)}async query(e,r){let n=r?.topK||10,s=[];for(let[t,o]of this.vectors){if(r?.filter&&!this.matchesFilter(o.metadata,r.filter))continue;let d=this.cosineSimilarity(e,o.values);s.push({id:t,score:d,metadata:r?.includeMetadata!==!1?o.metadata:void 0,values:r?.includeValues?o.values:void 0})}return s.sort((t,o)=>o.score-t.score),s.slice(0,n)}async delete(e){for(let r of e)this.vectors.delete(r)}async fetch(e){let r=[];for(let n of e){let s=this.vectors.get(n);s&&r.push(s)}return r}async deleteNamespace(){this.vectors.clear()}async disconnect(){}cosineSimilarity(e,r){if(e.length!==r.length)throw new Error("Vectors must have same length");let n=0,s=0,t=0;for(let o=0;o<e.length;o++)n+=e[o]*r[o],s+=e[o]*e[o],t+=r[o]*r[o];return n/(Math.sqrt(s)*Math.sqrt(t))}matchesFilter(e,r){if(!e)return!1;for(let[n,s]of Object.entries(r))if(e[n]!==s)return!1;return!0}getAll(){return Array.from(this.vectors.values())}count(){return this.vectors.size}};import{requireFeature as u}from"@neural-tools/core";async function w(i){switch(i.provider){case"local":return new c(i);case"pinecone":throw await u("vector-db","Vector Database Integration"),new Error("Pinecone integration coming soon. Use local provider for now.");case"qdrant":throw await u("vector-db","Vector Database Integration"),new Error("Qdrant integration coming soon. Use local provider for now.");case"chromadb":throw await u("vector-db","Vector Database Integration"),new Error("ChromaDB integration coming soon. Use local provider for now.");default:throw new Error(`Unknown vector store provider: ${i.provider}`)}}async function P(i){let e=l(i),r=384,n=new Array(r).fill(0);for(let t=0;t<r;t++)n[t]=Math.sin(e+t)*Math.cos(e/(t+1));let s=Math.sqrt(n.reduce((t,o)=>t+o*o,0));return n.map(t=>t/s)}function l(i){let e=0;for(let r=0;r<i.length;r++){let n=i.charCodeAt(r);e=(e<<5)-e+n,e=e&e}return Math.abs(e)}function x(i,e={}){let r=e.chunkSize||500,n=e.overlap||50,s=[],t=0;for(;t<i.length;){let o=Math.min(t+r,i.length);s.push(i.slice(t,o)),t+=r-n}return s}export{c as LocalVectorStore,a as VectorStore,x as chunkText,P as createEmbedding,w as createVectorStore};
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@neural-tools/vector-db",
3
- "version": "0.1.3",
3
+ "version": "0.1.5",
4
4
  "description": "Vector database abstraction layer for Neural Tools",
5
5
  "main": "dist/index.js",
6
6
  "types": "dist/index.d.ts",
@@ -10,11 +10,24 @@
10
10
  },
11
11
  "repository": {
12
12
  "type": "git",
13
- "url": "https://github.com/MacLeanLuke/ai-toolkit.git",
13
+ "url": "https://github.com/MacLeanLuke/neural-tools.git",
14
14
  "directory": "packages/vector-db"
15
15
  },
16
+ "homepage": "https://neural-tools.com/docs/vector-db.html",
17
+ "bugs": {
18
+ "url": "https://github.com/MacLeanLuke/neural-tools/issues"
19
+ },
20
+ "keywords": [
21
+ "vector-database",
22
+ "pinecone",
23
+ "chroma",
24
+ "qdrant",
25
+ "embeddings",
26
+ "neural-tools",
27
+ "ai"
28
+ ],
16
29
  "dependencies": {
17
- "@neural-tools/core": "0.1.3"
30
+ "@neural-tools/core": "0.1.5"
18
31
  },
19
32
  "devDependencies": {
20
33
  "@types/node": "^20.11.5",
@@ -40,8 +53,8 @@
40
53
  "dist"
41
54
  ],
42
55
  "scripts": {
43
- "build": "tsc",
44
- "dev": "tsc --watch",
56
+ "build": "tsup",
57
+ "dev": "tsup --watch",
45
58
  "clean": "rm -rf dist",
46
59
  "test": "echo 'Tests coming soon'"
47
60
  }
@@ -1,30 +0,0 @@
1
- import { VectorStore, Vector, QueryResult, QueryOptions, UpsertOptions } from './types';
2
- /**
3
- * Simple in-memory vector store for local development and testing
4
- */
5
- export declare class LocalVectorStore extends VectorStore {
6
- private vectors;
7
- connect(): Promise<void>;
8
- upsert(vectors: Vector[], options?: UpsertOptions): Promise<void>;
9
- query(vector: number[], options?: QueryOptions): Promise<QueryResult[]>;
10
- delete(ids: string[]): Promise<void>;
11
- fetch(ids: string[]): Promise<Vector[]>;
12
- deleteNamespace(): Promise<void>;
13
- disconnect(): Promise<void>;
14
- /**
15
- * Calculate cosine similarity between two vectors
16
- */
17
- private cosineSimilarity;
18
- /**
19
- * Check if metadata matches filter
20
- */
21
- private matchesFilter;
22
- /**
23
- * Get all vectors (for testing/debugging)
24
- */
25
- getAll(): Vector[];
26
- /**
27
- * Get count of vectors
28
- */
29
- count(): number;
30
- }
@@ -1,103 +0,0 @@
1
- "use strict";
2
- Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.LocalVectorStore = void 0;
4
- const types_1 = require("./types");
5
- /**
6
- * Simple in-memory vector store for local development and testing
7
- */
8
- class LocalVectorStore extends types_1.VectorStore {
9
- vectors = new Map();
10
- async connect() {
11
- // No connection needed for local store
12
- }
13
- async upsert(vectors, options) {
14
- for (const vector of vectors) {
15
- this.vectors.set(vector.id, vector);
16
- }
17
- }
18
- async query(vector, options) {
19
- const topK = options?.topK || 10;
20
- const results = [];
21
- // Calculate similarity for all vectors
22
- for (const [id, storedVector] of this.vectors) {
23
- // Skip if filter doesn't match
24
- if (options?.filter && !this.matchesFilter(storedVector.metadata, options.filter)) {
25
- continue;
26
- }
27
- const score = this.cosineSimilarity(vector, storedVector.values);
28
- results.push({
29
- id,
30
- score,
31
- metadata: options?.includeMetadata !== false ? storedVector.metadata : undefined,
32
- values: options?.includeValues ? storedVector.values : undefined
33
- });
34
- }
35
- // Sort by score descending and take topK
36
- results.sort((a, b) => b.score - a.score);
37
- return results.slice(0, topK);
38
- }
39
- async delete(ids) {
40
- for (const id of ids) {
41
- this.vectors.delete(id);
42
- }
43
- }
44
- async fetch(ids) {
45
- const results = [];
46
- for (const id of ids) {
47
- const vector = this.vectors.get(id);
48
- if (vector) {
49
- results.push(vector);
50
- }
51
- }
52
- return results;
53
- }
54
- async deleteNamespace() {
55
- this.vectors.clear();
56
- }
57
- async disconnect() {
58
- // No cleanup needed for local store
59
- }
60
- /**
61
- * Calculate cosine similarity between two vectors
62
- */
63
- cosineSimilarity(a, b) {
64
- if (a.length !== b.length) {
65
- throw new Error('Vectors must have same length');
66
- }
67
- let dotProduct = 0;
68
- let normA = 0;
69
- let normB = 0;
70
- for (let i = 0; i < a.length; i++) {
71
- dotProduct += a[i] * b[i];
72
- normA += a[i] * a[i];
73
- normB += b[i] * b[i];
74
- }
75
- return dotProduct / (Math.sqrt(normA) * Math.sqrt(normB));
76
- }
77
- /**
78
- * Check if metadata matches filter
79
- */
80
- matchesFilter(metadata, filter) {
81
- if (!metadata)
82
- return false;
83
- for (const [key, value] of Object.entries(filter)) {
84
- if (metadata[key] !== value) {
85
- return false;
86
- }
87
- }
88
- return true;
89
- }
90
- /**
91
- * Get all vectors (for testing/debugging)
92
- */
93
- getAll() {
94
- return Array.from(this.vectors.values());
95
- }
96
- /**
97
- * Get count of vectors
98
- */
99
- count() {
100
- return this.vectors.size;
101
- }
102
- }
103
- exports.LocalVectorStore = LocalVectorStore;
package/dist/types.d.ts DELETED
@@ -1,61 +0,0 @@
1
- export interface VectorDBConfig {
2
- provider: 'pinecone' | 'qdrant' | 'chromadb' | 'local';
3
- apiKey?: string;
4
- endpoint?: string;
5
- dimension?: number;
6
- metric?: 'cosine' | 'euclidean' | 'dotproduct';
7
- namespace?: string;
8
- }
9
- export interface Vector {
10
- id: string;
11
- values: number[];
12
- metadata?: Record<string, any>;
13
- }
14
- export interface QueryResult {
15
- id: string;
16
- score: number;
17
- metadata?: Record<string, any>;
18
- values?: number[];
19
- }
20
- export interface QueryOptions {
21
- topK?: number;
22
- filter?: Record<string, any>;
23
- includeValues?: boolean;
24
- includeMetadata?: boolean;
25
- }
26
- export interface UpsertOptions {
27
- namespace?: string;
28
- batch?: boolean;
29
- }
30
- export declare abstract class VectorStore {
31
- protected config: VectorDBConfig;
32
- constructor(config: VectorDBConfig);
33
- /**
34
- * Initialize the vector store connection
35
- */
36
- abstract connect(): Promise<void>;
37
- /**
38
- * Insert or update vectors
39
- */
40
- abstract upsert(vectors: Vector[], options?: UpsertOptions): Promise<void>;
41
- /**
42
- * Query for similar vectors
43
- */
44
- abstract query(vector: number[], options?: QueryOptions): Promise<QueryResult[]>;
45
- /**
46
- * Delete vectors by ID
47
- */
48
- abstract delete(ids: string[], namespace?: string): Promise<void>;
49
- /**
50
- * Get vector by ID
51
- */
52
- abstract fetch(ids: string[], namespace?: string): Promise<Vector[]>;
53
- /**
54
- * Delete all vectors in a namespace
55
- */
56
- abstract deleteNamespace(namespace: string): Promise<void>;
57
- /**
58
- * Close the connection
59
- */
60
- abstract disconnect(): Promise<void>;
61
- }
package/dist/types.js DELETED
@@ -1,10 +0,0 @@
1
- "use strict";
2
- Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.VectorStore = void 0;
4
- class VectorStore {
5
- config;
6
- constructor(config) {
7
- this.config = config;
8
- }
9
- }
10
- exports.VectorStore = VectorStore;