small fix

generate.py

@@ -1,5 +1,6 @@
 import os
 import re
+from typing import Optional
 from llms import build_prompt, get_completion, PROMPT_TEMPLATE, SYSTEM_PROMPT
 
 
@@ -25,8 +26,11 @@ def extract_code(completion: str) -> str:
     if code_block:
         return code_block.group(1)
 
-def get_solution_file_path(day: int, model: str):
-    """Returns the path to the solution file for the given day and model."""
+def get_solution_file_path(model: str, day: Optional[int] = None):
+    """Returns the path to the solution file for the given day and model. If day is None, returns the path to the solution file only."""
+    if day is None:
+        return f"solution_{model}.py"
+
     day_str = f"{day:02d}"
     return f"day{day_str}/solution_{model}.py"
 
@@ -57,7 +61,7 @@ if __name__ == "__main__":
                 print("-" * 80)
                 print(f"Generating code for {provider} {model}")
 
-                if os.path.exists(get_solution_file_path(day, model)):
+                if os.path.exists(get_solution_file_path(model=model, day=day)):
                     print(f"Skipping {provider} {model} for day {day} because it already exists")
                     continue