From 151da7fabfc251588b4c978edf7fb3c807c2f08f Mon Sep 17 00:00:00 2001
From: "jinli.yl" <jinli.yl@alibaba-inc.com>
Date: Fri, 2 Aug 2024 17:18:08 +0800
Subject: [PATCH] add embedding model to update_insight worker

---
 memoryscope/core/config/demo_config.yaml | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/memoryscope/core/config/demo_config.yaml b/memoryscope/core/config/demo_config.yaml
index 7c3a0030..73db079c 100644
--- a/memoryscope/core/config/demo_config.yaml
+++ b/memoryscope/core/config/demo_config.yaml
@@ -7,6 +7,7 @@ global:
   enable_ranker: false
   enable_today_contra_repeat: true
   enable_long_contra_repeat: false
+  output_memory_max_count: 20
 
 memory_chat:
   cli_memory_chat:
@@ -145,6 +146,7 @@ worker:
     class: core.worker.backend.update_insight_worker
     generation_model: generation_model
     rank_model: rank_model
+    embedding_model: embedding_model
     update_insight_threshold: 0.5
   long_contra_repeat:
     class: core.worker.backend.long_contra_repeat_worker