diff --git a/memory_scope/models/llama_index_embedding_model.py b/memory_scope/models/llama_index_embedding_model.py
index d79520c7..a2b866dd 100644
--- a/memory_scope/models/llama_index_embedding_model.py
+++ b/memory_scope/models/llama_index_embedding_model.py
@@ -22,6 +22,8 @@ class LlamaIndexEmbeddingModel(BaseModel):
     def after_call(self, model_response: ModelResponse, **kwargs) -> ModelResponse:
         embeddings = model_response.raw
         model_response.embedding_results = embeddings
+        if len(embeddings) == 1:
+            model_response.embedding_results = embeddings[0]
         return model_response
 
     def _call(self, **kwargs) -> ModelResponse:
diff --git a/tests/models/test_models_lli_embedding.py b/tests/models/test_models_lli_embedding.py
index ecc2d401..1c3c5dfe 100644
--- a/tests/models/test_models_lli_embedding.py
+++ b/tests/models/test_models_lli_embedding.py
@@ -2,7 +2,7 @@ import unittest
 from memory_scope.models.llama_index_embedding_model import LlamaIndexEmbeddingModel
 
 class TestLLIEmbedding(unittest.TestCase):
-    """Tests for LLIEmbedding"""
+    """Tests for LlamaIndexEmbeddingModel"""
 
     def setUp(self):
         config = {
diff --git a/tests/models/test_models_lli_generation.py b/tests/models/test_models_lli_generation.py
index bc9fdfc5..644c0e7b 100644
--- a/tests/models/test_models_lli_generation.py
+++ b/tests/models/test_models_lli_generation.py
@@ -4,7 +4,7 @@ from memory_scope.models.llama_index_generation_model import LlamaIndexGeneratio
 
 
 class TestLLILLM(unittest.TestCase):
-    """Tests for LLIEmbedding"""
+    """Tests for LlamaIndexGenerationModel"""
 
     def setUp(self):
         config = {
@@ -44,7 +44,6 @@ class TestLLILLM(unittest.TestCase):
             sys.stdout.flush()
             time.sleep(0.1)
         
-    @unittest.skip('tmp')
     def test_llm_messages(self):
         messages = [{"role": "system", "content": "you are a helpful assistant."},
                     {"role": "user", "content": "你如何看待黄金上涨？"}]
diff --git a/tests/models/test_models_lli_rerank.py b/tests/models/test_models_lli_rerank.py
index fbfd30ae..9b362a78 100644
--- a/tests/models/test_models_lli_rerank.py
+++ b/tests/models/test_models_lli_rerank.py
@@ -2,7 +2,7 @@ import unittest
 from memory_scope.models.llama_index_rerank_model import LlamaIndexRerankModel
 
 class TestLLIReRank(unittest.TestCase):
-    """Tests for LLIEmbedding"""
+    """Tests for LlamaIndexRerankModel"""
 
     def setUp(self):
         config = {