@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.
- package/dist/index.d.mts +110 -0
- package/dist/index.d.ts +97 -6
- package/dist/index.js +1 -87
- package/dist/index.mjs +1 -0
- package/package.json +18 -5
- package/dist/local-store.d.ts +0 -30
- package/dist/local-store.js +0 -103
- package/dist/types.d.ts +0 -61
- package/dist/types.js +0 -10
package/dist/index.d.mts
ADDED
|
@@ -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
|
-
|
|
2
|
-
|
|
3
|
-
|
|
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
|
-
|
|
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
|
-
|
|
101
|
+
declare function createEmbedding(text: string): Promise<number[]>;
|
|
13
102
|
/**
|
|
14
103
|
* Batch text into chunks for embedding
|
|
15
104
|
*/
|
|
16
|
-
|
|
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
|
+
"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/
|
|
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.
|
|
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": "
|
|
44
|
-
"dev": "
|
|
56
|
+
"build": "tsup",
|
|
57
|
+
"dev": "tsup --watch",
|
|
45
58
|
"clean": "rm -rf dist",
|
|
46
59
|
"test": "echo 'Tests coming soon'"
|
|
47
60
|
}
|
package/dist/local-store.d.ts
DELETED
|
@@ -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
|
-
}
|
package/dist/local-store.js
DELETED
|
@@ -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