Fix loading exp_dir using row_index in xray and add error handling.

[amanjaiswal73892]

Use _row_index column for getting exp_dir from agent_df instead of iloc. This fixes errors when multiple agents are loaded in result_df.

---

This pull request improves the robustness of the update_exp_result method in agent_xray.py by adding error handling and warnings for cases where the provided row_index is missing or duplicated in the DataFrame.

Error handling and data validation:

• Added logic to handle cases where no rows match the given _row_index, raising a ValueError and setting self.exp_result to None.
• Added a warning and fallback behavior when multiple rows have the same _row_index, using the first match.

Description by Korbit AI

What change is being made?

Enhance the loading of exp_dir by utilizing row_index in xray and introduce error handling for cases of missing or duplicate indices.

Why are these changes being made?

The changes address an issue where the incorrect or multiple rows could be accessed due to row_index mismatches, potentially causing errors or undefined behavior. Error handling is added to improve robustness by raising an error when no matching row_index is found and issuing a warning if multiple rows are found, selecting the first entry by default.

Is this description stale? Ask me to generate a new description by commenting /korbit-generate-pr-description

[korbit-ai]

I've completed my review and didn't find any issues... but I did find this squirrel.

         _.-"""-,
       .'  ..::. `\
      /  .::' `'` /
     / .::' .--.=;
     | ::' /  C ..\
     | :: |   \  _.)
      \ ':|   /  \
       '-, \./ \)\)
          `-|   );/
            '--'-'

Files scanned

​

File Path	Reviewed
src/agentlab/analyze/agent_xray.py	✅

Explore our documentation to understand the languages and file types we support and the files we ignore.

Check out our docs on how you can make Korbit work best for you and your team.

Loving Korbit!? Share us on LinkedIn Reddit and X