辅助函数,用于统一处理响应
(c *gin.Context, channel *model.Channel, startTime time.Time, modelName string, responseMessage string, err error)
| 129 | |
| 130 | // 辅助函数,用于统一处理响应 |
| 131 | func returnResponse(c *gin.Context, channel *model.Channel, startTime time.Time, modelName string, responseMessage string, err error) { |
| 132 | tok := time.Now() |
| 133 | milliseconds := tok.Sub(startTime).Milliseconds() |
| 134 | if err != nil { |
| 135 | milliseconds = 0 |
| 136 | } |
| 137 | go channel.UpdateResponseTime(milliseconds) |
| 138 | consumedTime := float64(milliseconds) / 1000.0 |
| 139 | if err != nil { |
| 140 | c.JSON(http.StatusOK, model.ParamError.ToResponse(err)) |
| 141 | return |
| 142 | } |
| 143 | c.JSON(http.StatusOK, model.Success.ToResponse(ChannelTestResponse{ |
| 144 | Success: true, |
| 145 | Message: responseMessage, |
| 146 | Model: modelName, |
| 147 | Time: consumedTime, |
| 148 | })) |
| 149 | } |
| 150 | |
| 151 | func testChannel(ctx context.Context, channel *model.Channel, request *relaymodel.GeneralOpenAIRequest) (responseMessage string, err error, openaiErr *relaymodel.Error, actualModel string) { |
| 152 | //startTime := time.Now() |