diff --git a/main.py b/main.py
index 30167ce..3a1b7e3 100644
--- a/main.py
+++ b/main.py
@@ -73,8 +73,18 @@ def test_database():
                     break
         print(f"   All records equal: {records_equal}")
         
+        # Test embedding functionality
+        print("\n5. Testing embedding functionality (Ollama API server)...")
+        try:
+            test_embedding = db._embed("This is a test text for embedding.")
+            print(f"   Embedding test PASSED: Generated vector of shape {test_embedding.shape}")
+            ollama_running = True
+        except Exception as e:
+            print(f"   Embedding test FAILED: {e}\n   Did you start ollama docker image?")
+            ollama_running = False
+
         # Summary
-        all_good = vectors_match and records_match and vectors_equal and records_equal
+        all_good = vectors_match and records_match and vectors_equal and records_equal and ollama_running
         print(f"\n✅ Test {'PASSED' if all_good else 'FAILED'}: All data preserved through save/load cycle")
         
     finally: