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: