Add info log for successful AI prediction parse in utils.py

This commit is contained in:
mrT23
2024-08-14 08:14:51 +03:00
parent f1981092d3
commit f4b06640d2

View File

@ -565,6 +565,9 @@ def load_yaml(response_text: str, keys_fix_yaml: List[str] = [], first_key="", l
data = try_fix_yaml(response_text, keys_fix_yaml=keys_fix_yaml, first_key=first_key, last_key=last_key)
if not data:
get_logger().error(f"Failed to parse AI prediction after fallbacks", artifact={'response_text': response_text})
else:
get_logger().info(f"Successfully parsed AI prediction after fallbacks",
artifact={'response_text': response_text})
return data