cherry-pick 83d99e14
This commit is contained in:
parent
f5e653451a
commit
84ac688666
1 changed files with 6 additions and 0 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue