优化代码-2

project/embeddings/local_embed.py

@@ -6,7 +6,7 @@ from typing import Any, Dict, List, Optional
 from langchain.embeddings.base import Embeddings
 from langchain.pydantic_v1 import BaseModel, root_validator
 from langchain.utils import get_from_dict_or_env
-from FlagEmbedding import LLMEmbedder
+from FlagEmbedding import FlagModel
 
 logger = logging.getLogger(__name__)
 
@@ -44,7 +44,7 @@ class LocalEmbed(BaseModel, Embeddings):
             "ZHIPUAI_API_KEY",
         )
 
-        values["client"] = LLMEmbedder('BAAI/bge-large-zh-v1.5',
+        values["client"] = FlagModel('BAAI/bge-large-zh-v1.5',
                                      query_instruction_for_retrieval="为这个句子生成表示以用于检索相关文章：",
                                      use_fp16=True)
         return values