Spaces:
Runtime error
Runtime error
| 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 | |