From 4c1c67aa5064f697ee74b44ff5f041db9abe4a97 Mon Sep 17 00:00:00 2001 From: chentianrui Date: Thu, 22 Aug 2024 17:06:28 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E5=BC=80=E5=90=AF=E4=BA=86?= =?UTF-8?q?=E6=B7=B7=E5=90=88=E6=A3=80=E7=B4=A2?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- backend/app/engine/engine.py | 3 ++- backend/app/engine/retriever/HybridRetriever.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/backend/app/engine/engine.py b/backend/app/engine/engine.py index 469a68a..379275e 100644 --- a/backend/app/engine/engine.py +++ b/backend/app/engine/engine.py @@ -39,7 +39,8 @@ def makeDescriptionByEngine(sql_database:SQLDatabase): return table_schema_objs def get_Retriever(index,**kwargs): - bEnableHybrid = True if os.getenv("HYBRID_ENABLED",False) == True else False + strEnableHybrid = os.getenv("HYBRID_ENABLED",'False') + bEnableHybrid = True if strEnableHybrid is not None and strEnableHybrid.title() == 'True' else False if bEnableHybrid: alpha = float(os.getenv("HYBRID_ALPHA", "0.5")) retriever = HybridRetriever(index,alpha = alpha,**kwargs) diff --git a/backend/app/engine/retriever/HybridRetriever.py b/backend/app/engine/retriever/HybridRetriever.py index b3e9dfb..4bf0b8d 100644 --- a/backend/app/engine/retriever/HybridRetriever.py +++ b/backend/app/engine/retriever/HybridRetriever.py @@ -4,7 +4,7 @@ from typing import Optional, Any, Dict, List from llama_index.core.base.base_retriever import BaseRetriever from llama_index.core.schema import NodeWithScore, QueryBundle -from app.engine.retriever import CHBM25Retriever +from app.engine.retriever.CHBM25Retriever import CHBM25Retriever class HybridRetriever(BaseRetriever):