bstraehle commited on
Commit
c4c7ba9
·
1 Parent(s): 621d0a8

Update rag_llamaindex.py

Browse files
Files changed (1) hide show
  1. rag_llamaindex.py +2 -2
rag_llamaindex.py CHANGED
@@ -63,7 +63,7 @@ class LlamaIndexRAG(BaseRAG):
63
  index_name = self.MONGODB_INDEX_NAME
64
  )
65
 
66
- def get_service_context(config):
67
  return ServiceContext.from_defaults(
68
  chunk_overlap = config["chunk_overlap"],
69
  chunk_size = config["chunk_size"],
@@ -71,7 +71,7 @@ class LlamaIndexRAG(BaseRAG):
71
  llm = self.get_llm(config)
72
  )
73
 
74
- def get_storage_context():
75
  return StorageContext.from_defaults(
76
  vector_store = self.get_vector_store()
77
  )
 
63
  index_name = self.MONGODB_INDEX_NAME
64
  )
65
 
66
+ def get_service_context(self, config):
67
  return ServiceContext.from_defaults(
68
  chunk_overlap = config["chunk_overlap"],
69
  chunk_size = config["chunk_size"],
 
71
  llm = self.get_llm(config)
72
  )
73
 
74
+ def get_storage_context(self):
75
  return StorageContext.from_defaults(
76
  vector_store = self.get_vector_store()
77
  )