current只保留role和content

EvAnhaodong · Apr 21, 2023 · 0662ae7 · 0662ae7
1 parent 98f9cb1
commit 0662ae7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -6,7 +6,7 @@
 
 setup(
     name="azureChatGPT",
-    version="0.0.4",
+    version="0.0.5",
     description="ChatGPT is a reverse engineering of Azure ChatGPT API",
     long_description=open(PATH, encoding="utf-8").read(),
     long_description_content_type="text/markdown",

diff --git a/src/azureChatGPT/azure.py b/src/azureChatGPT/azure.py
@@ -88,7 +88,7 @@ def __truncate_conversation(self, convo_id: str = "default") -> None:
         Truncate the conversation
         """
         self.conversation["current"] = [
-            dict((key, sentence[key]) for key in sentence if key != "name")
+            dict((key, sentence[key]) for key in ["role", "content"])
             for sentence in self.conversation[convo_id]
         ]