Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
| 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
|
| 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 FAISS.afrom_documents(
|
| 154 |
documents=chunks,
|
| 155 |
embedding=ml_models["embedder"]
|
| 156 |
)
|