Merge pull request #280 from huangye123/worktree-prompt-filter-rules-edit-support

feat: 支持 Prompt 检查额外规则编辑

Author: james-6-23

admin/handler.go

@@ -356,6 +356,7 @@ func (h *Handler) RegisterRoutes(r *gin.Engine) {
 	api.GET("/prompt-filter/logs/match", h.MatchPromptFilterLog)
 	api.DELETE("/prompt-filter/logs", h.ClearPromptFilterLogs)
 	api.POST("/prompt-filter/test", h.TestPromptFilter)
+	api.POST("/prompt-filter/rules/test", h.TestPromptFilterRulePattern)
 	api.GET("/prompt-filter/rules", h.GetPromptFilterRules)
 	api.GET("/models", h.ListModels)
 	api.POST("/models/sync", h.SyncModels)

admin/prompt_filter.go

@@ -3,6 +3,7 @@ package admin
 import (
 	"context"
 	"net/http"
+	"regexp"
 	"strconv"
 	"strings"
 	"time"
@@ -29,6 +30,16 @@ type promptFilterTestResponse struct {
 	Verdict promptfilter.Verdict `json:"verdict"`
 }
 
+type promptFilterRulePatternTestRequest struct {
+	Pattern string `json:"pattern"`
+	Text    string `json:"text"`
+}
+
+type promptFilterRulePatternTestResponse struct {
+	Matched bool   `json:"matched"`
+	Error   string `json:"error,omitempty"`
+}
+
 type promptFilterRuleItem struct {
 	Name     string `json:"name"`
 	Pattern  string `json:"pattern"`
@@ -205,6 +216,37 @@ func (h *Handler) TestPromptFilter(c *gin.Context) {
 	c.JSON(http.StatusOK, promptFilterTestResponse{Verdict: verdict})
 }
 
+func (h *Handler) TestPromptFilterRulePattern(c *gin.Context) {
+	var req promptFilterRulePatternTestRequest
+	if err := c.ShouldBindJSON(&req); err != nil {
+		writeError(c, http.StatusBadRequest, "请求体无效")
+		return
+	}
+	trimmedPattern := strings.TrimSpace(req.Pattern)
+	if trimmedPattern == "" {
+		writeError(c, http.StatusBadRequest, "pattern 不能为空")
+		return
+	}
+	if req.Text == "" {
+		writeError(c, http.StatusBadRequest, "text 不能为空")
+		return
+	}
+	if len([]rune(req.Pattern)) > 5000 {
+		writeError(c, http.StatusBadRequest, "pattern 不能超过 5000 个字符")
+		return
+	}
+	if len([]rune(req.Text)) > 20000 {
+		writeError(c, http.StatusBadRequest, "text 不能超过 20000 个字符")
+		return
+	}
+	re, err := regexp.Compile(req.Pattern)
+	if err != nil {
+		c.JSON(http.StatusOK, promptFilterRulePatternTestResponse{Matched: false, Error: err.Error()})
+		return
+	}
+	c.JSON(http.StatusOK, promptFilterRulePatternTestResponse{Matched: re.MatchString(req.Text)})
+}
+
 func (h *Handler) GetPromptFilterRules(c *gin.Context) {
 	cfg := h.store.GetPromptFilterConfig()
 	disabled := map[string]bool{}

frontend/src/api.ts

@@ -32,6 +32,7 @@ import type {
   OpsOverviewResponse,
   PromptFilterLog,
   PromptFilterLogsResponse,
+  PromptFilterRulePatternTestResponse,
   PromptFilterRulesResponse,
   PromptFilterTestResponse,
   PublicAPIKeyUsageResponse,
@@ -511,6 +512,8 @@ export const api = {
   },
   testPromptFilter: (data: { text: string; endpoint?: string; model?: string }) =>
     request<PromptFilterTestResponse>('/prompt-filter/test', { method: 'POST', body: JSON.stringify(data) }),
+  testPromptFilterRulePattern: (data: { pattern: string; text: string }) =>
+    request<PromptFilterRulePatternTestResponse>('/prompt-filter/rules/test', { method: 'POST', body: JSON.stringify(data) }),
   getPromptFilterRules: () =>
     request<PromptFilterRulesResponse>('/prompt-filter/rules'),
   getModels: () => request<ModelsResponse>('/models'),

frontend/src/components/ui/select.tsx

@@ -17,6 +17,7 @@ interface SelectProps {
   disabled?: boolean
   className?: string
   compact?: boolean
+  triggerClassName?: string
 }
 
 interface DropdownPosition {
@@ -39,6 +40,7 @@ export function Select({
   disabled = false,
   className,
   compact = false,
+  triggerClassName,
 }: SelectProps) {
   const [open, setOpen] = useState(false)
   const [position, setPosition] = useState<DropdownPosition | null>(null)
@@ -128,7 +130,8 @@ export function Select({
           'hover:border-primary/30 hover:bg-accent/50',
           'focus-visible:border-ring focus-visible:ring-[3px] focus-visible:ring-ring/20',
           'disabled:cursor-not-allowed disabled:opacity-60',
-          open && 'border-primary/35 ring-[3px] ring-primary/10'
+          open && 'border-primary/35 ring-[3px] ring-primary/10',
+          triggerClassName
         )}
         onClick={() => {
           if (!disabled) {

frontend/src/locales/en.json

@@ -1462,6 +1462,19 @@
     "customRulesTitle": "Extra Rules",
     "customRulesDesc": "Extra rules are stored in system settings and can be toggled or removed independently.",
     "addCustomRule": "Add Rule",
+    "addCustomRuleDesc": "Fill in the extra rule name, regex, weight, category, and strict flag. Saving adds it to system settings immediately.",
+    "editCustomRule": "Edit Rule",
+    "editCustomRuleDesc": "Update the extra rule name, regex, weight, category, and strict flag. Saving applies the change to system settings immediately.",
+    "rulePatternTesterTitle": "Regex Hit Test",
+    "rulePatternTesterDesc": "Enter a prompt and test whether the current rule matches it using backend Go regex.",
+    "rulePatternTest": "Test Hit",
+    "rulePatternTesting": "Testing...",
+    "ruleTestTextPlaceholder": "Enter a prompt to test against this regex",
+    "ruleTestMatched": "Matched: the current regex matches this prompt.",
+    "ruleTestNotMatched": "Not matched: the current regex does not match this prompt.",
+    "rulePatternInvalid": "Invalid regex",
+    "rulePatternRequired": "Enter a regex first",
+    "ruleTestTextRequired": "Enter test prompt text",
     "noCustomRules": "No extra rules yet",
     "filterByCategory": "Filter by Category",
     "selectAll": "Select All",

frontend/src/locales/zh.json

@@ -1462,6 +1462,19 @@
     "customRulesTitle": "额外规则",
     "customRulesDesc": "添加的额外规则会保存到系统设置中，可独立启停或删除。",
     "addCustomRule": "添加规则",
+    "addCustomRuleDesc": "填写额外规则的名称、正则、权重、分类和强规则标记。保存后会立即加入系统设置。",
+    "editCustomRule": "编辑规则",
+    "editCustomRuleDesc": "修改额外规则的名称、正则、权重、分类和强规则标记。保存后会立即更新系统设置。",
+    "rulePatternTesterTitle": "正则命中测试",
+    "rulePatternTesterDesc": "输入一段提示词，使用后端 Go 正则验证当前规则是否会命中。",
+    "rulePatternTest": "测试命中",
+    "rulePatternTesting": "测试中...",
+    "ruleTestTextPlaceholder": "输入用于测试命中的提示词",
+    "ruleTestMatched": "已命中：当前正则会匹配这段提示词。",
+    "ruleTestNotMatched": "未命中：当前正则不会匹配这段提示词。",
+    "rulePatternInvalid": "正则无效",
+    "rulePatternRequired": "请先填写正则",
+    "ruleTestTextRequired": "请输入测试提示词",
     "noCustomRules": "暂无额外规则",
     "filterByCategory": "按分类筛选",
     "selectAll": "全选",