fix: default value for max_tokens

jitsi · Nov 28, 2024 · a8aab7d · a8aab7d
1 parent abf3e7a
commit a8aab7d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/skynet/modules/ttt/summaries/v1/models.py b/skynet/modules/ttt/summaries/v1/models.py
@@ -22,7 +22,7 @@ class Priority(Enum):
 class DocumentPayload(BaseModel):
     text: str
     hint: HintType = HintType.MEETING
-    max_tokens: int = 0
+    max_tokens: int = None
     priority: Priority = Priority.NORMAL
     prompt: str | None = None
 
@@ -44,7 +44,7 @@ class SummaryDocumentPayload(DocumentPayload):
                 {
                     'text': 'Your text here',
                     'hint': 'meeting',
-                    'max_tokens': 0,
+                    'max_tokens': None,
                     'priority': 'normal',
                     'prompt': summary_meeting,
                 }