Update tool/rag.py
Browse files- 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=
|
| 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 |
|