fixed typing in test

Author: cemde

maseval/interface/agents/llamaindex.py

@@ -306,7 +306,7 @@ def _install_hooks(self):
             from llama_index_instrumentation import get_dispatcher
 
             dispatcher = get_dispatcher()
-            dispatcher.add_span_handler(self._span_handler)
+            dispatcher.add_span_handler(self._span_handler)  # type: ignore[invalid-argument-type]  # duck-typed handler
         except (ImportError, Exception):
             pass
 

tests/test_interface/test_agent_integration/test_langgraph_integration.py

@@ -382,7 +382,7 @@ class State(TypedDict):
         messages: list
 
     graph = StateGraph(State)  # type: ignore[arg-type]
-    graph.add_node("agent", lambda s: s)
+    graph.add_node("agent", lambda s: s)  # type: ignore[invalid-argument-type]
     graph.set_entry_point("agent")
     graph.add_edge("agent", END)
     compiled = graph.compile()

tests/test_interface/test_agent_integration/test_smolagents_integration.py

@@ -611,7 +611,7 @@ def test_smolagents_llm_user_get_tool():
 def test_smolagents_message_conversion_tool_call_attributes():
     """Test _convert_smolagents_messages preserves tool_calls from ChatMessage objects."""
     from maseval.interface.agents.smolagents import SmolAgentAdapter
-    from smolagents.models import ChatMessage, MessageRole
+    from smolagents.models import ChatMessage, ChatMessageToolCall, ChatMessageToolCallFunction, MessageRole
     from unittest.mock import Mock
 
     mock_agent = Mock()
@@ -621,18 +621,23 @@ def test_smolagents_message_conversion_tool_call_attributes():
 
     adapter = SmolAgentAdapter(agent_instance=mock_agent, name="msg_agent")
 
-    # Create ChatMessage with tool_calls attribute
-    msg = ChatMessage(role=MessageRole.ASSISTANT, content="Using tool")
-    msg.tool_calls = [{"id": "call_1", "function": {"name": "search", "arguments": '{"q": "test"}'}}]
+    # Create ChatMessage with tool_calls attribute using proper types
+    tool_call = ChatMessageToolCall(
+        id="call_1",
+        type="function",
+        function=ChatMessageToolCallFunction(name="search", arguments='{"q": "test"}'),
+    )
+    msg = ChatMessage(role=MessageRole.ASSISTANT, content="Using tool", tool_calls=[tool_call])
 
     history = adapter._convert_smolagents_messages([msg])
 
     assert len(history) == 1
     assert history[0]["role"] == "assistant"
     assert history[0]["content"] == "Using tool"
     assert "tool_calls" in history[0]
-    assert history[0]["tool_calls"][0]["id"] == "call_1"
-    assert history[0]["tool_calls"][0]["function"]["name"] == "search"
+    assert len(history[0]["tool_calls"]) == 1
+    assert history[0]["tool_calls"][0].id == "call_1"
+    assert history[0]["tool_calls"][0].function.name == "search"
 
 
 # =============================================================================