bstraehle commited on
Commit
d571c9b
·
1 Parent(s): 8730199

Update rag_llamaindex.py

Browse files
Files changed (1) hide show
  1. rag_llamaindex.py +1 -1
rag_llamaindex.py CHANGED
@@ -82,7 +82,7 @@ class LlamaIndexRAG:
82
 
83
  def retrieval(self, config, prompt):
84
  index = VectorStoreIndex.from_vector_store(
85
- vector_store = get_vector_store())
86
 
87
  query_engine = index.as_query_engine(
88
  similarity_top_k = config["k"]
 
82
 
83
  def retrieval(self, config, prompt):
84
  index = VectorStoreIndex.from_vector_store(
85
+ vector_store = get_vector_store(self))
86
 
87
  query_engine = index.as_query_engine(
88
  similarity_top_k = config["k"]