FCM Latest State: Fixed bug that did not actually store message history

esbmc_ai/latest_state_solution_generator.py

@@ -1,10 +1,12 @@
 # Author: Yiannis Charalambous
 
 from typing_extensions import override
-from langchain import BaseMessage
+from langchain_core.messages import BaseMessage
 from esbmc_ai.solution_generator import SolutionGenerator
 from esbmc_ai.chat_response import FinishReason
 
+# TODO Test me
+
 
 class LatestStateSolutionGenerator(SolutionGenerator):
     """SolutionGenerator that only shows the latest source code and verifier
@@ -18,6 +20,8 @@ def generate_solution(self) -> tuple[str, FinishReason]:
         messages: list[BaseMessage] = self.messages
         self.messages: list[BaseMessage] = []
         solution, finish_reason = super().generate_solution()
+        # Append last messages to the messages stack
+        messages.extend(self.messages)
         # Restore
         self.messages = messages
         return solution, finish_reason