Function SearchTokenByToken

controller/token_search.go:27–63

(c *gin.Context)

}

func SearchTokenByToken(c *gin.Context) {
	var req TokenSearchRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		common.ApiError(c, err)
		return
	}

	// 去掉前缀"sk-"后再查询数据库
	tokenKey := strings.TrimPrefix(req.Token, "sk-")

	// 获取token信息
	token, err := model.GetTokenByKey(tokenKey, false)
	if err != nil {
		common.ApiError(c, err)
		return
	}

	// 获取配比关系，默认使用gpt-3.5-turbo的配比
	modelRatio, _, _ := ratio_setting.GetModelRatio("gpt-3.5-turbo")

	// 构造返回信息
	response := TokenInfoResponse{
		TokenName:      token.Name,
		RemainQuota:    token.RemainQuota,
		UsedQuota:      token.UsedQuota,
		UnlimitedQuota: token.UnlimitedQuota,
		ExpiredTime:    token.ExpiredTime,
		Status:         token.Status,
		ModelRatio:     modelRatio,
	}

	c.JSON(http.StatusOK, gin.H{
		"success": true,
		"message": "",
		"data":    response,
	})
}