diff --git a/examples/langchain-python-rag-websummary/main.py b/examples/langchain-python-rag-websummary/main.py index cd2ef47f..d1b05ba8 100644 --- a/examples/langchain-python-rag-websummary/main.py +++ b/examples/langchain-python-rag-websummary/main.py @@ -1,12 +1,12 @@ -from langchain.llms import Ollama -from langchain.document_loaders import WebBaseLoader +from langchain_community.llms import Ollama +from langchain_community.document_loaders import WebBaseLoader from langchain.chains.summarize import load_summarize_chain loader = WebBaseLoader("https://ollama.com/blog/run-llama2-uncensored-locally") docs = loader.load() -llm = Ollama(model="llama2") +llm = Ollama(model="llama3") chain = load_summarize_chain(llm, chain_type="stuff") -result = chain.run(docs) +result = chain.invoke(docs) print(result)