Skip to content

[KYUUBI #7379][2a/4] Data Agent Engine: tool system, data source, and prompt templates#7381

Closed
wangzhigang1999 wants to merge 2 commits intoapache:masterfrom
wangzhigang1999:pr2a/data-agent-tools
Closed

[KYUUBI #7379][2a/4] Data Agent Engine: tool system, data source, and prompt templates#7381
wangzhigang1999 wants to merge 2 commits intoapache:masterfrom
wangzhigang1999:pr2a/data-agent-tools

Conversation

@wangzhigang1999
Copy link
Copy Markdown
Contributor

Summary

Part 2a of 4 for the Data Agent Engine (umbrella). Stacked on #7380, please review that first.

Pure Java layer: tool system, data source abstraction, and prompt templates.

  • SqlQueryTool with maxRows enforcement (hard limit 1000), output truncation (64KB), cell truncation (500 chars)
  • ToolRegistry with JSON schema generation for LLM function calling
  • DataSourceFactory with HikariCP, JdbcDialect auto-detection (Spark/SQLite/MySQL/Trino)
  • SystemPromptBuilder — composable prompt with per-dialect templates

Note: This PR is stacked on #7380. The diff includes PR1 changes. After #7380 merges, this PR will be rebased and the diff will shrink to only the incremental changes.

Test plan

  • SqlQueryToolTest — maxRows, truncation, NULL rendering, pipe escaping, error formatting, timeout
  • ToolTest / ToolSchemaGeneratorTest / ToolRegistryThreadSafetyTest
  • DataSourceFactoryAuthTest / JdbcDialectTest
  • SystemPromptBuilderTest

@wangzhigang1999 wangzhigang1999 deleted the pr2a/data-agent-tools branch April 4, 2026 17:00
@wangzhigang1999 wangzhigang1999 restored the pr2a/data-agent-tools branch April 10, 2026 11:12
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant