File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -42,7 +42,7 @@ def main(
4242 model : str ,
4343 port : int = 11434 ,
4444 pure : bool = False ,
45- thinking_budget : int = 1000 ,
45+ thinking_budget : int = 1024 ,
4646 output_file : str | None = None ,
4747 log_file : str = "evaluation_log.jsonl" ,
4848):
Original file line number Diff line number Diff line change @@ -103,7 +103,7 @@ def get_ant_ttc_model(
103103 client : Anthropic ,
104104 model : str = "claude-3-7-sonnet-20250219" ,
105105 pure : bool = False ,
106- thinking_budget : int = 1000 ,
106+ thinking_budget : int = 1024 ,
107107) -> Callable [[str ], str | None ]:
108108 def generate_type_signature (prompt : str ) -> str | None :
109109 try :
@@ -181,7 +181,7 @@ def get_gemini_ttc_model(
181181 client : genai .Client ,
182182 model : str = "gemini-2.5-flash-preview-04-17" ,
183183 pure : bool = False ,
184- thinking_budget : int = 1000 ,
184+ thinking_budget : int = 1024 ,
185185) -> Callable [[str ], str | None ]:
186186 def generate_type_signature (prompt : str ) -> str | None :
187187 response = client .models .generate_content (
You can’t perform that action at this time.
0 commit comments