[Misc] Lint code and fix code smells (#1871)

2024-09-16 17:39:54 -07:00
parent 0a78cb9f7a
commit 55c54beeab
57 changed files with 1178 additions and 1357 deletions
--- a/mem0/embeddings/openai.py
+++ b/mem0/embeddings/openai.py
@@ -29,8 +29,4 @@ class OpenAIEmbedding(EmbeddingBase):
            list: The embedding vector.
        """
        text = text.replace("\n", " ")
-        return (
-            self.client.embeddings.create(input=[text], model=self.config.model)
-            .data[0]
-            .embedding
-        )
+        return self.client.embeddings.create(input=[text], model=self.config.model).data[0].embedding