MikeMann commited on
Commit
a73c6b3
·
1 Parent(s): 750ba2f

added EvalDataset Generation

Browse files
Files changed (1) hide show
  1. app.py +5 -2
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)