From 00d216a16deb1fb74e7ee66935e6395876d48270 Mon Sep 17 00:00:00 2001
From: Miraclewang <549788813@qq.com>
Date: Sat, 15 Jul 2023 15:46:04 +0800
Subject: [PATCH] fix ai_token  method in collect_v2.py

---
 collect_v2.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/collect_v2.py b/collect_v2.py
index 8ada7aa..5ff44e4 100644
--- a/collect_v2.py
+++ b/collect_v2.py
@@ -83,7 +83,7 @@
                 model="gpt-3.5-turbo",
                 messages=conversation_state,
             )
-            ai_tokens = completion["usage"]["total_tokens"]
+            ai_tokens = ai_completion["usage"]["total_tokens"]
             total_tokens += ai_tokens
             ai_response = ai_completion["choices"][0]["message"]["content"]
             instruct += f"\n[Human] {response}\n[AI] {ai_response}"