@sjovanovic/recall.js 1.0.1 → 1.0.3

package/README.md

@@ -6,13 +6,13 @@
 
 Recall.js is long term memory for AI apps!
 
-It is a generic RAG (Retrieval-augmented generation) JavaScript library and command line utility focused on speed, ease of use and embeddability.
+It is a tool for building RAG (Retrieval-augmented generation) in a form of JavaScript library and command line utility focused on speed, ease of use and embeddability.
 
-It is versatile: use it for generic Semantic Search, as expert memory for your AI app, as a recommendation system, there are so many possibilities.
+It is versatile and you don't have to use it exclusively for RAG, use it for generic Semantic Search, as expert memory for your AI app, as a  recommendation system, there are so many possibilities...
 
 Recall.js supports multilingual embeddings out of the box so you can add data in one language and then query it in another.
 
-Under the hood, recall.js uses sentence vector embeddings and a vector database to index and query your data. It is a light wrapper around local language models such as [MiniLM-L12-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L12-v2) (optionally LLMs can be used) and [CozoDB](https://www.cozodb.org/) vector database.
+Under the hood, recall.js uses sentence vector embeddings and a vector database to index and query your data. It is a light wrapper around local language models such as [MiniLM-L12-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L12-v2) and [CozoDB](https://www.cozodb.org/) vector database.
 
 ## Install
 
@@ -20,7 +20,15 @@ Under the hood, recall.js uses sentence vector embeddings and a vector database
 
 ## Usage
 
-Warning: when this library is used for the first time, it will download a local language model MiniLM-L12-v2 which may take long time depending on your Internet connectivity. Please be patient.
+Console:
+
+```console
+recall --add 'The quick brown fox jumps over the lazy dog|Fox|{"foo":"bar"}'
+recall --query "Un animal saute par-dessus un autre animal" --limit 1
+```
+**Warning:** when this library is used for the first time, it will download a local language model MiniLM-L12-v2 which may take long time depending on your Internet connectivity. Please be patient.
+
+Below is the same example in JavaScript:
 
 ```javascript
 
@@ -68,18 +76,11 @@ response:
 
 ```
 
-Here's how the above example looks like in CLI:
-
-```log
-recall --add 'The quick brown fox jumps over the lazy dog|Fox|{"foo":"bar"}'
-recall --query "Un animal saute par-dessus un autre animal" --limit 1
-```
-
 ## Options
 
-Easiest way to get all the options is via command line:
+Easy way to view all the options is via command line:
 
-```log
+```console
 recall --help
 
 Usage:
@@ -99,12 +100,12 @@ Options:
 --json "FILE_NAME"                     - import from file which has one json object per line: {input:"", result:"", data:{}}
 ```
 
-Note when adding data recall will generate unique id automatically. To set custom id add it as a string property named "id" in the data object (i.e. `{"id":"customID"}`).
+**Note:** when adding data recall will generate unique id automatically. To set custom id add it as a string property named "id" in the data object (i.e. `{"id":"customID"}`).
 
 
 ## JavaScript API Reference
 
-**RECALL.config**
+### RECALL.config
 
 Configuration object.
 
@@ -118,19 +119,19 @@ export const config = {
 }
 ```
 
-**RECALL.getDb()**
+### RECALL.getDb()
 
 Returns reference to the CozoDB instance.
 
-**RECALL.getEmbeddings(text) -> Promise(Array)**
+### RECALL.getEmbeddings(text) -> Promise(Array)
 
 Given text calculates the embeddings vector
 
-**RECALL.add(input, result, data={}) -> Promise(Object)**
+### RECALL.add(input, result, data={}) -> Promise(Object)
 
 Add data. `input` is the sentence to get embeddings from. `result` is the string to show in the results. `data` is arbitrary object intended to hold related pieces of information and references. If `data` object contains `id` property it will be used as unique id of the record.
 
-**RECALL.addBatch(batch) -> Promise(Object)**
+### RECALL.addBatch(batch) -> Promise(Object)
 
 Add data in batches (faster than using add repeteadely). 
 `batch` is an Array that looks like this:
@@ -138,19 +139,19 @@ Add data in batches (faster than using add repeteadely).
 let batch = [{input:"", result:"", data:{}}]
 ```
 
-**RECALL.remove(id) -> Promise(Object)**
+### RECALL.remove(id) -> Promise(Object)
 
 Remove data by id. id is a string.
 
-**RECALL.searchText(text, numResults = 5) ->  Promise(Object)**
+### RECALL.searchText(text, numResults = 5) ->  Promise(Object)
 
 Query the vector database. Accepts query text and number of results to return.
 
-**RECALL.nuke()**
+### RECALL.nuke()
 
 Deletes the database.
 
-**RECALL.importFromJSONStream(fileName) -> Promise(object)**
+### RECALL.importFromJSONStream(fileName) -> Promise(object)
 
 Imports from readable stream or file which consists of JSON objects, one per line. e.g.
 ```
@@ -160,13 +161,13 @@ Imports from readable stream or file which consists of JSON objects, one per lin
 ```
 This is the most efficient way to import data.
 
-**RECALL.importFromCSVorTSV(fileName, inputHeader=null, resultHeader=null) -> Promise()**
+### RECALL.importFromCSVorTSV(fileName, inputHeader=null, resultHeader=null) -> Promise()
 
 Imports from CSV or TSV file. By default fist column is used as input, second as result and remaining columns are put in the data object.
 If `inputHeader` is specified, function will try to find the column by that name and use it as input.
 If `resultHeader` is specified, function will try to find the column by that name and use it as result.
 
-**RECALL.mcp() -> Promise()**
+### RECALL.mcp() -> Promise()
 
 (Experimental)
 Runs MCP server and makes the results available when mentioning `Recall search` in the prompt. Currently only supports STDIO.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sjovanovic/recall.js",
-  "version": "1.0.1",
+  "version": "1.0.3",
   "description": "Semantic search as long term memory for LLMs",
   "main": "recall.js",
   "bin": {

package/recall.js

@@ -23,22 +23,26 @@ export const config = {
     PATH: PATH // directory of recall.js
 }
 
-var db = null
+var db = null, initDone = false
 
 export const getDb = () => {
-    if(!db) db = new CozoDb('sqlite', config.DB_FILE)
+    if(!db) {
+        db = new CozoDb('sqlite', config.DB_FILE)
+    }
     return db
 }
 
 async function printQuery(query, params = {}) {
-    try {
-        if(!db) {
-            getDb()
-            try {
-                let isCreated = await createTable()
-                if(isCreated) console.log('Created embeddings table.')
-            }catch(err) {}
+
+    try{
+        if(!initDone) {
+            initDone = true
+            await createTable()
         }
+    }catch(err) {
+        //console.log('CREATE TABLE ERROR', err)
+    }
+    try {
         let data = getDb().run(query, params)
         return data
     }catch(err){
@@ -57,7 +61,7 @@ export const getEmbeddings = async (text) => {
 
 export const createTable = async () => {
     // create table (id, v, input, result, data)
-    let tableCreated = await printQuery(`:create embeddings {id: String => v: <F32; ${config.VECTOR_SIZE}>, input: String, result: String, data: Json}`)
+    let tableCreated = await printQuery(`:create embeddings {id: String, category: String => v: <F32; ${config.VECTOR_SIZE}>, input: String, result: String, data: Json}`)
     if(tableCreated){
         // create index
         let indexCreated = await printQuery(`::hnsw create embeddings:index_name {
@@ -67,7 +71,6 @@ export const createTable = async () => {
             fields: [v],
             distance: L2, # Cosine, IP
             ef_construction:50, # number of nearest neighbors
-            #filter: k != 'foo', # only those rows for which the expression evaluates to true are indexed
             extend_candidates: false, # include nearest neighbors of the nearest neighbors
             keep_pruned_connections: false,
         }`)
@@ -76,21 +79,15 @@ export const createTable = async () => {
     return false
 }
 
-export const add = async (input, result, data={}) => {
+export const add = async (input, result, data={}, category="") => {
     if(!input || !result) return
 
     input = sanitizeString(input)
     result = sanitizeString(result)
-
-    
-
     const embedding = await getEmbeddings(input)
-
-    console.log('Adding', input, '->', result)
-
     let id = data.id || Math.random().toString().substring(2)
-    return await printQuery(`?[id, v, input, result, data] <- [["${id}", ${JSON.stringify(embedding)}, ${JSON.stringify(input.replaceAll('"', "'"))}, ${JSON.stringify(result.replaceAll('"', "'"))}, ${JSON.stringify(data)} ]]
-        :put embeddings {id => v, input, result, data}
+    return await printQuery(`?[id, v, input, result, data, category] <- [["${id}", ${JSON.stringify(embedding)}, ${JSON.stringify(input.replaceAll('"', "'"))}, ${JSON.stringify(result.replaceAll('"', "'"))}, ${JSON.stringify(data)}, ${JSON.stringify(category.replaceAll('"', "'"))} ]]
+        :put embeddings {id, category => v, input, result, data}
     `)
 }
 
@@ -106,25 +103,26 @@ export const addBatch = async (batch) => {
     if(!batch || !Array.isArray(batch)) return
     let vectorBatch = []
     for(let i=0;i<batch.length; i++){
-        let {input, result, data} = batch[i]
+        let {input, result, data, category} = batch[i]
 
         if(!input || !result) continue
         if(!data) data = {}
+        if(!category) category = ''
         const embedding = await getEmbeddings(input)
         batch[i].embedding = embedding
         let item = ''
         if(i == 0) {
-            item += `?[id, v, input, result, data] <- [`
+            item += `?[id, v, input, result, data, category] <- [`
         }
 
         input = sanitizeString(input)
         result = sanitizeString(result)
 
         let id = data?.id ? data.id : Math.random().toString().substring(2)
-        item += `["${id}", ${JSON.stringify(embedding)}, ${JSON.stringify(input)}, ${JSON.stringify(result)}, ${JSON.stringify(data)} ],`
+        item += `["${id}", ${JSON.stringify(embedding)}, ${JSON.stringify(input)}, ${JSON.stringify(result)}, ${JSON.stringify(data)}, ${JSON.stringify(category)} ],`
         if(i == batch.length-1) {
             item += `]
-            :put embeddings {id => v, input, result, data}`
+            :put embeddings {id, category => v, input, result, data}`
         }
         vectorBatch.push(item)
     }
@@ -135,31 +133,38 @@ const sanitizeString = (str)=>{
     return str.replace(/[\/#$%\^&\*{}=_`~()\"]/g," ").replace(/\s{2,}/g, " ")
 }
 
-export const remove = async (id) => {
+export const remove = async (id, category="") => {
     if(!id || typeof id != 'string') return 
     id.replace(/[^a-zA-Z0-9]/g, '')
     if(!id) return
     let results = await printQuery(
-        `?[id] <- [['${id}']]
+        `?[id, category] <- [['${id}', '${category}']]
         ::remove embeddings {id}`)
     return results
 }
 
-export const searchText = async (text, numResults = 5) => {
+export const searchText = async (text, category="", numResults = 5) => {
     const embedding = await getEmbeddings(text)
-    let results = await printQuery(`?[dist, result, id, data] := ~embeddings:index_name{ id, v, input, result, data |
+    let results = await printQuery(`?[dist, result, id, data, category] := ~embeddings:index_name { id, v, input, result, data, category |
         query: q,
         k: ${numResults}, # number of results
-        ef: 90, # number of neighbours to consider 
+        ef: 50, # number of neighbours to consider 
         bind_distance: dist,
+        filter: category==${JSON.stringify(category)},
         radius: 10.0
     }, q = vec(${JSON.stringify(embedding)})
-    :sort dist`)
+    :sort -dist`)
     return results
 }
 
-export const vectorSearch = async (query, numResults=5) => {
-    return await searchText(query, numResults)
+export const vectorSearch = async (query, category='', numResults=5) => {
+    let result = undefined
+    try{
+        result = await searchText(query, category, numResults)
+    }catch(err){
+        if(config.SHOW_ERRORS) console.error(err.display || err.message)
+    }
+    return result
 }
 
 const cmdArgs = (list = []) => {
@@ -274,19 +279,6 @@ export const importFromCSVorTSV = async (fileName, inputHeader, resultHeader) =>
     
     let results = await fetchFromFile(fileName)
 
-    // // split results to sentences
-    // let results_raw = await fetchFromFile(fileName)
-    // let results = []
-    // for(let i=0;i<results_raw.length; i++){
-    //     let sentences = splitSentences(results_raw[i].input)
-    //     for(let j=0; j<sentences.length; j++){
-    //         results.push({
-    //             ...results_raw[i],
-    //             ...{ input: sentences[j] }
-    //         })
-    //     }
-    // }
-
     let batchSize = 40, batch = [], currentBatch = 0, totalBatches = Math.ceil(results.length / batchSize), dataHeaders = Object.keys(results[results.length-1]).filter(k => k != 'input' && k != 'result'), data
     for(let i=0; i<results.length; i++){
         if(i % batchSize === 0){
@@ -431,18 +423,22 @@ const splitSentences = (text) => {
 }
 
 const runCLI = async () => {
-    let args = cmdArgs(['--query', '-q', '--add', '--db', '--import', '--json', '--mcp', '--nuke', '--input-header', '--result-header', '--test', '--limit'])
+    let args = cmdArgs(['--query', '-q', '--add', '--db', '--import', '--json', '--mcp', '--nuke', '--input-header', '--result-header', '--test', '--limit', '--category'])
     let query = args['--query'] || args['-q']
     if(args['--db']){
         config.DB_FILE = args['--db']
     }
+    let category = ''
+    if(args['--category']) {
+        category = args['--category']
+    }
     if(query){
         let numResults = 5
         if(args['--limit'] && parseInt(args['--limit'])) {
             numResults = parseInt(args['--limit'])
         }
         console.time('Search time')
-        let result = await vectorSearch(query, numResults)
+        let result = await vectorSearch(query, category, numResults)
         console.timeEnd('Search time')
         console.log('Results:')
         console.log(JSON.stringify(result, null, 2))
@@ -451,17 +447,17 @@ const runCLI = async () => {
         if(!input || !result) {
             console.log('Usage:')
             return console.log(args._cmd + `--add 'input|result|{"foo":"bar"}'`)
-        } 
+        }
         let data = {}
         if(dataString) {
             try {data = JSON.parse(dataString)}catch(err) {}
         }
-        let resp = await add(input, result, data)
+        let resp = await add(input, result, data, category)
         console.log(JSON.stringify(resp, null, 2))
     }else if(args['--remove']){
         let id = args['--remove']
         if(!id) return console.log('Please specify ID to remove')
-        let resp = await remove(id)
+        let resp = await remove(id, category)
         console.log(JSON.stringify(resp, null, 2))
     }else if(args['--nuke'] != undefined){
         nuke()
@@ -481,17 +477,18 @@ const runCLI = async () => {
         console.log('Usage:')
         console.log(args._cmd + ' --query "Foo Bar"')
         console.log("\n" + 'Options:')
-        console.log('--query "SEARCH_STRING"                - search')
-        console.log('--limit 2                              - limit number of results (used with --query)')
-        console.log(`--add 'input|result|{"foo":"bar"}'     - add data`)
-        console.log(`--remove 'id'                          - remove data`)
-        console.log(`--nuke                                 - destroy database`)
-        console.log(`--mcp                                  - run as MCP server (experimental)`)
-        console.log(`--db "FILE_NAME"                       - database file (SQLite)`)
-        console.log(`--import "file.csv | file.tsv"         - import from CSV or TSV w/ columns: 1. input 2. result 3. and remaining columns are additional data`)
-        console.log('--input-header "foo"                   - when used with --import designates specific header column as input')
-        console.log('--result-header "bar"                  - when used with --import designates specific header column as result')
-        console.log(`--json "FILE_NAME"                     - import from file which has one json object per line: {input:"", result:"", data:{}}`)
+        console.log('--query "SEARCH_STRING"                    - search')
+        console.log('--limit 2                                  - limit number of results (used with --query)')
+        console.log(`--add 'input|result|{"foo":"bar"}|categ'   - add data`)
+        console.log(`--remove 'id'                              - remove data`)
+        console.log(`--nuke                                     - destroy database`)
+        console.log(`--mcp                                      - run as MCP server (experimental)`)
+        console.log(`--db "FILE_NAME"                           - database file (SQLite)`)
+        console.log(`--import "file.csv | file.tsv"             - import from CSV or TSV w/ columns: 1. input 2. result 3. and remaining columns are additional data`)
+        console.log('--input-header "foo"                       - when used with --import designates specific header column as input')
+        console.log('--result-header "bar"                      - when used with --import designates specific header column as result')
+        console.log(`--json "FILE_NAME"                         - import from file which has one json object per line: {input:"", result:"", data:{}}`)
+        console.log(`--category "CATEGORY"                      - specify category when adding data and to filter by when querying (defaults to empty string)`)
     }
 }