set max_token

bioerrorlog · Feb 4, 2024 · bb20259 · bb20259
1 parent 2db7b47
commit bb20259
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/robot_gpt/robot.py b/robot_gpt/robot.py
@@ -95,7 +95,8 @@ def call_gpt4v(self) -> str:
         logger.info(f"Prompts: {self.prompts}")
         response = client.chat.completions.create(
             model="gpt-4-vision-preview",
-            messages=self.prompts
+            messages=self.prompts,
+            max_tokens=1200,
         )
         content = response.choices[0].message.content
         logger.info(f"Response content: {content}")