From 1db8577ca695e53f829a4df44cd6dec4251e4981 Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com>
Date: Thu, 15 May 2025 21:53:10 +0800
Subject: [PATCH] perf: Refine the Model Manager code (#3091)

---
 apps/common/config/embedding_config.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/apps/common/config/embedding_config.py b/apps/common/config/embedding_config.py
index d360221c0..03d585e2d 100644
--- a/apps/common/config/embedding_config.py
+++ b/apps/common/config/embedding_config.py
@@ -25,16 +25,14 @@ class ModelManage:
             with _lock:
                 model_instance = get_model(_id)
                 ModelManage.cache.set(_id, model_instance, timeout=60 * 60 * 8)
-                ModelManage.clear_timeout_cache()
-                return model_instance
         else:
             if model_instance.is_cache_model():
                 ModelManage.cache.touch(_id, timeout=60 * 60 * 8)
-                return model_instance
             else:
                 model_instance = get_model(_id)
                 ModelManage.cache.set(_id, model_instance, timeout=60 * 60 * 8)
-                return model_instance
+        ModelManage.clear_timeout_cache()
+        return model_instance
 
     @staticmethod
     def clear_timeout_cache():