feat: 添加查询余额的交互能力 (#68)

README.md

@@ -81,6 +81,8 @@
 $ docker run -itd --name chatgpt -p 8090:8090 -e APIKEY=换成你的key -e SESSION_TIMEOUT=600 -e HTTP_PROXY="" -e DEFAULT_MODE="单聊" --restart=always dockerproxy.com/eryajf/chatgpt-dingtalk:latest
 ```
 
+`📢 注意：`如果你使用docker部署，那么proxy指定地址的时候，请指定宿主机的IP，而不要写成127，以免代理不生效。
+
 运行命令中映射的配置文件参考下边的配置文件说明。
 
 `第二种：基于配置文件挂载运行`
@@ -170,6 +172,12 @@ $ curl --location --request POST 'http:https://chat.eryajf.net/' \
 > 📢 注意：串聊模式下，群里每个人的聊天上下文是独立的。
 > 📢 注意：默认对话模式为单聊，因此不必发送单聊即可进入单聊模式，而要进入串聊，则需要发送串聊关键字进行切换，当串聊内容超过最大限制的时候，你可以发送重置，然后再次进入串聊模式。
 
+`查询余额`
+
+> 艾特机器人发送 `余额` 二字，会返回当前key对应的账号的剩余额度以及可用日期。
+
+![image_20230304_222522](https://cdn.staticaly.com/gh/eryajf/tu/main/img/image_20230304_222522.jpg)
+
 `实际聊天效果如下`
 
 ![image_20221209_163739](https://cdn.staticaly.com/gh/eryajf/tu/main/img/image_20221209_163739.png)

go.mod

@@ -3,13 +3,15 @@ module github.com/eryajf/chatgpt-dingtalk
 go 1.18
 
 require (
+ github.com/go-resty/resty/v2 v2.7.0
  github.com/patrickmn/go-cache v2.1.0+incompatible
  github.com/solywsh/chatgpt v0.0.14
 )
 
 require (
  github.com/joho/godotenv v1.5.1 // indirect
  github.com/sashabaranov/go-gpt3 v1.3.0 // indirect
+ golang.org/x/net v0.0.0-20211029224645-99673261e6eb // indirect
 )
 
 replace github.com/solywsh/chatgpt => ./pkg/chatgpt

go.sum

@@ -1,6 +1,15 @@
+github.com/go-resty/resty/v2 v2.7.0 h1:me+K9p3uhSmXtrBZ4k9jcEAfJmuC8IivWHwaLZwPrFY=
+github.com/go-resty/resty/v2 v2.7.0/go.mod h1:9PWDzw47qPphMRFfhsyk0NnSgvluHcljSMVIq3w7q0I=
 github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=
 github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4=
 github.com/patrickmn/go-cache v2.1.0+incompatible h1:HRMgzkcYKYpi3C8ajMPV8OFXaaRUnok+kx1WdO15EQc=
 github.com/patrickmn/go-cache v2.1.0+incompatible/go.mod h1:3Qf8kWWT7OJRJbdiICTKqZju1ZixQ/KpMGzzAfe6+WQ=
 github.com/sashabaranov/go-gpt3 v1.3.0 h1:IbvaK2yTnlm7f/oiC2HC9cbzu/4Znt4GkarFiwZ60uI=
 github.com/sashabaranov/go-gpt3 v1.3.0/go.mod h1:BIZdbwdzxZbCrcKGMGH6u2eyGe1xFuX9Anmh3tCP8lQ=
+golang.org/x/net v0.0.0-20211029224645-99673261e6eb h1:pirldcYWx7rx7kE5r+9WsOXPXK0+WH5+uZ7uPmJ44uM=
+golang.org/x/net v0.0.0-20211029224645-99673261e6eb/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
+golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
+golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
+golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=

main.go

@@ -6,6 +6,7 @@ import (
  "io/ioutil"
  "net/http"
  "strings"
+ "time"
 
  "github.com/eryajf/chatgpt-dingtalk/config"
  "github.com/eryajf/chatgpt-dingtalk/public"
@@ -25,6 +26,7 @@ var Welcome string = `Commands:
 🙋 单聊 👉 单独聊天
 📣 串聊 👉 带上下文聊天
 🔃 重置 👉 重置带上下文聊天
+💵 余额 👉 查询剩余额度
 🚀 帮助 👉 显示帮助信息
 =================================
 🚜 例：@我发送 空 或 帮助 将返回此帮助信息
@@ -107,6 +109,20 @@ func ProcessRequest(rmsg public.ReceiveMsg) error {
  if err != nil {
  logger.Warning(fmt.Errorf("send message error: %v", err))
  }
+ case "余额":
+ rst, err := public.GetBalance()
+ if err != nil {
+ logger.Warning(fmt.Errorf("get balance error: %v", err))
+ return err
+ }
+ t1 := time.Unix(int64(rst.Grants.Data[0].EffectiveAt), 0)
+ t2 := time.Unix(int64(rst.Grants.Data[0].ExpiresAt), 0)
+ msg := fmt.Sprintf("💵 已用: 💲%v\n💵 剩余: 💲%v\n⏳ 有效时间: 从 %v 到 %v\n", fmt.Sprintf("%.2f", rst.TotalUsed), fmt.Sprintf("%.2f", rst.TotalAvailable), t1.Format("2006-01-02 15:04:05"), t2.Format("2006-01-02 15:04:05"))
+
+ _, err = rmsg.ReplyText(msg, rmsg.SenderStaffId)
+ if err != nil {
+ logger.Warning(fmt.Errorf("send message error: %v", err))
+ }
  default:
  if public.FirstCheck(rmsg) {
  return Do("串聊", rmsg)

public/gpt.go

@@ -0,0 +1,46 @@
+package public
+
+import (
+ "encoding/json"
+ "fmt"
+ "time"
+
+ "github.com/eryajf/chatgpt-dingtalk/config"
+ "github.com/go-resty/resty/v2"
+)
+
+func InitAiCli() *resty.Client {
+ return resty.New().SetTimeout(30*time.Second).SetHeader("Authorization", fmt.Sprintf("Bearer %s", config.LoadConfig().ApiKey)).SetProxy(config.LoadConfig().HttpProxy)
+}
+
+type Billing struct {
+ Object string `json:"object"`
+ TotalGranted float64 `json:"total_granted"`
+ TotalUsed float64 `json:"total_used"`
+ TotalAvailable float64 `json:"total_available"`
+ Grants struct {
+ Object string `json:"object"`
+ Data []struct {
+ Object string `json:"object"`
+ ID string `json:"id"`
+ GrantAmount float64 `json:"grant_amount"`
+ UsedAmount float64 `json:"used_amount"`
+ EffectiveAt float64 `json:"effective_at"`
+ ExpiresAt float64 `json:"expires_at"`
+ } `json:"data"`
+ } `json:"grants"`
+}
+
+func GetBalance() (Billing, error) {
+ var data Billing
+ url := "https://api.openai.com/dashboard/billing/credit_grants"
+ resp, err := InitAiCli().R().Get(url)
+ if err != nil {
+ return data, err
+ }
+ err = json.Unmarshal(resp.Body(), &data)
+ if err != nil {
+ return data, err
+ }
+ return data, nil
+}