jinysun commited on
Commit
632ea55
·
verified ·
1 Parent(s): a99dabc

Update tool/rag.py

Browse files
Files changed (1) hide show
  1. tool/rag.py +1 -1
tool/rag.py CHANGED
@@ -30,7 +30,7 @@ from langchain_community.vectorstores import FAISS
30
  from torch import cuda, bfloat16
31
  device = f'cuda:{cuda.current_device()}' if cuda.is_available() else 'cpu'
32
  from langchain_openai import OpenAIEmbeddings
33
- embeddings = OpenAIEmbeddings(api_key=api_key=os.getenv("OPENAI_API_KEY"),
34
  base_url=os.getenv("OPENAI_API_BASE"))
35
  vectorstore=FAISS.load_local(r"rag", embeddings,allow_dangerous_deserialization =True)
36
 
 
30
  from torch import cuda, bfloat16
31
  device = f'cuda:{cuda.current_device()}' if cuda.is_available() else 'cpu'
32
  from langchain_openai import OpenAIEmbeddings
33
+ embeddings = OpenAIEmbeddings(api_key=os.getenv("OPENAI_API_KEY"),
34
  base_url=os.getenv("OPENAI_API_BASE"))
35
  vectorstore=FAISS.load_local(r"rag", embeddings,allow_dangerous_deserialization =True)
36