Fix product data eval export format
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user