dev #2
@@ -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):
|
||||||
|
|||||||
Reference in New Issue
Block a user