Fix bug of overwriting request token number

devchat-ai · May 12, 2023 · 604ea31 · 604ea31
1 parent e3b2064
commit 604ea31
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 9 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,4 +1,5 @@
 .DS_Store
+.vscode
 
 # PyCharm
 .idea/

diff --git a/.vscode/settings.json b/.vscode/settings.json
diff --git a/devchat/openai/openai_prompt.py b/devchat/openai/openai_prompt.py
@@ -77,7 +77,7 @@ def set_request(self, content: str) -> int:
             raise ValueError("The request cannot be empty.")
         message = OpenAIMessage(content, 'user')
         self._new_messages['request'] = message
-        self._request_tokens = message_tokens(message.to_dict(), self.model)
+        self._request_tokens += message_tokens(message.to_dict(), self.model)
 
     def set_response(self, response_str: str):
         """