Spaces:
Runtime error
Runtime error
File size: 424 Bytes
7a74583 8fe992b 7a74583 8fe992b 7a74583 | 1 2 3 4 5 6 7 8 9 10 11 | 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
|