This commit is contained in:
Raphaël MANSUY 2025-12-04 19:14:29 +08:00
parent f5e653451a
commit 84ac688666

View file

@ -505,6 +505,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