Update rag_llamaindex.py
Browse files- rag_llamaindex.py +2 -2
rag_llamaindex.py
CHANGED
@@ -5,7 +5,7 @@ from llama_index import download_loader, PromptTemplate, ServiceContext
|
|
5 |
from llama_index.embeddings import OpenAIEmbedding
|
6 |
from llama_index.indices.vector_store.base import VectorStoreIndex
|
7 |
from llama_index.llms import OpenAI
|
8 |
-
from llama_index.prompts import
|
9 |
from llama_index.storage.storage_context import StorageContext
|
10 |
from llama_index.vector_stores.mongodb import MongoDBAtlasVectorSearch
|
11 |
|
@@ -105,7 +105,7 @@ class LlamaIndexRAG(BaseRAG):
|
|
105 |
)
|
106 |
|
107 |
###
|
108 |
-
prompt_template =
|
109 |
os.environ["RAG_TEMPLATE"]
|
110 |
)
|
111 |
|
|
|
5 |
from llama_index.embeddings import OpenAIEmbedding
|
6 |
from llama_index.indices.vector_store.base import VectorStoreIndex
|
7 |
from llama_index.llms import OpenAI
|
8 |
+
from llama_index.prompts import PromptTemplate
|
9 |
from llama_index.storage.storage_context import StorageContext
|
10 |
from llama_index.vector_stores.mongodb import MongoDBAtlasVectorSearch
|
11 |
|
|
|
105 |
)
|
106 |
|
107 |
###
|
108 |
+
prompt_template = PromptTemplate(
|
109 |
os.environ["RAG_TEMPLATE"]
|
110 |
)
|
111 |
|