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