audit_assistant / auditqa /engine /vectorstore.py
ppsingh's picture
Update auditqa/engine/vectorstore.py
607e865 verified
raw
history blame
No virus
299 Bytes
from langchain.embeddings import OpenAIEmbeddings, HuggingFaceEmbeddings, HuggingFaceInferenceAPIEmbeddings
from dotenv import load_dotenv
provider_retrieval_model = "HF"
embeddingmodel = "BAAI/bge-small-en-v1.5"
load_dotenv()
HF_Token = os.environ.get("HF_TOKEN")
def token():
print(HF_Token)