panda
bd5bfbac2d
Root cause: LLM receiving full 34k-char JRXML would regenerate from scratch
instead of modifying coordinates in-place, shrinking output to ~3k chars.
Solution (programmatic node control, not prompt engineering):
- New agent/jrxml_windower.py: decompose JRXML into header (never sent to
LLM) + individual bands. Split bands >4000 chars at element boundaries.
Reassemble with element count validation (>10% change = rollback).
- Rewrite refine_layout: per-band windowed LLM processing (~2-4k chars
each). LLM cannot "reimagine" the entire report.
- Rewrite map_fields: 100% programmatic regex $F{field_N} -> real name
replacement. Zero LLM calls, zero content loss.
- _sanitize_field_name: non-ASCII chars escaped to _uXXXX_ format for
valid JRXML identifiers.
- Tests: 48 new unit tests (windower 28 + map_fields 20). All passing.
Full suite 385 tests, zero regressions.
116 lines
3.9 KiB
Python
116 lines
3.9 KiB
Python
"""数据源模式解析模块。
|
||
|
||
默认使用 $P{xxx} 参数模式；用户可选择 JDBC 直连模式。
|
||
"""
|
||
|
||
import json
|
||
import os
|
||
import re
|
||
from typing import Optional
|
||
|
||
from dotenv import load_dotenv
|
||
|
||
from agent.state import AgentState
|
||
|
||
load_dotenv()
|
||
|
||
|
||
def resolve_datasource_mode(state: AgentState) -> str:
|
||
"""返回数据源模式: "parameter" 或 "jdbc"。
|
||
|
||
优先读取 state 中已设定的模式，否则根据用户输入检测。
|
||
"""
|
||
existing = state.get("datasource_mode", "")
|
||
if existing in ("parameter", "jdbc"):
|
||
return existing
|
||
|
||
user_input = state.get("user_input", "")
|
||
if _detect_jdbc_intent(user_input):
|
||
return "jdbc"
|
||
return "parameter"
|
||
|
||
|
||
def _detect_jdbc_intent(user_input: str) -> bool:
|
||
"""检测用户是否想要 JDBC 直连数据库模式。"""
|
||
patterns = [
|
||
r"(直连|直连数据库|数据库直连)",
|
||
r"(从|在)(数据库|DB|MySQL|PostgreSQL|Oracle|SQL Server)\w*",
|
||
r"(jdbc|JDBC)",
|
||
r"(连接|连)(数据库|DB)",
|
||
r"(查询|select|SELECT)\s",
|
||
]
|
||
for pat in patterns:
|
||
if re.search(pat, user_input):
|
||
return True
|
||
return False
|
||
|
||
|
||
def _sanitize_url(url: str) -> str:
|
||
"""剥离 JDBC URL 中的 user:password@ 片段，防止泄露到 LLM prompt。"""
|
||
return re.sub(r"://[^@]*@", "://***:***@", url)
|
||
|
||
|
||
def build_datasource_context(mode: str, kb_fields: list, db_config: Optional[dict] = None) -> str:
|
||
"""构建数据源上下文字符串，注入生成 prompt。"""
|
||
if mode == "jdbc":
|
||
if not db_config or not db_config.get("url"):
|
||
return (
|
||
"[数据源模式: JDBC]\n"
|
||
"⚠ 用户想要 JDBC 直连模式，但尚未配置数据库连接信息。\n"
|
||
"请先生成带 $P{xxx} 参数占位符的 JRXML，并提醒用户配置 JDBC 连接。"
|
||
)
|
||
safe_url = _sanitize_url(db_config.get("url", ""))
|
||
return (
|
||
"[数据源模式: JDBC]\n"
|
||
f"连接URL: {safe_url}\n"
|
||
f"驱动: {db_config.get('driver', '')}\n"
|
||
"请使用 <queryString><![CDATA[...]]></queryString> 中的 SQL 查询。"
|
||
)
|
||
|
||
# parameter mode
|
||
if kb_fields:
|
||
field_list = "\n".join(
|
||
f"| {f['name']} | {f.get('description', '')} | {f.get('type', 'java.lang.String')} |"
|
||
for f in kb_fields
|
||
)
|
||
return (
|
||
"[数据源模式: 参数]\n"
|
||
"使用 $P{xxx} 参数模式，以下为可用参数：\n"
|
||
f"| 参数名 | 含义 | 类型 |\n|---|---|---|\n{field_list}"
|
||
)
|
||
return "[数据源模式: 参数]\n使用 $P{xxx} 参数模式生成 JRXML。"
|
||
|
||
|
||
def configure_jdbc(state: AgentState, url: str = "", driver: str = "",
|
||
username: str = "", password: str = "") -> dict:
|
||
"""配置 JDBC 连接并返回更新字段。
|
||
|
||
注意：db_config 会被存入 AgentState 并持久化到会话文件。
|
||
生产环境应使用外部密钥管理服务，避免明文存储密码。
|
||
"""
|
||
return {
|
||
"datasource_mode": "jdbc",
|
||
"db_config": {
|
||
"url": url,
|
||
"driver": driver or "com.mysql.cj.jdbc.Driver",
|
||
"username": username,
|
||
"password": password,
|
||
},
|
||
}
|
||
|
||
|
||
def ask_db_config(state: AgentState) -> Optional[str]:
|
||
"""如果用户选了 JDBC 模式但未配置 DB 连接，返回反问消息。"""
|
||
mode = resolve_datasource_mode(state)
|
||
if mode == "jdbc":
|
||
db_config = state.get("db_config", {})
|
||
if not db_config or not db_config.get("url"):
|
||
return (
|
||
"您选择了数据库直连模式，请提供以下信息：\n"
|
||
"1. JDBC URL（如 jdbc:mysql://localhost:3306/dbname）\n"
|
||
"2. 数据库用户名\n"
|
||
"3. 数据库密码\n"
|
||
"4. 驱动类名（可选，默认 com.mysql.cj.jdbc.Driver）"
|
||
)
|
||
return None
|