Research graph test cassette
This commit is contained in:
parent
7cc561d1db
commit
a9ed178a90
3 changed files with 3746 additions and 0 deletions
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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."""
|
||||
|
|
|
|||
3729
tests/cassettes/test_research_graph/test_graph_end_to_end.yaml
Normal file
3729
tests/cassettes/test_research_graph/test_graph_end_to_end.yaml
Normal file
File diff suppressed because one or more lines are too long
Loading…
Reference in a new issue