From 649349a8405a039ff201897dbc05a491fc2518c5 Mon Sep 17 00:00:00 2001
From: broman-pl <2329842+broman-pl@users.noreply.github.com>
Date: Wed, 21 Jan 2026 13:04:32 -0800
Subject: [PATCH] Update local ollama sample to new version of langchain Change
 file name to remove collision with module name

---
 local-llm/{ollama.py => ollama-chat.py} | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)
 rename local-llm/{ollama.py => ollama-chat.py} (73%)

diff --git a/local-llm/ollama.py b/local-llm/ollama-chat.py
similarity index 73%
rename from local-llm/ollama.py
rename to local-llm/ollama-chat.py
index 8757fdc0a..600b8e3db 100644
--- a/local-llm/ollama.py
+++ b/local-llm/ollama-chat.py
@@ -1,13 +1,13 @@
-from langchain.llms.ollama import Ollama
-from langchain.prompts import ChatPromptTemplate
-from langchain.schema import StrOutputParser
-from langchain.schema.runnable import Runnable
-from langchain.schema.runnable.config import RunnableConfig
+from langchain_ollama.llms import OllamaLLM
+from langchain_core.prompts import ChatPromptTemplate
+from langchain_core.output_parsers import StrOutputParser
+from langchain_core.runnables.base import Runnable
+from langchain_core.runnables.config import RunnableConfig
 
 import chainlit as cl
 
-model = Ollama(
-    model="llama2",
+model = OllamaLLM(
+    model="gemma3",
 )