remove max_tokens

VisionaireLabs · Dec 19, 2024 · b8a6d10 · b8a6d10
1 parent a0674db
commit b8a6d10
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 5 deletions.
diff --git a/evaluate.py b/evaluate.py
@@ -109,7 +109,6 @@ def evaluate_final_screenshot(guideline):
             presence_penalty=1,
             frequency_penalty=1,
             temperature=0.7,
-            max_tokens=300,
         )
 
         eval_content = response.choices[0].message.content

diff --git a/operate/models/apis.py b/operate/models/apis.py
@@ -110,7 +110,6 @@ def call_gpt_4o(messages):
             presence_penalty=1,
             frequency_penalty=1,
             temperature=0.7,
-            max_tokens=3000,
         )
 
         content = response.choices[0].message.content
@@ -237,7 +236,6 @@ async def call_gpt_4o_with_ocr(messages, objective, model):
             model="o1",
             messages=messages,
             temperature=0.7,
-            max_tokens=3000,
         )
 
         content = response.choices[0].message.content
@@ -352,7 +350,6 @@ async def call_o1_with_ocr(messages, objective, model):
             model="gpt-4o",
             messages=messages,
             temperature=0.7,
-            max_tokens=3000,
         )
 
         content = response.choices[0].message.content
@@ -478,7 +475,6 @@ async def call_gpt_4o_labeled(messages, objective, model):
             presence_penalty=1,
             frequency_penalty=1,
             temperature=0.7,
-            max_tokens=3000,
         )
 
         content = response.choices[0].message.content