Change var name and increase default

NirantK · Aug 2, 2023 · 470a83e · 470a83e
1 parent ea72cdf
commit 470a83e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/agentai/conversation.py b/agentai/conversation.py
@@ -13,7 +13,7 @@ class Message(BaseModel):
 
 
 class Conversation:
-    def __init__(self, history: List[Message] = [], id: Optional[str] = None, max_tokens: int = 100):
+    def __init__(self, history: List[Message] = [], id: Optional[str] = None, max_history_tokens: int = 200):
         self.history: List[Message] = history
         self.role_to_color = {
             "system": "red",