mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-31 17:55:01 +00:00
feat: add message role validation function
This commit is contained in:
parent
cdc9ec2854
commit
bb61be630a
1 changed files with 13 additions and 1 deletions
|
@ -15,7 +15,19 @@ CACHE = None
|
|||
|
||||
RETRY_TIMEOUT = 60
|
||||
|
||||
# sanity_check_messages(messages) -> check if messages alternate role=user and role=assistant (it's ok if role=system messages are interspersed) ai!
|
||||
def sanity_check_messages(messages):
|
||||
"""Check if messages alternate between user and assistant roles.
|
||||
System messages can be interspersed anywhere.
|
||||
Returns True if valid, False otherwise."""
|
||||
last_role = None
|
||||
for msg in messages:
|
||||
role = msg.get("role")
|
||||
if role == "system":
|
||||
continue
|
||||
if last_role and role == last_role:
|
||||
return False
|
||||
last_role = role
|
||||
return True
|
||||
|
||||
def send_completion(
|
||||
model_name,
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue