|
@@ -140,7 +140,7 @@ func relayTextHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
|
|
|
quotaDelta := quota - preConsumedQuota
|
|
quotaDelta := quota - preConsumedQuota
|
|
|
err := model.PostConsumeTokenQuota(tokenId, quotaDelta)
|
|
err := model.PostConsumeTokenQuota(tokenId, quotaDelta)
|
|
|
if err != nil {
|
|
if err != nil {
|
|
|
- common.SysError("Error consuming token remain quota: " + err.Error())
|
|
|
|
|
|
|
+ common.SysError("error consuming token remain quota: " + err.Error())
|
|
|
}
|
|
}
|
|
|
tokenName := c.GetString("token_name")
|
|
tokenName := c.GetString("token_name")
|
|
|
model.RecordLog(userId, model.LogTypeConsume, fmt.Sprintf("通过令牌「%s」使用模型 %s 消耗 %s(模型倍率 %.2f,分组倍率 %.2f)", tokenName, textRequest.Model, common.LogQuota(quota), modelRatio, groupRatio))
|
|
model.RecordLog(userId, model.LogTypeConsume, fmt.Sprintf("通过令牌「%s」使用模型 %s 消耗 %s(模型倍率 %.2f,分组倍率 %.2f)", tokenName, textRequest.Model, common.LogQuota(quota), modelRatio, groupRatio))
|
|
@@ -173,7 +173,7 @@ func relayTextHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
|
|
|
for scanner.Scan() {
|
|
for scanner.Scan() {
|
|
|
data := scanner.Text()
|
|
data := scanner.Text()
|
|
|
if len(data) < 6 { // must be something wrong!
|
|
if len(data) < 6 { // must be something wrong!
|
|
|
- common.SysError("Invalid stream response: " + data)
|
|
|
|
|
|
|
+ common.SysError("invalid stream response: " + data)
|
|
|
continue
|
|
continue
|
|
|
}
|
|
}
|
|
|
dataChan <- data
|
|
dataChan <- data
|
|
@@ -184,7 +184,7 @@ func relayTextHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
|
|
|
var streamResponse ChatCompletionsStreamResponse
|
|
var streamResponse ChatCompletionsStreamResponse
|
|
|
err = json.Unmarshal([]byte(data), &streamResponse)
|
|
err = json.Unmarshal([]byte(data), &streamResponse)
|
|
|
if err != nil {
|
|
if err != nil {
|
|
|
- common.SysError("Error unmarshalling stream response: " + err.Error())
|
|
|
|
|
|
|
+ common.SysError("error unmarshalling stream response: " + err.Error())
|
|
|
return
|
|
return
|
|
|
}
|
|
}
|
|
|
for _, choice := range streamResponse.Choices {
|
|
for _, choice := range streamResponse.Choices {
|
|
@@ -194,7 +194,7 @@ func relayTextHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
|
|
|
var streamResponse CompletionsStreamResponse
|
|
var streamResponse CompletionsStreamResponse
|
|
|
err = json.Unmarshal([]byte(data), &streamResponse)
|
|
err = json.Unmarshal([]byte(data), &streamResponse)
|
|
|
if err != nil {
|
|
if err != nil {
|
|
|
- common.SysError("Error unmarshalling stream response: " + err.Error())
|
|
|
|
|
|
|
+ common.SysError("error unmarshalling stream response: " + err.Error())
|
|
|
return
|
|
return
|
|
|
}
|
|
}
|
|
|
for _, choice := range streamResponse.Choices {
|
|
for _, choice := range streamResponse.Choices {
|