Merge pull request #2 from chinu0609/fix-for-newer-version-ollama

Fix for newer version ollama

cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py

@@ -111,7 +111,7 @@ class OllamaEmbeddingEngine(EmbeddingEngine):
                 self.endpoint, json=payload, headers=headers, timeout=60.0
             ) as response:
                 data = await response.json()
-                return data["embeddings"]
+                return data["embeddings"][0]
 
     def get_vector_size(self) -> int:
         """

cognee/infrastructure/databases/vector/lancedb/LanceDBAdapter.py

@@ -182,8 +182,6 @@ class LanceDBAdapter(VectorDBInterface):
             properties = get_own_properties(data_point)
             properties["id"] = str(properties["id"])
             
-            if isinstance(vector, list) and len(vector) == 1 and isinstance(vector[0], list):
-                vector = vector[0]
             return LanceDataPoint[str, self.get_data_point_schema(type(data_point))](
                 id=str(data_point.id),
                 vector=vector,