update cache rule

coaidev · Oct 25, 2023 · c248c35 · c248c35
1 parent bf56062
commit c248c35
Show file tree

Hide file tree

Showing 4 changed files with 50 additions and 0 deletions.
diff --git a/app/src/dialogs/Invitation.tsx b/app/src/dialogs/Invitation.tsx
diff --git a/app/src/i18n.ts b/app/src/i18n.ts
@@ -198,6 +198,15 @@ const resources = {
       docs: {
         title: "Open Docs",
       },
+      invitation: {
+        title: "Invitation Code",
+        "input-placeholder": "Please enter the invitation code",
+        "cancel": "Cancel",
+        "check": "Check",
+        "check-success": "Redeem Success",
+        "check-success-description": "Redeem Success! You have received {{amount}} points, start your AI journey!",
+        "check-failed": "Redeem Failed",
+      }
     },
   },
   cn: {
@@ -382,6 +391,15 @@ const resources = {
       docs: {
         title: "开放文档",
       },
+      invitation: {
+        title: "邀请码",
+        "input-placeholder": "请输入邀请码",
+        "cancel": "取消",
+        "check": "验证",
+        "check-success": "兑换成功",
+        "check-success-description": "兑换成功！您已获得 {{amount}} 点数，开始您的 AI 之旅吧！",
+        "check-failed": "兑换失败",
+      }
     },
   },
   ru: {
@@ -577,6 +595,15 @@ const resources = {
       docs: {
         title: "Открыть документы",
       },
+      invitation: {
+        title: "Код приглашения",
+        "input-placeholder": "Введите код приглашения",
+        "cancel": "Отмена",
+        "check": "Проверить",
+        "check-success": "Успешно",
+        "check-success-description": "Успешно! Вы получили {{amount}} очков, начните свое путешествие в мир AI!",
+        "check-failed": "Не удалось",
+      }
     },
   },
 };

diff --git a/globals/variables.go b/globals/variables.go
@@ -116,6 +116,21 @@ var LongContextModelArray = []string{
 	Claude2100k,
 }
 
+var FreeModelArray = []string{
+	GPT3Turbo,
+	GPT3Turbo0613,
+	GPT3Turbo0301,
+	GPT3Turbo16k,
+	GPT3Turbo16k0613,
+	GPT3Turbo16k0301,
+	Claude2,
+	ChatBison001,
+	BingCreative,
+	BingBalanced,
+	BingPrecise,
+	ZhiPuChatGLMLite,
+}
+
 func in(value string, slice []string) bool {
 	for _, item := range slice {
 		if item == value {
@@ -172,3 +187,7 @@ func IsZhiPuModel(model string) bool {
 func IsLongContextModel(model string) bool {
 	return in(model, LongContextModelArray)
 }
+
+func IsFreeModel(model string) bool {
+	return in(model, FreeModelArray)
+}
diff --git a/manager/cache.go b/manager/cache.go
@@ -29,6 +29,10 @@ func ExtractCacheData(c *gin.Context, props *CacheProps) *CacheData {
 }
 
 func SaveCacheData(c *gin.Context, props *CacheProps, data *CacheData) {
+	if !globals.IsFreeModel(props.Model) {
+		return
+	}
+
 	hash := utils.Md5Encrypt(utils.Marshal(props))
 	utils.GetCacheFromContext(c).Set(c, fmt.Sprintf(":niodata:%s", hash), utils.Marshal(data), time.Hour*12)
 }