From 854abd9a6b9adbcaa3ec360717a1c4a68624b317 Mon Sep 17 00:00:00 2001 From: nexy7574 Date: Thu, 6 Jun 2024 00:39:47 +0100 Subject: [PATCH] FIx ollama endpoints not using ollama router --- src/server.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/server.py b/src/server.py index b6feed4..a1c9c44 100644 --- a/src/server.py +++ b/src/server.py @@ -150,7 +150,7 @@ ollama_router = APIRouter( ) -@app.get("/threads") +@ollama_router.get("/threads") def get_ollama_threads(db: redis.Redis = Depends(get_db_factory(0))): """ Retrieves all stored threads @@ -161,7 +161,7 @@ def get_ollama_threads(db: redis.Redis = Depends(get_db_factory(0))): return keys -@app.get("/thread/{thread_id}") +@ollama_router.get("/thread/{thread_id}") def get_ollama_thread(thread_id: str, db: redis.Redis = Depends(get_db_factory(0))): """Retrieves a stored thread""" data: str = db.get(thread_id) @@ -170,7 +170,7 @@ def get_ollama_thread(thread_id: str, db: redis.Redis = Depends(get_db_factory(0 return json.loads(data) -@app.delete("/thread/{thread_id}", status_code=204) +@ollama_router.delete("/thread/{thread_id}", status_code=204) def delete_ollama_thread(thread_id: str, db: redis.Redis = Depends(get_db_factory(0))): """Deletes a stored thread""" if not db.delete(thread_id):