Changes
1 changed files (+6/-7)
-
-
@@ -14,9 +14,7 @@ import modelprint("Connecting to DB") # Possible corruption? con = sqlite3.connect("index.db", check_same_thread=False) con.execute("PRAGMA journal_mode=wal") con.enable_load_extension(True) sqlite_vec.load(con) cur = con.cursor()
-
@@ -27,7 +25,6 @@ cur.execute("CREATE VIRTUAL TABLE IF NOT EXISTS emb USING vec0(id INTEGER PRIMARY KEY, embedding float[1024] distance_metric=cosine)" ) con.commit() lock = threading.Lock() def get_parent(path):
-
@@ -37,17 +34,20 @@ def get_parent(path):class EventHandler(FileSystemEventHandler): lock = threading.Lock() def dispatch(self, event): lock.acquire() self.lock.acquire() print(event) super().dispatch(event) lock.release() self.lock.release() def on_created(self, event): index(event.src_path, get_parent(event.src_path)) def on_modified(self, event): self.on_created(event) if not event.is_directory: self.on_created(event) def on_deleted(self, event): res = cur.execute("SELECT id FROM idx WHERE path = ?", (event.src_path,))
-
@@ -186,4 +186,3 @@ print("Starting RPC server")server = SimpleXMLRPCServer(("localhost", int(os.environ.get("PORT", 8000)))) server.register_function(search) server.serve_forever() con.close()
-