bstraehle commited on
Commit
32dedd9
1 Parent(s): cc34500

Update rag_llamaindex.py

Browse files
Files changed (1) hide show
  1. 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 LangchainPromptTemplate
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 = LangchainPromptTemplate(
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