Update tool/rag.py
Browse files- tool/rag.py +1 -1
tool/rag.py
CHANGED
|
@@ -32,7 +32,7 @@ device = f'cuda:{cuda.current_device()}' if cuda.is_available() else 'cpu'
|
|
| 32 |
from langchain_openai import OpenAIEmbeddings
|
| 33 |
embeddings = OpenAIEmbeddings(api_key='sk-itPrztYm9F6XZZpsBMJB9O7Vq0pYUABVVBSoThuBxEGTnDik',
|
| 34 |
base_url="https://www.dmxapi.com/v1")
|
| 35 |
-
vectorstore=FAISS.load_local(r"tool/rag", embeddings,allow_dangerous_deserialization =True)
|
| 36 |
|
| 37 |
|
| 38 |
template = """
|
|
|
|
| 32 |
from langchain_openai import OpenAIEmbeddings
|
| 33 |
embeddings = OpenAIEmbeddings(api_key='sk-itPrztYm9F6XZZpsBMJB9O7Vq0pYUABVVBSoThuBxEGTnDik',
|
| 34 |
base_url="https://www.dmxapi.com/v1")
|
| 35 |
+
vectorstore=FAISS.load_local(r"https://huggingface.co/spaces/jinysun/TeLLAgent/blob/main/tool/rag", embeddings,allow_dangerous_deserialization =True)
|
| 36 |
|
| 37 |
|
| 38 |
template = """
|