|
from pr.config import MAX_AUTONOMOUS_ITERATIONS
|
|
from pr.ui import Colors
|
|
|
|
|
|
def is_task_complete(response, iteration):
|
|
if "error" in response:
|
|
return True
|
|
|
|
if "choices" not in response or not response["choices"]:
|
|
return True
|
|
|
|
message = response["choices"][0]["message"]
|
|
content = message.get("content", "").lower()
|
|
|
|
completion_keywords = [
|
|
"task complete",
|
|
"task is complete",
|
|
"finished",
|
|
"done",
|
|
"successfully completed",
|
|
"task accomplished",
|
|
"all done",
|
|
"implementation complete",
|
|
"setup complete",
|
|
"installation complete",
|
|
]
|
|
|
|
error_keywords = [
|
|
"cannot proceed",
|
|
"unable to continue",
|
|
"fatal error",
|
|
"cannot complete",
|
|
"impossible to",
|
|
]
|
|
|
|
has_tool_calls = "tool_calls" in message and message["tool_calls"]
|
|
mentions_completion = any(keyword in content for keyword in completion_keywords)
|
|
mentions_error = any(keyword in content for keyword in error_keywords)
|
|
|
|
if mentions_error:
|
|
return True
|
|
|
|
if mentions_completion and not has_tool_calls:
|
|
return True
|
|
|
|
if iteration > 5 and not has_tool_calls:
|
|
return True
|
|
|
|
if iteration >= MAX_AUTONOMOUS_ITERATIONS:
|
|
print(f"{Colors.YELLOW}⚠ Maximum iterations reached{Colors.RESET}")
|
|
return True
|
|
|
|
return False
|