Research graph test cassette

This commit is contained in:
Yiorgis Gozadinos 2026-01-12 12:36:06 +02:00
parent 7cc561d1db
commit a9ed178a90
No known key found for this signature in database
3 changed files with 3746 additions and 0 deletions

View file

@ -1,6 +1,14 @@
# Changelog
## [Unreleased]
### Changed
- **BREAKING: Module Reorganization**: Consolidated all agent code under `haiku.rag.agents`
- Moved `haiku.rag.qa``haiku.rag.agents.qa`
- Moved `haiku.rag.graph.research``haiku.rag.agents.research`
- Added `haiku.rag.agents.chat` module with conversational RAG agent (`create_chat_agent`, `SearchAgent`, `ChatDeps`, `ChatSessionState`)
- Deleted `haiku.rag.graph` module (research graph now at `haiku.rag.agents.research.graph`)
### Removed
- **BREAKING: Custom AG-UI Infrastructure**: Removed custom AG-UI event handling in favor of pydantic-ai's native AG-UI support

View file

@ -1,3 +1,5 @@
from pathlib import Path
import pytest
from haiku.rag.agents.research.dependencies import ResearchContext
@ -7,6 +9,13 @@ from haiku.rag.agents.research.state import ResearchDeps, ResearchState
from haiku.rag.client import HaikuRAG
@pytest.fixture(scope="module")
def vcr_cassette_dir():
return str(
Path(__file__).parent.parent.parent / "cassettes" / "test_research_graph"
)
@pytest.mark.vcr()
async def test_graph_end_to_end(allow_model_requests, temp_db_path, qa_corpus):
"""Test research graph with real LLM calls recorded via VCR."""

File diff suppressed because one or more lines are too long