[dev] add logger to test

memory_scope/cli.py

@@ -19,7 +19,7 @@ class CliJob(object):
 
     def __init__(self):
         self.config: Dict[str, Any] = {}
-        self.logger: Logger = Logger.get_logger("cli_job")
+        self.logger: Logger = Logger.get_logger("cli_job", to_stream=False)
 
     def load_config(self, path: str):
         with open(path) as f:

tests/models/test_models_lli_embedding.py

@@ -6,7 +6,7 @@ import asyncio
 import unittest
 
 from memory_scope.models.llama_index_embedding_model import LlamaIndexEmbeddingModel
-
+from memory_scope.utils.logger import Logger
 
 class TestLLIEmbedding(unittest.TestCase):
     """Tests for LlamaIndexEmbeddingModel"""
@@ -18,21 +18,22 @@ class TestLLIEmbedding(unittest.TestCase):
             "clazz": "models.base_embedding_model"
         }
         self.emb = LlamaIndexEmbeddingModel(**config)
+        self.logger = Logger.get_logger()
 
     def test_single_embedding(self):
         text = "您吃了吗？"
         result = self.emb.call(text=text)
-        print(result)
+        self.logger.info(result)
 
     def test_batch_embedding(self):
         texts = ["您吃了吗？",
                  "吃了吗您？"]
         result = self.emb.call(text=texts)
-        print(result)
+        self.logger.info(result)
 
     def test_async_embedding(self):
         texts = ["您吃了吗？",
                  "吃了吗您？"]
         # 调用异步函数并等待其结果
         result = asyncio.run(self.emb.async_call(text=texts))
-        print(result)
+        self.logger.info(result)