Fix TypeError in MLE-STAR get_text_from_response when handling None text values #299
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Problem
The MLE-STAR agent crashes with a
TypeError: can only concatenate str (not "NoneType") to str
when processing LLM responses that contain parts withNone
text values.Solution
Added a null check in
get_text_from_response()
to verify thatresponse.content.parts[i].text
is notNone
before concatenating it to the final text string.Changes
machine_learning_engineering/shared_libraries/common_util.py
and response.content.parts[i].text is not None
Impact
Fixes a critical bug that could cause the MLE-STAR system to crash unexpectedly.