added EvalDataset Generation
Browse files
app.py
CHANGED
@@ -30,7 +30,9 @@ from langchain_community.llms.huggingface_pipeline import HuggingFacePipeline
|
|
30 |
from langchain_core.indexing import index
|
31 |
from langchain_core.vectorstores import VectorStore
|
32 |
from llama_index.core.node_parser import TextSplitter
|
33 |
-
from langchain.retrievers import BM25Retriever, EnsembleRetriever
|
|
|
|
|
34 |
from llama_index.legacy.vector_stores import FaissVectorStore
|
35 |
from pycparser.ply.yacc import token
|
36 |
from ragatouille import RAGPretrainedModel
|
@@ -329,6 +331,7 @@ class BSIChatbot:
|
|
329 |
def retrieval(self, query, rerankingStep, hybridSearch):
|
330 |
global vectorstore
|
331 |
global bm25_retriever
|
|
|
332 |
if hybridSearch == True:
|
333 |
allDocs = self.retrieveDocFromFaiss()
|
334 |
if bm25_retriever == None:
|
@@ -363,7 +366,7 @@ class BSIChatbot:
|
|
363 |
print("initializing Reranker-Model..")
|
364 |
self.initializeRerankingModel()
|
365 |
print("Starting Reranking Chunks...")
|
366 |
-
rerankingModel
|
367 |
#newprint("DBG:retrieved_chunks_text")
|
368 |
#newprint(type(retrieved_chunks_text))
|
369 |
#new print(retrieved_chunks_text)
|
|
|
30 |
from langchain_core.indexing import index
|
31 |
from langchain_core.vectorstores import VectorStore
|
32 |
from llama_index.core.node_parser import TextSplitter
|
33 |
+
#from langchain.retrievers import BM25Retriever, EnsembleRetriever
|
34 |
+
from langchain.retrievers import EnsembleRetriever
|
35 |
+
from langchain_community.retrievers import BM25Retriever
|
36 |
from llama_index.legacy.vector_stores import FaissVectorStore
|
37 |
from pycparser.ply.yacc import token
|
38 |
from ragatouille import RAGPretrainedModel
|
|
|
331 |
def retrieval(self, query, rerankingStep, hybridSearch):
|
332 |
global vectorstore
|
333 |
global bm25_retriever
|
334 |
+
global rerankingModel
|
335 |
if hybridSearch == True:
|
336 |
allDocs = self.retrieveDocFromFaiss()
|
337 |
if bm25_retriever == None:
|
|
|
366 |
print("initializing Reranker-Model..")
|
367 |
self.initializeRerankingModel()
|
368 |
print("Starting Reranking Chunks...")
|
369 |
+
#rerankingModel
|
370 |
#newprint("DBG:retrieved_chunks_text")
|
371 |
#newprint(type(retrieved_chunks_text))
|
372 |
#new print(retrieved_chunks_text)
|