ly/zjdataai-app
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

增加开启了混合检索

Browse Source
This commit is contained in:
chentianrui
2024-08-22 17:06:28 +08:00
parent 59ef831a41
commit 4c1c67aa50
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/app/engine/engine.py
+2 -1
View File
@@ -39,7 +39,8 @@ def makeDescriptionByEngine(sql_database:SQLDatabase):
return table_schema_objs return table_schema_objs
def get_Retriever(index,**kwargs): 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: if bEnableHybrid:
alpha = float(os.getenv("HYBRID_ALPHA", "0.5")) alpha = float(os.getenv("HYBRID_ALPHA", "0.5"))
retriever = HybridRetriever(index,alpha = alpha,**kwargs) retriever = HybridRetriever(index,alpha = alpha,**kwargs)
backend/app/engine/retriever/HybridRetriever.py
+1 -1
View File
@@ -4,7 +4,7 @@ from typing import Optional, Any, Dict, List
from llama_index.core.base.base_retriever import BaseRetriever from llama_index.core.base.base_retriever import BaseRetriever
from llama_index.core.schema import NodeWithScore, QueryBundle from llama_index.core.schema import NodeWithScore, QueryBundle
from app.engine.retriever import CHBM25Retriever from app.engine.retriever.CHBM25Retriever import CHBM25Retriever
class HybridRetriever(BaseRetriever): class HybridRetriever(BaseRetriever):