Method queryByEmbedding

embedding/embeddingHelper.py:168–195

(self, embedding, topK=5, where=None, where_document=None)

        return result

    def queryByEmbedding(self, embedding, topK=5, where=None, where_document=None):
        if where is None:
            if where_document is None:
                result = self.collection.query(
                    query_embeddings=[embedding],
                    n_results=topK
                )
            else:
                result = self.collection.query(
                    query_texts=[embedding],
                    n_results=topK,
                )
        else:
            if where_document is None:
                result = self.collection.query(
                    query_texts=[embedding],
                    n_results=topK,
                    where=where
                )
            else:
                result = self.collection.query(
                    query_texts=[embedding],
                    n_results=topK,
                    where=where
                )

        result = self.processResult(result=result)
        return result[0]

    def batchQueryByEmbedding(self, embeddings, topK=5, where=None, where_document=None):
        if where is None: