From 0494004f5c07bd8243bc52d15e549fc18e0047b4 Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Tue, 21 May 2024 16:07:08 -0700
Subject: [PATCH] fix

---
 examples/llamaindex_ollama_pipeline.py | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

diff --git a/examples/llamaindex_ollama_pipeline.py b/examples/llamaindex_ollama_pipeline.py
index 768ef0e..7bf7fda 100644
--- a/examples/llamaindex_ollama_pipeline.py
+++ b/examples/llamaindex_ollama_pipeline.py
@@ -2,18 +2,15 @@ from typing import List, Union, Generator
 from schemas import OpenAIChatMessage
 
 from llama_index.embeddings.ollama import OllamaEmbedding
-from llama_index.llms.openai import OpenAI
+from llama_index.llms.ollama import Ollama
 from llama_index.core import Settings
 
-ollama_embedding = OllamaEmbedding(
+
+Settings.embed_model = OllamaEmbedding(
     model_name="nomic-embed-text",
     base_url="http://localhost:11434",
 )
-
-Settings.embed_model = ollama_embedding
-Settings.llm = OpenAI(
-    temperature=0, model="llama3", api_key="none", api_base="http://localhost:11434"
-)
+Settings.llm = Ollama(model="llama3")
 
 from llama_index.core import VectorStoreIndex, SimpleDirectoryReader