First_agent_template / tools /final_answer.py
alexander1198's picture
Update tools/final_answer.py
7a74583 verified
raw
history blame contribute delete
424 Bytes
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