Skip to content

Commit d1fd159

Browse files
committed
fix: set default thinking_budget as 1024
1 parent 32902f5 commit d1fd159

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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
):

src/experiment.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff 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(

0 commit comments

Comments
 (0)