Spaces:
Sleeping
Sleeping
from core.lifecycle import Lifecycle | |
from llama.context import ServiceContextManager | |
from llama_index.indices.vector_store import VectorStoreIndex | |
from typing import Optional | |
class IndexManager(Lifecycle): | |
index: Optional[VectorStoreIndex] | |
def __init__(self, context_manager: ServiceContextManager) -> None: | |
super().__init__() | |
self.index = None | |
self.context_manager = context_manager | |
def get_index(self) -> Optional[VectorStoreIndex]: | |
if not self.lifecycle_state.is_started(): | |
raise Exception("Lifecycle state is not correct") | |
return self.index | |