From 2674af5b39b314402644ca6f9af24d11d20a6ebd Mon Sep 17 00:00:00 2001
From: "jinli.yl" <jinli.yl@alibaba-inc.com>
Date: Thu, 18 Jul 2024 17:42:01 +0800
Subject: [PATCH] modify en version by cb

---
 config/demo_config_en.yaml      | 25 +++++++++++++++++++++++--
 tests/worker/test_workers_en.py |  4 ++--
 2 files changed, 25 insertions(+), 4 deletions(-)

diff --git a/config/demo_config_en.yaml b/config/demo_config_en.yaml
index ded645e3..36cc3b93 100644
--- a/config/demo_config_en.yaml
+++ b/config/demo_config_en.yaml
@@ -72,12 +72,14 @@ worker:
   extract_time:
     class: memory.worker.frontend.extract_time_worker
     generation_model: dashscope_generation
+    generation_model_kwargs:
+      top_k: 1
   semantic_rank:
     class: memory.worker.frontend.semantic_rank_worker
     rank_model: dashscope_rank
   fuse_rerank:
     class: memory.worker.frontend.fuse_rerank_worker
-    fuse_score_threshold: 0.1
+    fuse_score_threshold: 0.05
     fuse_ratio_dict:
       conversation: 0.5
       observation: 1
@@ -109,18 +111,26 @@ worker:
   info_filter:
     class: memory.worker.write.info_filter_worker
     generation_model: dashscope_generation
+    generation_model_kwargs:
+      top_k: 1
   load_today_memory:
     class: memory.worker.write.load_memory_worker
     retrieve_today_top_k: 100
   get_observation:
     class: memory.worker.write.get_observation_worker
     generation_model: dashscope_generation
+    generation_model_kwargs:
+      top_k: 1
   get_observation_with_time:
     class: memory.worker.write.get_observation_with_time_worker
     generation_model: dashscope_generation
+    generation_model_kwargs:
+      top_k: 1
   contra_repeat:
     class: memory.worker.write.contra_repeat_worker
     generation_model: dashscope_generation
+    generation_model_kwargs:
+      top_k: 1
   store_memory:
     class: memory.worker.write.update_memory_worker
     method: from_memory_key
@@ -132,16 +142,27 @@ worker:
     retrieve_insight_top_k: 100
   get_reflection_subject:
     class: memory.worker.summary.get_reflection_subject_worker
+    generation_model: dashscope_generation
+    generation_model_kwargs:
+      top_k: 1
   update_insight:
     class: memory.worker.summary.update_insight_worker
+    generation_model: dashscope_generation
+    generation_model_kwargs:
+      top_k: 1
+    rank_model: dashscope_rank
   long_contra_repeat:
     class: memory.worker.summary.long_contra_repeat_worker
+    generation_model: dashscope_generation
+    generation_model_kwargs:
+      top_k: 1
 
 models:
   dashscope_generation:
     class: models.llama_index_generation_model
     module_name: dashscope_generation
     model_name: qwen-max
+    max_tokens: 2000
   dashscope_embedding:
     class: models.llama_index_embedding_model
     module_name: dashscope_embedding
@@ -160,7 +181,7 @@ memory_store:
   embedding_model: dashscope_embedding
   index_name: memory_index
   es_url: http://localhost:9200
-  use_hybrid: false
+  use_hybrid: true
 
 monitor:
   class: storage.dummy_monitor
\ No newline at end of file
diff --git a/tests/worker/test_workers_en.py b/tests/worker/test_workers_en.py
index e035fae8..fa3e9d6a 100644
--- a/tests/worker/test_workers_en.py
+++ b/tests/worker/test_workers_en.py
@@ -13,7 +13,7 @@ from memory_scope.utils.global_context import G_CONTEXT
 from memory_scope.utils.tool_functions import init_instance_by_config
 
 
-class TestWorkersCn(unittest.TestCase):
+class TestWorkersEn(unittest.TestCase):
     """Tests for LLIEmbedding"""
 
     def setUp(self):
@@ -21,7 +21,7 @@ class TestWorkersCn(unittest.TestCase):
         self.logger: Logger = Logger.get_logger(f"test_worker_{datetime_suffix}", to_stream=True)
 
         ms = MemoryScope()
-        ms.load_config("config/demo_config_cn.yaml")
+        ms.load_config("config/demo_config_en.yaml")
         ms.init_global_content_by_config()
 
     def tearDown(self):