Rauhan commited on
Commit
9801927
·
verified ·
1 Parent(s): 9e8e7cd

Update functions.py

Browse files
Files changed (1) hide show
  1. functions.py +2 -2
functions.py CHANGED
@@ -15,7 +15,7 @@ from pandasai import SmartDataframe
15
  from langchain_core.chat_history import BaseChatMessageHistory
16
  from langchain_community.document_loaders import YoutubeLoader
17
  from langchain.docstore.document import Document
18
- from langchain_huggingface import HuggingFaceEmbeddings
19
  from langchain_qdrant import FastEmbedSparse
20
  from supabase.client import create_client
21
  from qdrant_client import QdrantClient
@@ -38,7 +38,7 @@ client = create_client(os.environ["SUPABASE_URL"], os.environ["SUPABASE_KEY"])
38
  qdrantClient = QdrantClient(url=os.environ["QDRANT_URL"], api_key=os.environ["QDRANT_API_KEY"])
39
  model_kwargs = {"device": "cuda"}
40
  encode_kwargs = {"normalize_embeddings": True}
41
- vectorEmbeddings = HuggingFaceEmbeddings(
42
  model_name="BAAI/bge-large-en-v1.5",
43
  model_kwargs=model_kwargs,
44
  encode_kwargs=encode_kwargs
 
15
  from langchain_core.chat_history import BaseChatMessageHistory
16
  from langchain_community.document_loaders import YoutubeLoader
17
  from langchain.docstore.document import Document
18
+ from langchain_huggingface import HuggingFaceBgeEmbeddings
19
  from langchain_qdrant import FastEmbedSparse
20
  from supabase.client import create_client
21
  from qdrant_client import QdrantClient
 
38
  qdrantClient = QdrantClient(url=os.environ["QDRANT_URL"], api_key=os.environ["QDRANT_API_KEY"])
39
  model_kwargs = {"device": "cuda"}
40
  encode_kwargs = {"normalize_embeddings": True}
41
+ vectorEmbeddings = HuggingFaceBgeEmbeddings(
42
  model_name="BAAI/bge-large-en-v1.5",
43
  model_kwargs=model_kwargs,
44
  encode_kwargs=encode_kwargs