memechain/test/0_langchain_ollama_test.py

17 lines
358 B
Python

from langchain_core.prompts import ChatPromptTemplate
from langchain_ollama.llms import OllamaLLM
template = """Question: {question}
Answer: Let's think step by step."""
prompt = ChatPromptTemplate.from_template(template)
model = OllamaLLM(model="llama3")
chain = prompt | model
result = chain.invoke({"question": "What is LangChain?"})
print(result)