Allow router session tools to read external parquet paths

This commit is contained in:
武阳
2026-05-09 11:55:54 +08:00
parent b87b9c14b7
commit 95c265d2a1
3 changed files with 25 additions and 6 deletions
+23
View File
@@ -123,6 +123,29 @@ class DataAgentRouterSessionTests(unittest.TestCase):
self.assertEqual(candidate['matched_turn']['query'], '导航去公司')
self.assertEqual(candidate['prev_turns'][0]['query'], '你好小爱')
@unittest.skipUnless(HAS_PYARROW, 'pyarrow is required for parquet tests')
def test_router_session_tools_allow_external_absolute_paths(self) -> None:
with tempfile.TemporaryDirectory() as workspace_dir, tempfile.TemporaryDirectory() as data_dir:
_write_router_parquet(Path(data_dir))
external_partition = Path(data_dir) / 'router_session_parquet' / 'date=20260428'
profile = profile_router_sessions(
workspace_dir,
paths=[str(external_partition)],
max_files=1,
)
search = search_router_sessions(
workspace_dir,
paths=[str(external_partition)],
query_keywords=['导航'],
max_files=1,
max_candidates=5,
)
self.assertEqual(profile['sampled_row_count'], 2)
self.assertEqual(search['candidate_count'], 1)
self.assertEqual(search['candidates'][0]['matched_turn']['query'], '导航去公司')
@unittest.skipUnless(HAS_PYARROW, 'pyarrow is required for parquet tests')
def test_router_session_tools_execute_against_registry(self) -> None:
with tempfile.TemporaryDirectory() as tmp_dir: