Method batchQueryByEmbedding

embedding/embeddingHelper.py:197–224

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

        return result[0]

    def batchQueryByEmbedding(self, embeddings, topK=5, where=None, where_document=None):
        if where is None:
            if where_document is None:
                result = self.collection.query(
                    query_embeddings=embeddings,
                    n_results=topK
                )
            else:
                result = self.collection.query(
                    query_embeddings=embeddings,
                    n_results=topK,
                )
        else:
            if where_document is None:
                result = self.collection.query(
                    query_embeddings=embeddings,
                    n_results=topK,
                    where=where
                )
            else:
                result = self.collection.query(
                    query_embeddings=embeddings,
                    n_results=topK,
                    where=where
                )

        result = self.processResult(result=result)
        return result

    def saveFile(self, fileName, filePath, chunk_size=500, chunk_overlap=0, text_splitter=None, autoSpliter=False,
                 csv_delimiter=",",