| @@ -60,14 +60,17 @@ class CacheEmbedding(Embeddings): | |||
| db.session.rollback() | |||
| except Exception as e: | |||
| logging.exception('Failed transform embedding: ', e) | |||
| cache_embeddings = [] | |||
| for i, embedding in zip(embedding_queue_indices, embedding_queue_embeddings): | |||
| text_embeddings[i] = embedding | |||
| hash = helper.generate_text_hash(texts[i]) | |||
| embedding_cache = Embedding(model_name=self._model_instance.model, | |||
| hash=hash, | |||
| provider_name=self._model_instance.provider) | |||
| embedding_cache.set_embedding(embedding) | |||
| db.session.add(embedding_cache) | |||
| if hash not in cache_embeddings: | |||
| embedding_cache = Embedding(model_name=self._model_instance.model, | |||
| hash=hash, | |||
| provider_name=self._model_instance.provider) | |||
| embedding_cache.set_embedding(embedding) | |||
| db.session.add(embedding_cache) | |||
| cache_embeddings.append(hash) | |||
| db.session.commit() | |||
| except Exception as ex: | |||
| db.session.rollback() | |||