File size: 977 Bytes
8578816 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
from . import Tool, LLMChain, llm
from langchain.memory import ConversationBufferMemory
from langchain_core.messages import SystemMessage
from langchain_core.prompts.chat import (
ChatPromptTemplate,
HumanMessagePromptTemplate,
MessagesPlaceholder,
)
system_message = "You are AI Assistant. You need to give crystal clear answer.\n"
template_messages = [
SystemMessage(content=system_message),
MessagesPlaceholder(variable_name="chat_history"),
HumanMessagePromptTemplate.from_template("{text}"),
]
prompt_template = ChatPromptTemplate.from_messages(template_messages)
memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
word_problem_chain = LLMChain(llm=llm, prompt=prompt_template, memory=memory)
word_problem_tool = Tool.from_function(name="Reasoning Tool", \
func=word_problem_chain.run, \
description="Useful for when you need to answer logic-based/reasoning \
questions.",
) |