剩余token数改为4000 - history token

EvAnhaodong · Mar 27, 2023 · 23220a9 · 23220a9
1 parent 09ee6ca
commit 23220a9
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -1,10 +1,15 @@
 from setuptools import find_namespace_packages
 from setuptools import setup
+from pathlib import Path
+
+PATH = Path("README.md")
 
 setup(
     name="azureChatGPT",
-    version="0.0.1",
+    version="0.0.2",
     description="ChatGPT is a reverse engineering of Azure ChatGPT API",
+    long_description=open(PATH, encoding="utf-8").read(),
+    long_description_content_type="text/markdown",
     url="https://github.com/EvAnhaodong/azureChatGPT",
     author="yehaodong",
     author_email="yehaodong@genomics.cn",

diff --git a/src/azureChatGPT/azure.py b/src/azureChatGPT/azure.py
@@ -116,9 +116,9 @@ def get_token_count(self, convo_id: str = "default") -> int:
 
     def get_max_tokens(self, convo_id: str) -> int:
         """
-        Get max tokens
+        Get remaining tokens
         """
-        return self.max_tokens - self.get_token_count(convo_id)
+        return 4000 - self.get_token_count(convo_id)
 
     def __messages2prompt(self, convo_id: str = "default"):
         prompt = "".join(