18 lines
561 B
Python
18 lines
561 B
Python
from typing import get_type_hints, List
|
|
from langchain_core.messages import BaseMessage
|
|
from ea_chatbot.graph.workers.data_analyst.state import WorkerState
|
|
|
|
def test_data_analyst_worker_state():
|
|
"""Verify that DataAnalyst WorkerState has the required fields."""
|
|
hints = get_type_hints(WorkerState)
|
|
|
|
assert "messages" in hints
|
|
assert "task" in hints
|
|
assert hints["task"] == str
|
|
|
|
assert "code" in hints
|
|
assert "output" in hints
|
|
assert "error" in hints
|
|
assert "iterations" in hints
|
|
assert hints["iterations"] == int
|