From 1c2ff001103f32dfc6e7d53fff3edccbb81619b2 Mon Sep 17 00:00:00 2001
From: goulustis <cnimaiv@gmail.com>
Date: Mon, 13 Oct 2025 20:02:37 +0800
Subject: [PATCH] add logs

---
 lang_agent/rag/simple.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/lang_agent/rag/simple.py b/lang_agent/rag/simple.py
index d548745..8ecb78c 100644
--- a/lang_agent/rag/simple.py
+++ b/lang_agent/rag/simple.py
@@ -3,6 +3,7 @@ from typing import Type, List
 import tyro
 from mcp.server.fastmcp import FastMCP
 import tyro
+from loguru import logger
 
 from langchain_community.vectorstores import FAISS
 from langchain_core.documents.base import Document
@@ -23,11 +24,16 @@ class SimpleRagConfig(ToolConfig):
     """embedding model name"""
 
     api_key:str = "wrong-key"
-    """api_key for model; for generic text splitting; give a wrong key"""
+    """api_key for model; for generic text splitting; give a wrong key <-- wrong, MUST have api key"""
 
     folder_path:str = "assets/xiaozhan_emb"
     """path to local database"""
 
+    def __post_init__(self):
+        if self.api_key == "wrong-key":
+            logger.info("wrong embedding key, using simple retrieval method")
+
+
 
 class SimpleRag(LangToolBase):
     def __init__(self, config:SimpleRagConfig):