14 lines
442 B
Python
14 lines
442 B
Python
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
|