feat(workers): Define WorkerState for the Data Analyst subgraph

This commit is contained in:
Yunxiao Xu
2026-02-23 04:55:50 -08:00
parent eeb2be409b
commit 5324cbe851
4 changed files with 45 additions and 0 deletions

View File

@@ -0,0 +1,17 @@
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