danvics/haiku.rag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Make test_client_resolve_document not require embeddings

Browse source
This commit is contained in:
Yiorgis Gozadinos 2026-02-06 10:33:19 +01:00
parent a0c4ddbd01
commit dc83dde9ab
No known key found for this signature in database
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/test_client.py
View file

@ -88,11 +88,13 @@ async def test_client_document_crud(qa_corpus: Dataset, temp_db_path):
async def test_client_resolve_document(temp_db_path): async def test_client_resolve_document(temp_db_path):
"""Test resolve_document finds documents by ID, title, or URI.""" """Test resolve_document finds documents by ID, title, or URI."""
async with HaikuRAG(temp_db_path, create=True) as client: async with HaikuRAG(temp_db_path, create=True) as client:
doc = await client.create_document( # Insert document directly via repository (no embeddings needed)
doc = Document(
content="Test content", content="Test content",
uri="test://resolve-test", uri="test://resolve-test",
title="Resolve Test Doc", title="Resolve Test Doc",
) )
doc = await client.document_repository.create(doc)
# Resolve by ID # Resolve by ID
by_id = await client.resolve_document(doc.id) by_id = await client.resolve_document(doc.id)