通过文本长度限制每次提问的长度

app.py

@@ -1,11 +1,8 @@
 import openai
 import os
 import gradio as gr
-import tiktoken as tk
 
 openai.api_key = os.environ.get("OPENAI_API_KEY")
-# 定义tiktoken编码器
-encoding = tk.get_encoding("cl100k_base")
 # 定义提示词长度不能大于1024
 MAX_PROMPT_LENGTH = 1024
 # 提示词
@@ -14,13 +11,12 @@ prompt = """你是一个AI个人助手。你的回答需要满足以下要求:
 2. 回答限制在1000个字以内"""
 
 
-# 利用tiktoken校验提示词长度,防止提示词过长,方法返回Boolean值
-def check_prompt(question):
-    check_len = len(encoding.encode(prompt)) + len(encoding.encode(question))
-    if check_len > MAX_PROMPT_LENGTH:
+# 校验提示词长度
+def check_prompt_length(question):
+    text_length = len(prompt) + len(question)
+    if len(text_length) > MAX_PROMPT_LENGTH:
         return False
-    else:
-        return True
+    return True
 
 
 class Conversation:
@@ -31,10 +27,8 @@ class Conversation:
         self.messages.append({"role": "system", "content": self.prompt})
 
     def ask(self, question):
-        # 校验提示词长度，若长度大于限制长度，则直接返回错误提示
-        if not check_prompt(question):
-            return "Error: The length of prompt is too long!"
-
+        if not check_prompt_length(question):
+            return "你的输入太长了，请将文本长度限制在900以内，请重新输入"
         try:
             self.messages.append({"role": "user", "content": question})
             response = openai.ChatCompletion.create(

requirements.txt

@@ -1,2 +1 @@
-openai
-tiktoken
+openai