jan.mrna/semantic_doc_search
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Test Ollama API

Browse Source
This commit is contained in:
Mrna
2025-11-03 14:19:57 +01:00
parent b31eb4ebfa
commit 0287e16fd8
1 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
main.py
View File

@@ -73,8 +73,18 @@ def test_database():
break break
print(f" All records equal: {records_equal}") 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 # 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") print(f"\n✅ Test {'PASSED' if all_good else 'FAILED'}: All data preserved through save/load cycle")
finally: finally: