From ee036dc0a8857cbef5c288d9c98b2eb5d8649d34 Mon Sep 17 00:00:00 2001 From: "jinli.yl" Date: Thu, 27 Jun 2024 16:46:10 +0800 Subject: [PATCH] [dev] add logger to test --- memory_scope/cli.py | 2 +- tests/models/test_models_lli_embedding.py | 9 +++++---- 2 files changed, 6 insertions(+), 5 deletions(-) diff --git a/memory_scope/cli.py b/memory_scope/cli.py index a4a04316..a76c61db 100644 --- a/memory_scope/cli.py +++ b/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: diff --git a/tests/models/test_models_lli_embedding.py b/tests/models/test_models_lli_embedding.py index 1bbfae5e..5bd3f28b 100644 --- a/tests/models/test_models_lli_embedding.py +++ b/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)