소스 검색

Add type assertion for task_request in adaptor.go

IcedTangerine 4 달 전
부모
커밋
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 {
 		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{