INNER CODE UNIT · Python
__init__
handrew/browserpilot · browserpilot/agents/memories/__init__.py:23
def __init__(self, memory_folder=None, index_type="vector"):
assert index_type in INDEX_TYPES, f"Invalid index type: {index_type}"
self.texts = []
if memory_folder and os.path.exists(memory_folder):
logger.info("Loading memory from disk.")
storage_context = StorageContext.from_defaults(persist_dir=memory_folder)
self.index = load_index_from_storage(storage_context)
else:
self.index = INDEX_TYPES[index_type].from_documents([])
def query(self, prompt, similarity_top_k=3):
query_engine = self.index.as_query_engine(similarity_top_k=similarity_top_k)
return query_engine.query(prompt)
def add(self, text):
if text in self.texts: