From ee2e85ca24330be0a00f9310b301469ab5b77934 Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Wed, 29 May 2024 00:42:19 -0700
Subject: [PATCH] fix

---
 main.py                                        | 4 ++--
 pipelines/examples/ollama_manifold_pipeline.py | 7 ++++++-
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/main.py b/main.py
index f0c68de..d94a746 100644
--- a/main.py
+++ b/main.py
@@ -600,8 +600,8 @@ async def delete_pipeline(
         )
 
 
-@app.post("/v1/reload")
-@app.post("/reload")
+@app.post("/v1/pipelines/reload")
+@app.post("/pipelines/reload")
 async def reload_pipelines(user: str = Depends(get_current_user)):
     if user == API_KEY:
         await reload()
diff --git a/pipelines/examples/ollama_manifold_pipeline.py b/pipelines/examples/ollama_manifold_pipeline.py
index 199947f..c4a1ae9 100644
--- a/pipelines/examples/ollama_manifold_pipeline.py
+++ b/pipelines/examples/ollama_manifold_pipeline.py
@@ -30,7 +30,6 @@ class Pipeline:
     async def on_startup(self):
         # This function is called when the server is started.
         print(f"on_startup:{__name__}")
-        self.pipelines = self.get_ollama_models()
         pass
 
     async def on_shutdown(self):
@@ -38,6 +37,12 @@ class Pipeline:
         print(f"on_shutdown:{__name__}")
         pass
 
+    async def on_valves_update(self):
+        # This function is called when the valves are updated.
+        print(f"on_valves_update:{__name__}")
+        self.pipelines = self.get_ollama_models()
+        pass
+
     def get_ollama_models(self):
         if self.valves.OLLAMA_BASE_URL:
             try: