SuperAPI AI 工具集

byted-bytehouse-ai-query

数据与分析

查看项目

更新于 10-15 17:30

技能介绍

用途:ByteHouse AI Query Skill 面向字节 ByteHouse/火山引擎云数仓的日常查询场景,把自然语言需求转为 SQL 并执行,同时支持多模态向量化与检索。

主要能力:Text2SQL 将自然语言查询转换为 ByteHouse SQL;列出数据库/表;执行 SQL 查询并返回结果,内置非 DQL 保护;支持文本、图片、视频向量化,以及向量/混合检索。

使用方式:配置保存在 ~/.bytehouse_config.json,需用户提供 BYTEHOUSE_HOST 与 BYTEHOUSE_PASSWORD,多模态场景还需 BH_ARK_API_KEY,其它字段建议保留默认值。先执行 source scripts/export_config.sh 注入环境变量;未指定库表时用 scripts/list_tables.py 查看;Text2SQL 用 scripts/text2sql.py,目标表以位置参数传入;执行用 scripts/execute_sql.py,默认展示前 5 条结果;多模态用 scripts/embedding.py 与 scripts/search_client.py。示例:python3 scripts/text2sql.py "get count of all call centers" "tpcds.call_center";python3 scripts/execute_sql.py "SELECT * FROM tpcds.call_center LIMIT 5"。

适用场景:自然语言查询 ByteHouse 数据、明确要求 Text2SQL、在 ByteHouse 上执行 SQL,以及文本/图片/视频的向量化存储与检索。

限制:非 DQL(INSERT/UPDATE/DELETE/DROP 等)默认被阻断,需向用户确认后附加 --force 强制执行;Agent 必须通过脚本发起查询,不允许自行拼装 SQL 或直接连接 ByteHouse;敏感字段需做 Mask,禁止臆测数据。前置条件为 Python 3.8+、jq、requirements.txt 依赖及有效连接信息。