fix #I6Y10K [app_chatgpt]发问时增加token相关处理

This commit is contained in:
Chill
2023-04-24 15:22:48 +08:00
parent 28c28a480c
commit 641b443391

View File

@@ -197,6 +197,9 @@ class Channel(models.Model):
if c_history: if c_history:
messages += c_history messages += c_history
messages.append({"role": "user", "content": msg}) messages.append({"role": "user", "content": msg})
msg_len = sum(len(str(m)) for m in messages)
if msg_len * 2 >= ai.max_tokens * 0.6:
messages = [{"role": "user", "content": msg}]
if sync_config == 'sync': if sync_config == 'sync':
self.get_ai_response(ai, messages, channel, user_id, message) self.get_ai_response(ai, messages, channel, user_id, message)
else: else: