Explorar o código

Add type assertion for task_request in adaptor.go

IcedTangerine hai 4 meses
pai
achega
5c792263ba
Modificáronse 1 ficheiros con 4 adicións e 1 borrados
  1. 4 1
      relay/channel/task/gemini/adaptor.go

+ 4 - 1
relay/channel/task/gemini/adaptor.go

@@ -129,7 +129,10 @@ func (a *TaskAdaptor) BuildRequestBody(c *gin.Context, info *relaycommon.RelayIn
 	if !ok {
 		return nil, fmt.Errorf("request not found in context")
 	}
-	req := v.(relaycommon.TaskSubmitReq)
+	req, ok := v.(relaycommon.TaskSubmitReq)
+	if !ok {
+		return nil, fmt.Errorf("unexpected task_request type")
+	}
 
 	// Create structured video generation request
 	body := GeminiVideoPayload{