from langchain.schema import HumanMessage, SystemMessage
from langchain.chat_models import ChatOpenAI

class FinalAnswerTool:
def init(self, model_name: str = "Qwen2.5-Coder"):
self.client = ChatOpenAI(model_name=model_name)

def __call__(self, conversation: list) -> str:
    messages = [SystemMessage(content=conversation[0])] + [HumanMessage(content=m) for m in conversation[1:]]
    return self.client(messages).content