fix(plugin): preserve image routing metadata

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Author: DouDOU-start

backend/internal/plugin/forwarder.go

@@ -3,6 +3,7 @@ package plugin
 import (
 	"log/slog"
 	"net/http"
+	"strings"
 	"time"
 
 	"github.com/gin-gonic/gin"
@@ -78,14 +79,9 @@ func (f *Forwarder) Forward(c *gin.Context) {
 	}
 	defer releaseClientQuota()
 
-	routes, err := routing.ListEligibleGroups(c.Request.Context(), f.db, state.keyInfo.UserID, state.requestedPlatform, state.keyInfo.UserGroupRates, routing.Requirements{
+	routes := routesForAPIKey(state, routing.Requirements{
 		NeedsImage: requestNeedsImage(state.requestPath, state.model),
 	})
-	if err != nil {
-		slog.Error("查询候选分组失败", "platform", state.requestedPlatform, "model", state.model, "error", err)
-		openAIError(c, http.StatusServiceUnavailable, "server_error", "routing_unavailable", "请求暂时无法完成，请稍后重试")
-		return
-	}
 	if len(routes) == 0 {
 		slog.Warn("没有可用候选分组", "platform", state.requestedPlatform, "model", state.model, "user_id", state.keyInfo.UserID)
 		openAIError(c, http.StatusServiceUnavailable, "server_error", "no_available_route", "请求暂时无法完成，请稍后重试")
@@ -198,6 +194,69 @@ func (f *Forwarder) Forward(c *gin.Context) {
 	openAIError(c, 503, "server_error", "all_routes_failed", "请求暂时无法完成，请稍后重试")
 }
 
+func routesForAPIKey(state *forwardState, requirements routing.Requirements) []routing.Candidate {
+	if state == nil || state.keyInfo == nil {
+		return nil
+	}
+	if !apiKeyGroupMatchesRequirements(state.keyInfo, requirements) {
+		return nil
+	}
+	return []routing.Candidate{keyInfoRoute(state.keyInfo)}
+}
+
+func apiKeyGroupMatchesRequirements(keyInfo *auth.APIKeyInfo, requirements routing.Requirements) bool {
+	if keyInfo == nil {
+		return false
+	}
+	if requirements.NeedsImage && strings.EqualFold(keyInfo.GroupPlatform, "openai") {
+		return pluginSettingEnabledForKey(keyInfo.GroupPluginSettings, "openai", "image_enabled")
+	}
+	return true
+}
+
+func pluginSettingEnabledForKey(settings map[string]map[string]string, plugin, key string) bool {
+	for pluginName, kv := range settings {
+		if !strings.EqualFold(pluginName, plugin) {
+			continue
+		}
+		for k, v := range kv {
+			if strings.EqualFold(k, key) {
+				return strings.EqualFold(strings.TrimSpace(v), "true")
+			}
+		}
+	}
+	return false
+}
+
+func keyInfoRoute(keyInfo *auth.APIKeyInfo) routing.Candidate {
+	return routing.Candidate{
+		GroupID:                keyInfo.GroupID,
+		Platform:               keyInfo.GroupPlatform,
+		EffectiveRate:          billing.ResolveBillingRateForGroup(keyInfo.UserGroupRates, keyInfo.GroupID, keyInfo.GroupRateMultiplier),
+		GroupRateMultiplier:    keyInfo.GroupRateMultiplier,
+		GroupServiceTier:       keyInfo.GroupServiceTier,
+		GroupForceInstructions: keyInfo.GroupForceInstructions,
+		GroupPluginSettings:    clonePluginSettingsForKey(keyInfo.GroupPluginSettings),
+	}
+}
+
+func clonePluginSettingsForKey(in map[string]map[string]string) map[string]map[string]string {
+	if len(in) == 0 {
+		return nil
+	}
+	out := make(map[string]map[string]string, len(in))
+	for plugin, settings := range in {
+		if len(settings) == 0 {
+			continue
+		}
+		out[plugin] = make(map[string]string, len(settings))
+		for k, v := range settings {
+			out[plugin][k] = v
+		}
+	}
+	return out
+}
+
 func keyInfoForRoute(base *auth.APIKeyInfo, route routing.Candidate) *auth.APIKeyInfo {
 	info := *base
 	info.GroupID = route.GroupID

backend/internal/plugin/forwarder_test.go

@@ -2,6 +2,9 @@ package plugin
 
 import (
 	"testing"
+
+	"github.com/DouDOU-start/airgate-core/internal/auth"
+	"github.com/DouDOU-start/airgate-core/internal/routing"
 )
 
 func TestParseBody(t *testing.T) {
@@ -37,3 +40,53 @@ func TestParseBody_StreamTrue(t *testing.T) {
 		t.Fatalf("SessionID = %q, want %q", parsed.SessionID, "sess-1")
 	}
 }
+
+func TestRoutesForAPIKeyUsesBoundGroupOnly(t *testing.T) {
+	t.Parallel()
+
+	settings := map[string]map[string]string{"openai": {"image_enabled": "true"}}
+	state := &forwardState{keyInfo: &auth.APIKeyInfo{
+		GroupID:                42,
+		GroupPlatform:          "openai",
+		GroupRateMultiplier:    1.5,
+		UserGroupRates:         map[int64]float64{42: 0.7, 99: 0.1},
+		GroupPluginSettings:    settings,
+		GroupServiceTier:       "priority",
+		GroupForceInstructions: "stay concise",
+	}}
+
+	routes := routesForAPIKey(state, routing.Requirements{NeedsImage: true})
+	if len(routes) != 1 {
+		t.Fatalf("len(routes) = %d, want 1", len(routes))
+	}
+	route := routes[0]
+	if route.GroupID != 42 {
+		t.Fatalf("GroupID = %d, want 42", route.GroupID)
+	}
+	if route.EffectiveRate != 0.7 {
+		t.Fatalf("EffectiveRate = %v, want 0.7", route.EffectiveRate)
+	}
+	if route.GroupPluginSettings["openai"]["image_enabled"] != "true" {
+		t.Fatalf("image_enabled not preserved")
+	}
+
+	settings["openai"]["image_enabled"] = "false"
+	if route.GroupPluginSettings["openai"]["image_enabled"] != "true" {
+		t.Fatalf("route plugin settings should be cloned")
+	}
+}
+
+func TestRoutesForAPIKeyRejectsImageWhenBoundGroupDisabled(t *testing.T) {
+	t.Parallel()
+
+	state := &forwardState{keyInfo: &auth.APIKeyInfo{
+		GroupID:             42,
+		GroupPlatform:       "openai",
+		GroupPluginSettings: map[string]map[string]string{"openai": {"image_enabled": "false"}},
+	}}
+
+	routes := routesForAPIKey(state, routing.Requirements{NeedsImage: true})
+	if len(routes) != 0 {
+		t.Fatalf("len(routes) = %d, want 0", len(routes))
+	}
+}

web/src/i18n/en.json

@@ -985,6 +985,11 @@
     "no_response": "No response from AI service",
     "send": "Send",
     "stop": "Stop",
+    "edit": "Edit",
+    "image": "Image",
+    "attach_images": "Attach images",
+    "select_model_first": "Select a model first",
+    "select_image_model_first": "Select an image model first",
     "input_placeholder": "Type a message...",
     "input_hint": "Enter to send, Shift+Enter for newline",
     "platform": "Platform",
@@ -1004,6 +1009,18 @@
     "retry": "Retry",
     "regenerate": "Regenerate",
     "retry_image": "Retry image",
-    "no_image_prompt": "No prompt found for this image"
+    "no_image_prompt": "No prompt found for this image",
+    "edit_image_region": "Edit image region",
+    "edit_image_region_hint": "Drag on the image to box-select the area to replace.",
+    "choose_source_image_region_hint": "Choose a source image and select a region.",
+    "replace_source": "Replace source",
+    "choose_source": "Choose source",
+    "region_selected": "Region selected",
+    "drag_to_select": "Drag to select",
+    "clear_selection": "Clear selection",
+    "choose_source_image_for_regional_editing": "Choose a source image for regional editing",
+    "choose_source_image_first": "Choose a source image first",
+    "select_edit_area_first": "Select the area to edit first",
+    "describe_image_change_first": "Describe the image change first"
   }
 }

web/src/i18n/zh.json

@@ -989,6 +989,11 @@
     "no_response": "AI 服务未返回结果",
     "send": "发送",
     "stop": "停止",
+    "edit": "编辑",
+    "image": "图片",
+    "attach_images": "添加图片",
+    "select_model_first": "请先选择模型",
+    "select_image_model_first": "请先选择图片模型",
     "input_placeholder": "输入消息...",
     "input_hint": "Enter 发送，Shift+Enter 换行",
     "platform": "平台",
@@ -1008,6 +1013,18 @@
     "retry": "重试",
     "regenerate": "重新生成",
     "retry_image": "重试图片生成",
-    "no_image_prompt": "未找到这张图片对应的提示词"
+    "no_image_prompt": "未找到这张图片对应的提示词",
+    "edit_image_region": "编辑图片区域",
+    "edit_image_region_hint": "在图片上拖拽框选要替换的区域。",
+    "choose_source_image_region_hint": "选择一张源图片并框选区域。",
+    "replace_source": "替换源图",
+    "choose_source": "选择源图",
+    "region_selected": "已选择区域",
+    "drag_to_select": "拖拽选择",
+    "clear_selection": "清除选择",
+    "choose_source_image_for_regional_editing": "选择源图片以进行区域编辑",
+    "choose_source_image_first": "请先选择源图片",
+    "select_edit_area_first": "请先选择要编辑的区域",
+    "describe_image_change_first": "请先描述图片修改内容"
   }
 }