chenchenandClaude Opus 4.6
f43aa269d6
- Replace all encoding/json direct calls with common.Marshal/Unmarshal/DecodeJson per Rule 1 - Fix Dify nil pointer dereference on remote image upload (relay-dify.go) - Fix Claude relay file content type detection for text/* and PDF (relay-claude.go) - Fix unsafe type assertions in Claude relay and Vertex GetModelRegion - Fix StreamScanner unconditionally resetting pre-existing StreamStatus - Add inferMimeTypeFromFilename() for proper MIME type handling in DTO - Fix Mac build script hardcoded DMG version (now reads from tauri.conf.json) Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
67 lines
1.8 KiB
Go
67 lines
1.8 KiB
Go
package middleware
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"net/http"
|
|
|
|
"github.com/heicode/manager/common"
|
|
"github.com/heicode/manager/constant"
|
|
relayconstant "github.com/heicode/manager/relay/constant"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func JimengRequestConvert() func(c *gin.Context) {
|
|
return func(c *gin.Context) {
|
|
action := c.Query("Action")
|
|
if action == "" {
|
|
abortWithOpenAiMessage(c, http.StatusBadRequest, "Action query parameter is required")
|
|
return
|
|
}
|
|
|
|
// Handle Jimeng official API request
|
|
var originalReq map[string]interface{}
|
|
if err := common.UnmarshalBodyReusable(c, &originalReq); err != nil {
|
|
abortWithOpenAiMessage(c, http.StatusBadRequest, "Invalid request body")
|
|
return
|
|
}
|
|
model, _ := originalReq["req_key"].(string)
|
|
prompt, _ := originalReq["prompt"].(string)
|
|
|
|
unifiedReq := map[string]interface{}{
|
|
"model": model,
|
|
"prompt": prompt,
|
|
"metadata": originalReq,
|
|
}
|
|
|
|
jsonData, err := common.Marshal(unifiedReq)
|
|
if err != nil {
|
|
abortWithOpenAiMessage(c, http.StatusInternalServerError, "Failed to marshal request body")
|
|
return
|
|
}
|
|
|
|
// Update request body
|
|
c.Request.Body = io.NopCloser(bytes.NewBuffer(jsonData))
|
|
c.Set(common.KeyRequestBody, jsonData)
|
|
|
|
if image, ok := originalReq["image"]; !ok || image == "" {
|
|
c.Set("action", constant.TaskActionTextGenerate)
|
|
}
|
|
|
|
c.Request.URL.Path = "/v1/video/generations"
|
|
|
|
if action == "CVSync2AsyncGetResult" {
|
|
taskId, ok := originalReq["task_id"].(string)
|
|
if !ok || taskId == "" {
|
|
abortWithOpenAiMessage(c, http.StatusBadRequest, "task_id is required for CVSync2AsyncGetResult")
|
|
return
|
|
}
|
|
c.Request.URL.Path = "/v1/video/generations/" + taskId
|
|
c.Request.Method = http.MethodGet
|
|
c.Set("task_id", taskId)
|
|
c.Set("relay_mode", relayconstant.RelayModeVideoFetchByID)
|
|
}
|
|
c.Next()
|
|
}
|
|
}
|