INNER CODE UNIT · Python

query

handrew/browserpilot · browserpilot/agents/memories/__init__.py:35

    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:
            logger.info("Skipping duplicate text.")
            return
        self.texts.append(text)
        self.index.insert(Document(text=text))

    def save(self, path):
        self.index.storage_context.persist(path)

View source record →