Ajaykanth Maddi commited on
Commit
af1da78
·
1 Parent(s): 407ca13

Code Changes - Updated the imports

Browse files
Files changed (1) hide show
  1. ragbench.py +4 -2
ragbench.py CHANGED
@@ -1,13 +1,15 @@
1
  import os
2
  import pickle
3
- from typing import List, Dict, Any
4
  import re, json
5
  from dataclasses import dataclass
6
  from nltk.tokenize import sent_tokenize
7
  from langchain_community.embeddings import HuggingFaceEmbeddings
8
  from huggingface_hub import InferenceClient
9
  from langchain_community.vectorstores import FAISS
10
- from langchain_community.retrievers import BM25Retriever, EnsembleRetriever
 
 
11
  from langchain.schema import Document as LangchainDocument
12
 
13
  import torch
 
1
  import os
2
  import pickle
3
+ from typing import List
4
  import re, json
5
  from dataclasses import dataclass
6
  from nltk.tokenize import sent_tokenize
7
  from langchain_community.embeddings import HuggingFaceEmbeddings
8
  from huggingface_hub import InferenceClient
9
  from langchain_community.vectorstores import FAISS
10
+ from langchain_community.retrievers import BM25Retriever
11
+ from langchain.retrievers import EnsembleRetriever # Main LangChain package
12
+
13
  from langchain.schema import Document as LangchainDocument
14
 
15
  import torch