mirror of
https://github.com/Aider-AI/aider.git
synced 2025-06-02 10:45:00 +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
|
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(
|
def send_completion(
|
||||||
model_name,
|
model_name,
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue