feat: implement mvp with email-first login flow and langgraph architecture

This commit is contained in:
Yunxiao Xu
2026-02-09 23:22:30 -08:00
parent af227d40e6
commit 5a943b902a
79 changed files with 8200 additions and 1 deletions

View File

@@ -0,0 +1,45 @@
import pytest
import logging
from unittest.mock import MagicMock, patch
from ea_chatbot.graph.nodes.query_analyzer import query_analyzer_node, QueryAnalysis
@pytest.fixture
def mock_state():
return {
"messages": [],
"question": "Show me the 2024 results for Florida",
"analysis": None,
"next_action": ""
}
@patch("ea_chatbot.graph.nodes.query_analyzer.get_llm_model")
@patch("ea_chatbot.graph.nodes.query_analyzer.get_logger")
def test_query_analyzer_logs_actions(mock_get_logger, mock_get_llm, mock_state):
"""Test that query_analyzer_node logs its main actions."""
# Mock Logger
mock_logger = MagicMock()
mock_get_logger.return_value = mock_logger
# Mock LLM
mock_llm_instance = MagicMock()
mock_get_llm.return_value = mock_llm_instance
mock_structured_llm = MagicMock()
mock_llm_instance.with_structured_output.return_value = mock_structured_llm
expected_analysis = QueryAnalysis(
data_required=["2024 results", "Florida"],
unknowns=[],
ambiguities=[],
conditions=[],
next_action="plan"
)
mock_structured_llm.invoke.return_value = expected_analysis
query_analyzer_node(mock_state)
# Check that logger was called
# We expect at least one log at the start and one at the end
assert mock_logger.info.called
# Verify specific log messages if we decide on them
# For now, just ensuring it's called is enough for Red phase