feat(workers): Define WorkerState for the Researcher subgraph
This commit is contained in:
13
backend/tests/test_researcher_state.py
Normal file
13
backend/tests/test_researcher_state.py
Normal file
@@ -0,0 +1,13 @@
|
||||
from typing import get_type_hints, List
|
||||
from ea_chatbot.graph.workers.researcher.state import WorkerState
|
||||
|
||||
def test_researcher_worker_state():
|
||||
"""Verify that Researcher WorkerState has the required fields."""
|
||||
hints = get_type_hints(WorkerState)
|
||||
|
||||
assert "messages" in hints
|
||||
assert "task" in hints
|
||||
assert "queries" in hints
|
||||
assert "raw_results" in hints
|
||||
assert "iterations" in hints
|
||||
assert "result" in hints
|
||||
Reference in New Issue
Block a user