修复 model_provider_lite 的 bug

2025-05-06 22:45:14 +08:00 · 2025-05-06 22:45:14 +08:00 · 42fa8c19a1
commit 42fa8c19a1
parent a20143b09c
1 changed files with 2 additions and 2 deletions
--- a/src/core/retriever.py
+++ b/src/core/retriever.py
@ -162,8 +162,8 @@ class Retriever:
    def reco_entities(self, query, history, refs):
        """识别句子中的实体"""
        query = refs.get("rewritten_query", query)
-        model_provider = config.model_provider_lite
-        model_name = config.model_name_lite
+        model_provider = config.model_provider
+        model_name = config.model_name
        model = select_model(model_provider=model_provider, model_name=model_name)

        entities = []