From 0287e16fd87d82a4f481c15f394b26c2d1f7be98 Mon Sep 17 00:00:00 2001 From: Mrna Date: Mon, 3 Nov 2025 14:19:57 +0100 Subject: [PATCH] Test Ollama API --- main.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) 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: