dev #2

Merged
ly merged 17 commits from dev into main 2024-08-23 09:37:06 +08:00
2 changed files with 3 additions and 2 deletions
Showing only changes of commit 4c1c67aa50 - Show all commits
+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)
@@ -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):