main.py

#{
# Using VirtualchatbotTest memory
#(venv) python3 -m venv chatbotTest
#(venv) source chatbotTest/bin/activate
#(chatbotTest) (venv)
#}

from langchain_ollama import OllamaLLM
from langchain_core.prompts import ChatPromptTemplate

template="""
Answer the question below, use less than 200 words.

Here is the conversation history: {history}

Question: {question}

Answer:
"""
model = OllamaLLM(model="ollamaPirate")
prompt = ChatPromptTemplate.from_template(template)
chain = prompt | model

def handle_conv():
    history=""
    print("Welcome to pirateChat, type exit to quit.")
    while True:
        userInput = input("Input: ")
        if userInput.lower() =="exit":
            break
        result = chain.invoke({
            "history": history,
            "question": userInput
        })
        print("Pirate: ", result)
        history = history + f"\nUser: {userInput}\nAI: {result}"

if __name__ == "__main__":
    handle_conv()