nex/college-bot-v2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Activity Actions

FIx ollama endpoints not using ollama router
All checks were successful
Build and Publish college-bot-v2 / build_and_publish (push) Successful in 13s
Details

Browse source
This commit is contained in:
Nexus 2024-06-06 00:39:47 +01:00
parent 41e2daa265
commit 854abd9a6b
Signed by: nex
GPG key ID: 0FA334385D0B689F
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/server.py
View file

@ -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))): def get_ollama_threads(db: redis.Redis = Depends(get_db_factory(0))):
""" """
Retrieves all stored threads Retrieves all stored threads
@ -161,7 +161,7 @@ def get_ollama_threads(db: redis.Redis = Depends(get_db_factory(0))):
return keys 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))): def get_ollama_thread(thread_id: str, db: redis.Redis = Depends(get_db_factory(0))):
"""Retrieves a stored thread""" """Retrieves a stored thread"""
data: str = db.get(thread_id) 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) 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))): def delete_ollama_thread(thread_id: str, db: redis.Redis = Depends(get_db_factory(0))):
"""Deletes a stored thread""" """Deletes a stored thread"""
if not db.delete(thread_id): if not db.delete(thread_id):