logger for update

Author: Zhe Yu

src/vectorcode/subcommands/update.py

@@ -1,4 +1,5 @@
 import asyncio
+import logging
 import os
 import sys
 from asyncio import Lock
@@ -11,6 +12,8 @@
 from vectorcode.common import get_client, get_collection, verify_ef
 from vectorcode.subcommands.vectorise import chunked_add, show_stats
 
+logger = logging.getLogger(name=__name__)
+
 
 async def update(configs: Config) -> int:
     client = await get_client(configs)
@@ -49,6 +52,7 @@ async def update(configs: Config) -> int:
     with tqdm.tqdm(
         total=len(files), desc="Vectorising files...", disable=configs.pipe
     ) as bar:
+        logger.info(f"Updating embeddings for {len(files)} file(s).")
         try:
             tasks = [
                 asyncio.create_task(
@@ -73,6 +77,7 @@ async def update(configs: Config) -> int:
             return 1
 
     if len(orphanes):
+        logger.info(f"Removing {len(orphanes)} orphaned files from database.")
         await collection.delete(where={"path": {"$in": list(orphanes)}})
 
     show_stats(configs, stats)

src/vectorcode/subcommands/vectorise.py

@@ -64,6 +64,7 @@ async def chunked_add(
         async with collection_lock:
             await collection.delete(where={"path": full_path_str})
 
+    logger.debug(f"Vectorising {file_path}")
     try:
         async with semaphore:
             chunks: list[Chunk | str] = list(
@@ -192,7 +193,7 @@ async def vectorise(configs: Config) -> int:
                 with open(spec_path) as fin:
                     spec = pathspec.GitIgnoreSpec.from_lines(fin.readlines())
                 files = exclude_paths_by_spec((str(i) for i in files), spec)
-    else:
+    else:  # pragma: nocover
         logger.info("Ignoring exclude specs.")
 
     stats = {"add": 0, "update": 0, "removed": 0}