Fix product data eval export format

This commit is contained in:
wuyang6
2026-05-11 15:27:11 +08:00
parent be731caed7
commit f22de8a40b
15 changed files with 414 additions and 71 deletions
+4
View File
@@ -57,6 +57,7 @@ class DataAgentRouterSessionTests(unittest.TestCase):
[candidate],
dataset_label='总结类边界评测集',
default_target='Summarize',
default_complex=True,
batch_id='summary',
)
@@ -68,6 +69,7 @@ class DataAgentRouterSessionTests(unittest.TestCase):
self.assertEqual(record['prev_session'], [{'query': '这篇文章讲了什么', 'tts': '', 'timestamp': 10}])
self.assertEqual(record['context']['domain'], 'QA')
self.assertEqual(record['label']['target'], 'Summarize')
self.assertEqual(record['dimensions'], {'complex': True})
def test_convert_router_candidates_to_records_applies_review_decisions(self) -> None:
candidates = [
@@ -84,6 +86,7 @@ class DataAgentRouterSessionTests(unittest.TestCase):
'matched_turn_index': 0,
'decision': 'include',
'target': 'Summarize',
'complex': True,
'notes': '前文有可总结内容',
},
{
@@ -98,6 +101,7 @@ class DataAgentRouterSessionTests(unittest.TestCase):
self.assertEqual(result['record_count'], 1)
self.assertEqual(result['skipped_count'], 1)
self.assertEqual(result['records'][0]['meta']['notes'], '前文有可总结内容')
self.assertEqual(result['records'][0]['dimensions']['complex'], True)
@unittest.skipUnless(HAS_PYARROW, 'pyarrow is required for parquet tests')
def test_profile_and_search_router_sessions_read_parquet(self) -> None: