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 # Changelog
## [Unreleased] ## [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 ### Removed
- **BREAKING: Custom AG-UI Infrastructure**: Removed custom AG-UI event handling in favor of pydantic-ai's native AG-UI support - **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 import pytest
from haiku.rag.agents.research.dependencies import ResearchContext 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 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() @pytest.mark.vcr()
async def test_graph_end_to_end(allow_model_requests, temp_db_path, qa_corpus): 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.""" """Test research graph with real LLM calls recorded via VCR."""

File diff suppressed because one or more lines are too long