Update tests to handle new invocation logic

Author: Alex Lambrou

tests/agent/test_SemanticTerminalAgent.py

@@ -119,4 +119,4 @@ def test_invoke(self, mrkl_agent):
 
         mrkl_agent.terminal_agent_executor.invoke = MagicMock()
         mrkl_agent.invoke(user_input)
-        mrkl_agent.terminal_agent_executor.invoke.assert_called_with(input=user_input)
+        mrkl_agent.terminal_agent_executor.invoke.assert_called_with({"input": user_input, "chat_history": []})

tests/agent/test_TerminalAgent.py

@@ -7,6 +7,8 @@
     AIMessage,
     SystemMessage,
 )
+from langchain_core.messages import HumanMessage
+
 from semterm.agent.TerminalAgent import TerminalAgent
 from semterm.agent.TerminalAgentPrompt import PREFIX, SUFFIX
 
@@ -30,7 +32,9 @@ def test_create_prompt(self, terminal_agent, mock_tools):
         # Extract properties from the returned ChatPromptTemplate
         system_message_from_prompt = prompt.messages[0].format_messages()[0].content
         human_message_from_prompt = (
-            prompt.messages[2].format_messages(input="test input")[0].content
+            prompt.messages[2].format_messages(input="test input", chat_history=[HumanMessage(
+                content="test input"
+            )])[0].content
         )
 
         # Assert that the properties have the expected values