fix : refine handling of leading '+' in response text

This commit is contained in:
Benedict Lee 2025-02-24 09:15:00 +09:00 committed by GitHub
parent 237a6ffb5f
commit feb306727e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -782,7 +782,8 @@ def try_fix_yaml(response_text: str,
# fifth fallback - try to remove leading '+' (sometimes added by AI for 'existing code' and 'improved code')
response_text_lines_copy = response_text_lines.copy()
for i in range(0, len(response_text_lines_copy)):
response_text_lines_copy[i] = ' ' + response_text_lines_copy[i][1:]
if response_text_lines_copy[i].startswith('+'):
response_text_lines_copy[i] = ' ' + response_text_lines_copy[i][1:]
try:
data = yaml.safe_load('\n'.join(response_text_lines_copy))
get_logger().info(f"Successfully parsed AI prediction after removing leading '+'")