wuyang/zk-data-agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Allow router session tools to read external parquet paths

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
src/agent_tool_specs/data_agent.py
+2 -2
View File
@@ -254,7 +254,7 @@ def build_data_agent_tools(handlers: Mapping[str, ToolHandler]) -> list[AgentToo
'paths': {
'type': 'array',
'items': {'type': 'string'},
'description': 'Workspace-relative parquet files or date directories. Use when dates is not enough.',
'description': 'Parquet files or date directories. Relative paths resolve from workspace root; absolute paths are allowed for external online data.',
},
'max_files': {'type': 'integer', 'minimum': 1, 'maximum': 200},
'max_rows_per_file': {'type': 'integer', 'minimum': 1, 'maximum': 100000},
@@ -280,7 +280,7 @@ def build_data_agent_tools(handlers: Mapping[str, ToolHandler]) -> list[AgentToo
'paths': {
'type': 'array',
'items': {'type': 'string'},
'description': 'Workspace-relative parquet files or date directories. Use when dates is not enough.',
'description': 'Parquet files or date directories. Relative paths resolve from workspace root; absolute paths are allowed for external online data.',
},
'devices': {'type': 'array', 'items': {'type': 'string'}},
'domains': {'type': 'array', 'items': {'type': 'string'}},
src/data_agent_router_sessions.py
-4
View File
@@ -278,10 +278,6 @@ def _resolve_parquet_files(
files: list[Path] = []
for path in requested:
resolved = path.resolve()
try:
resolved.relative_to(root_path)
except ValueError as exc:
raise DataAgentRouterSessionError(f'path escapes workspace root: {path}') from exc
if not resolved.exists():
raise DataAgentRouterSessionError(f'path not found: {path}')
if resolved.is_dir():