Generate embeddings before clearing

griptape-ai · Sep 9, 2024 · 37a41ba · 37a41ba
1 parent 1e5974c
commit 37a41ba
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/griptape/artifacts/text_artifact.py b/griptape/artifacts/text_artifact.py
@@ -31,11 +31,12 @@ def to_bytes(self) -> bytes:
         return str(self.value).encode(encoding=self.encoding, errors=self.encoding_error_handler)
 
     def generate_embedding(self, driver: BaseEmbeddingDriver) -> list[float]:
+        embedding = driver.embed_string(str(self.value))
+
         if self.embedding is None:
             self.embedding = []
-
         self.embedding.clear()
-        self.embedding.extend(driver.embed_string(str(self.value)))
+        self.embedding.extend(embedding)
 
         return self.embedding