diff --git a/lightrag/api/routers/ollama_api.py b/lightrag/api/routers/ollama_api.py index 426583f8..f9353dda 100644 --- a/lightrag/api/routers/ollama_api.py +++ b/lightrag/api/routers/ollama_api.py @@ -483,6 +483,12 @@ class OllamaAPI: if not messages: raise HTTPException(status_code=400, detail="No messages provided") + # Validate that the last message is from a user + if messages[-1].role != "user": + raise HTTPException( + status_code=400, detail="Last message must be from user role" + ) + # Get the last message as query and previous messages as history query = messages[-1].content # Convert OllamaMessage objects to dictionaries