Просмотр исходного кода

Add type assertion for task_request in adaptor.go

IcedTangerine 4 месяцев назад
Родитель
Сommit
5c792263ba
1 измененных файлов с 4 добавлено и 1 удалено
  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 {
 	if !ok {
 		return nil, fmt.Errorf("request not found in context")
 		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
 	// Create structured video generation request
 	body := GeminiVideoPayload{
 	body := GeminiVideoPayload{