feat: 优化密钥测试 (#140)

* fix: rm max_tokens

* feat: 测试耗时

Author: tbphp

internal/channel/anthropic_channel.go

@@ -90,8 +90,7 @@ func (ch *AnthropicChannel) ValidateKey(ctx context.Context, key string) (bool,
 
 	// Use a minimal, low-cost payload for validation
 	payload := gin.H{
-		"model":      ch.TestModel,
-		"max_tokens": 100,
+		"model": ch.TestModel,
 		"messages": []gin.H{
 			{"role": "user", "content": "hi"},
 		},

internal/channel/openai_channel.go

@@ -93,7 +93,6 @@ func (ch *OpenAIChannel) ValidateKey(ctx context.Context, key string) (bool, err
 		"messages": []gin.H{
 			{"role": "user", "content": "hi"},
 		},
-		"max_tokens": 100,
 	}
 	body, err := json.Marshal(payload)
 	if err != nil {

internal/handler/key_handler.go

@@ -8,6 +8,7 @@ import (
 	"log"
 	"strconv"
 	"strings"
+	"time"
 
 	"github.com/gin-gonic/gin"
 	"gorm.io/gorm"
@@ -247,7 +248,9 @@ func (s *Server) TestMultipleKeys(c *gin.Context) {
 		return
 	}
 
+	start := time.Now()
 	results, err := s.KeyService.TestMultipleKeys(group, req.KeysText)
+	duration := time.Since(start).Milliseconds()
 	if err != nil {
 		if strings.Contains(err.Error(), "batch size exceeds the limit") {
 			response.Error(c, app_errors.NewAPIError(app_errors.ErrValidation, err.Error()))
@@ -259,7 +262,10 @@ func (s *Server) TestMultipleKeys(c *gin.Context) {
 		return
 	}
 
-	response.Success(c, results)
+	response.Success(c, gin.H{
+		"results":        results,
+		"total_duration": duration,
+	})
 }
 
 // ValidateGroupKeys initiates a manual validation task for all keys in a group.

web/src/api/keys.ts

@@ -91,13 +91,14 @@ export const keysApi = {
   async testKeys(
     group_id: number,
     keys_text: string
-  ): Promise<
-    {
+  ): Promise<{
+    results: {
       key_value: string;
       is_valid: boolean;
       error: string;
-    }[]
-  > {
+    }[];
+    total_duration: number;
+  }> {
     const res = await http.post(
       "/keys/test-multiple",
       {

web/src/components/keys/GroupFormModal.vue

@@ -843,15 +843,15 @@ async function handleSubmit() {
                         <template #trigger>
                           <n-icon :component="HelpCircleOutline" class="help-icon config-help" />
                         </template>
-                        使用JSON格式定义要覆盖的API请求参数。例如： {&quot;temperature&quot;: 0.7,
-                        &quot;max_tokens&quot;: 2000}。这些参数会在发送请求时合并到原始参数中
+                        使用JSON格式定义要覆盖的API请求参数。例如： {&quot;temperature&quot;:
+                        0.7}。这些参数会在发送请求时合并到原始参数中
                       </n-tooltip>
                     </div>
                   </template>
                   <n-input
                     v-model:value="formData.param_overrides"
                     type="textarea"
-                    placeholder='{"temperature": 0.7, "max_tokens": 2000}'
+                    placeholder='{"temperature": 0.7}'
                     :rows="4"
                   />
                 </n-form-item>

web/src/components/keys/KeyTable.vue

@@ -223,10 +223,10 @@ async function testKey(_key: KeyRow) {
   });
 
   try {
-    const res = await keysApi.testKeys(props.selectedGroup.id, _key.key_value);
-    const curValid = res?.[0] || {};
+    const response = await keysApi.testKeys(props.selectedGroup.id, _key.key_value);
+    const curValid = response.results?.[0] || {};
     if (curValid.is_valid) {
-      window.$message.success("密钥测试成功");
+      window.$message.success(`密钥测试成功 (耗时: ${formatDuration(response.total_duration)})`);
     } else {
       window.$message.error(curValid.error || "密钥测试失败: 无效的API密钥", {
         keepAliveOnHover: true,
@@ -245,6 +245,29 @@ async function testKey(_key: KeyRow) {
   }
 }
 
+function formatDuration(ms: number): string {
+  if (ms < 0) {
+    return "0ms";
+  }
+
+  const minutes = Math.floor(ms / 60000);
+  const seconds = Math.floor((ms % 60000) / 1000);
+  const milliseconds = ms % 1000;
+
+  let result = "";
+  if (minutes > 0) {
+    result += `${minutes}m`;
+  }
+  if (seconds > 0) {
+    result += `${seconds}s`;
+  }
+  if (milliseconds > 0 || result === "") {
+    result += `${milliseconds}ms`;
+  }
+
+  return result;
+}
+
 function toggleKeyVisibility(key: KeyRow) {
   key.is_visible = !key.is_visible;
 }