fixed error of logs not printing out

Author: codinglabsong

src/any_chatbot/agent.py

@@ -1,6 +1,6 @@
 import argparse
-import random
 import sqlite3
+import logging
 from pathlib import Path
 
 from langgraph.prebuilt import create_react_agent
@@ -12,6 +12,8 @@
 from any_chatbot.prompts import system_message
 from any_chatbot.utils import load_environ_vars
 
+logger = logging.getLogger(__name__)
+
 BASE = Path(__file__).parent.parent.parent
 
 
@@ -22,7 +24,7 @@ def parse_args() -> argparse.Namespace:
         "--ask",
         type=str,
         default=(
-            "What kinds (text docs, images, or excel sheets) are available in the documents I have provided to you?\n\n"
+            "What kinds (text docs, images, or excel/CSV sheets) are available in the documents I have provided to you?\n\n"
         ),
         help="Your input prompt to the agent.",
     )
@@ -34,7 +36,7 @@ def parse_args() -> argparse.Namespace:
     p.add_argument(
         "--thread_id",
         type=str,
-        default=str(random.random()),
+        default="thread123",
         help="Your conversation history ID. Different IDs save different chat histories with agent.",
     )
     p.add_argument(
@@ -53,6 +55,7 @@ def parse_args() -> argparse.Namespace:
 
 
 def main() -> None:
+    logging.basicConfig(level=logging.INFO)
     cfg = parse_args()
     load_environ_vars()
     # INDEXING

src/any_chatbot/indexing.py

@@ -41,7 +41,7 @@ def load_and_split_text_docs(data_dir):
         glob=globs,
         loader_cls=UnstructuredFileLoader,
     )
-    logger.info(f"Loading text files from {data_dir}")
+    logger.info("Loading text files...")
     docs = loader.load()
     logger.info(f"Loaded {len(docs)} text files")
     # split
@@ -79,7 +79,7 @@ def load_image_docs_as_text(data_dir):
         glob=globs,
         loader_cls=UnstructuredFileLoader,
     )
-    logger.info(f"Loading images from {data_dir}")
+    logger.info("Loading images' OCR texts...")
     image_text_docs = loader.load()
     logger.info(f"Loaded {len(image_text_docs)} image files")
     # tag
@@ -107,7 +107,8 @@ def build_duckdb_and_summary_cards(
     if not any(next(data_dir.rglob(p), None) for p in patterns):
         logger.info(f"No CSV or Excel files found under {data_dir}; skipping.")
         return summary_cards
-    logger.info(f"Detected CSV or Excel files under {data_dir}")
+    logger.info(f"Detected CSV/Excel files under {data_dir}")
+    logger.info("Loading CSV/Excel files...")
     # ensure the DB folder exists
     os.makedirs(db_path.parent, exist_ok=True)
     # empty the entire DB