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

Stop hitting the embedder in test_sandbox_toc fixtures

Browse source
This commit is contained in:
Yiorgis Gozadinos 2026-05-18 15:55:44 +03:00
parent 3b6bef3bfb
commit 84e91097a8
No known key found for this signature in database
1 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
tests/agents/analysis/test_sandbox_toc.py
View file

@ -16,14 +16,17 @@ from haiku.rag.agents.analysis.dependencies import AnalysisContext
from haiku.rag.agents.analysis.sandbox import Sandbox
from haiku.rag.client import HaikuRAG
from haiku.rag.config.models import AppConfig
from haiku.rag.store.models.document import Document
from haiku.rag.store.models.document_item import DocumentItem
async def _empty_doc(client, *, uri: str, title: str) -> str:
"""Create a Document row and drop the auto-extracted items so the test
controls the items table exactly. Returns the document id."""
doc = await client.create_document(content="x", uri=uri, title=title)
await client.document_item_repository.delete_by_document_id(doc.id)
"""Create a Document row directly via the repository (no chunking, no
embedder) so these tests can run without a reachable embedding endpoint.
Returns the document id."""
doc = await client.document_repository.create(
Document(content="x", uri=uri, title=title)
)
return doc.id