singhankur01 commited on
Commit
bde0d57
·
verified ·
1 Parent(s): 0b810e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -15,11 +15,11 @@ from fastapi.responses import JSONResponse
15
  from utils.DocsLoader import load_and_chunk
16
  from utils.Schemas import RunRequest, RunResponse
17
  # from concurrent.futures import ThreadPoolExecutor
18
- from langchain_community.vectorstores import FAISS
19
  from langchain.schema import Document
20
  from langchain_google_genai import ChatGoogleGenerativeAI
21
  from langchain_huggingface import HuggingFaceEmbeddings # Correct new import
22
- # from langchain_chroma import Chroma
23
  from langchain_community.retrievers import BM25Retriever
24
  from langchain.retrievers import EnsembleRetriever, ContextualCompressionRetriever
25
  from langchain.retrievers.document_compressors import CrossEncoderReranker
@@ -150,7 +150,7 @@ async def run_hackrx(req: RunRequest):
150
 
151
  ####code for parallel####################################################################################
152
 
153
- vectorstore = await FAISS.afrom_documents(
154
  documents=chunks,
155
  embedding=ml_models["embedder"]
156
  )
 
15
  from utils.DocsLoader import load_and_chunk
16
  from utils.Schemas import RunRequest, RunResponse
17
  # from concurrent.futures import ThreadPoolExecutor
18
+ # from langchain_community.vectorstores import FAISS
19
  from langchain.schema import Document
20
  from langchain_google_genai import ChatGoogleGenerativeAI
21
  from langchain_huggingface import HuggingFaceEmbeddings # Correct new import
22
+ from langchain_chroma import Chroma
23
  from langchain_community.retrievers import BM25Retriever
24
  from langchain.retrievers import EnsembleRetriever, ContextualCompressionRetriever
25
  from langchain.retrievers.document_compressors import CrossEncoderReranker
 
150
 
151
  ####code for parallel####################################################################################
152
 
153
+ vectorstore = await Chroma.afrom_documents(
154
  documents=chunks,
155
  embedding=ml_models["embedder"]
156
  )