Update rag_llamaindex.py
Browse files- 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 |
)
|