Allow router session tools to read external parquet paths
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user