Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 10 additions & 0 deletions agentplatform/_genai/_evals_common.py
Original file line number Diff line number Diff line change
Expand Up @@ -357,6 +357,16 @@ def _eval_cases_to_dataframe(
case.user_scenario.conversation_plan
)

if case.interactions_data_source:
row["interaction"] = getattr(
case.interactions_data_source, "interaction", None
)
gemini_cfg = getattr(
case.interactions_data_source, "gemini_agent_config", None
)
if gemini_cfg:
row["gemini_agent"] = getattr(gemini_cfg, "gemini_agent", None)

rows.append(row)
return pd.DataFrame(rows)

Expand Down
18 changes: 11 additions & 7 deletions agentplatform/_genai/_evals_visualization.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@
import pandas as pd
from pydantic import errors

from . import _evals_common
from . import types


Expand Down Expand Up @@ -1552,15 +1553,18 @@ def display_evaluation_dataset(eval_dataset_obj: types.EvaluationDataset) -> Non
else:
from IPython import display

if (
eval_dataset_obj.eval_dataset_df is None
or eval_dataset_obj.eval_dataset_df.empty
):
logger.warning("No inference data to display.")
return
df = eval_dataset_obj.eval_dataset_df

# Fall back to eval_cases when eval_dataset_df is not populated (e.g.
# when the dataset was constructed manually with eval_cases).
if df is None or df.empty:
if eval_dataset_obj.eval_cases:
df = _evals_common._eval_cases_to_dataframe(eval_dataset_obj.eval_cases)
if df is None or df.empty:
logger.warning("No inference data to display.")
return

processed_rows = []
df = eval_dataset_obj.eval_dataset_df

for _, row in df.iterrows():
processed_row = {}
Expand Down
Loading