|
|
|
@@ -189,6 +189,7 @@ func OaiStreamHandler(c *gin.Context, info *relaycommon.RelayInfo, resp *http.Re |
|
|
|
applyUsagePostProcessing(info, usage, common.StringToByteSlice(lastStreamData)) |
|
|
|
|
|
|
|
HandleFinalResponse(c, info, lastStreamData, responseId, createAt, model, systemFingerprint, usage, containStreamUsage) |
|
|
|
relaycommon.SetRelayChatID(c, responseId) |
|
|
|
|
|
|
|
return usage, nil |
|
|
|
} |
|
|
|
@@ -225,6 +226,8 @@ func OpenaiHandler(c *gin.Context, info *relaycommon.RelayInfo, resp *http.Respo |
|
|
|
return nil, types.NewOpenAIError(err, types.ErrorCodeBadResponseBody, http.StatusInternalServerError) |
|
|
|
} |
|
|
|
|
|
|
|
relaycommon.SetRelayChatID(c, simpleResponse.Id) |
|
|
|
|
|
|
|
if oaiError := simpleResponse.GetOpenAIError(); oaiError != nil && oaiError.Type != "" { |
|
|
|
return nil, types.WithOpenAIError(*oaiError, resp.StatusCode) |
|
|
|
} |
|
|
|
|