sqlite-lembed

Author: asg017

llama.cpp/embedr/BUILD.mk

@@ -22,6 +22,9 @@ o/$(MODE)/llama.cpp/embedr/sqlite3.a: o/$(MODE)/llama.cpp/embedr/sqlite3.o
 o/$(MODE)/llama.cpp/embedr/sqlite-vec.o: llama.cpp/embedr/sqlite-vec.c
 o/$(MODE)/llama.cpp/embedr/sqlite-vec.a: o/$(MODE)/llama.cpp/embedr/sqlite-vec.o
 
+o/$(MODE)/llama.cpp/embedr/sqlite-lembed.o: llama.cpp/embedr/sqlite-lembed.c
+o/$(MODE)/llama.cpp/embedr/sqlite-lembed.a: o/$(MODE)/llama.cpp/embedr/sqlite-lembed.o o/$(MODE)/llama.cpp/llama.cpp.a
+
 o/$(MODE)/llama.cpp/embedr/shell.o: llama.cpp/embedr/shell.c
 o/$(MODE)/llama.cpp/embedr/shell.a: o/$(MODE)/llama.cpp/embedr/shell.o
 
@@ -35,7 +38,8 @@ o/$(MODE)/llama.cpp/embedr/embedr:					\
 		o/$(MODE)/llama.cpp/embedr/embedr.1.asc.zip.o	\
 		o/$(MODE)/llama.cpp/llama.cpp.a \
 		o/$(MODE)/llama.cpp/embedr/sqlite3.a \
-		o/$(MODE)/llama.cpp/embedr/sqlite-vec.a
+		o/$(MODE)/llama.cpp/embedr/sqlite-vec.a \
+		o/$(MODE)/llama.cpp/embedr/sqlite-lembed.a
 
 $(LLAMA_CPP_EMBEDR_OBJS): private CCFLAGS += -DSQLITE_CORE
 

llama.cpp/embedr/embedr.c

@@ -4,16 +4,18 @@
 #include "llamafile/version.h"
 #include "llama.cpp/embedr/sqlite3.h"
 #include "llama.cpp/embedr/sqlite-vec.h"
+#include "llama.cpp/embedr/sqlite-lembed.h"
 #include "llama.cpp/embedr/shell.h"
 #include "string.h"
 int main(int argc, char ** argv) {
     int rc;
     sqlite3* db;
     sqlite3_stmt* stmt;
     rc = sqlite3_auto_extension((void (*)())sqlite3_vec_init);
+    rc = sqlite3_auto_extension((void (*)())sqlite3_lembed_init);
 
     if(argc > 1 &&  (strcmp(argv[1], "sh") == 0)) {
-      return mn(argc, argv);
+      return mn(argc-1, argv+1);
     }
     printf("%d\n", argc);
     printf("llamafile-embed %s, SQLite %s, sqlite-vec=%s, %d\n", LLAMAFILE_VERSION_STRING, sqlite3_version, SQLITE_VEC_VERSION, LLAMA_FTYPE_MOSTLY_Q4_1);
@@ -24,9 +26,9 @@ int main(int argc, char ** argv) {
       return 1;
     }
 
-    rc = sqlite3_prepare_v2(db, "select vec_version()", -1, &stmt, NULL);
+    rc = sqlite3_prepare_v2(db, "select sqlite_version(), vec_version(), lembed_version()", -1, &stmt, NULL);
     if(rc != SQLITE_OK) {
-      printf("a\n");
+      printf("a %s\n", sqlite3_errmsg(db));
       return 1;
     }
     rc = sqlite3_step(stmt);
@@ -35,7 +37,7 @@ int main(int argc, char ** argv) {
       sqlite3_finalize(stmt);
       return 1;
     }
-    printf("x=%s\n", sqlite3_column_text(stmt, 0));
+    printf("%s %s %s\n", sqlite3_column_text(stmt, 0), sqlite3_column_text(stmt, 1), sqlite3_column_text(stmt, 2));
 
     sqlite3_finalize(stmt);