Increase max tokens and add gpt-5

src/pips/core.py

@@ -30,7 +30,7 @@ class PIPSSolver:
     *,
     max_iterations: int = 8,
     temperature: float = 0.0,
-    max_tokens: int = 4096,
+    max_tokens: int = 50000,
     top_p: float = 1.0,
     interactive: bool = False,
     critic_model: Optional[LLMModel] = None,
@@ -477,10 +477,10 @@ Make sure to follow these formatting requirements exactly.
             img_b64 = img2base64(sample.image_input)
             content = [
                 {"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{img_b64}"}},
-                {"type": "text", "text": sample.text_input},
+                {"type": "text", "text": f"Question: {sample.text_input}"},
             ]
         else:
-            content = sample.text_input
+            content = f"Question: {sample.text_input}"
 
         conv = [
             {"role": "system", "content": system_content},

src/pips/model_registry.py

@@ -61,6 +61,8 @@ def _initialize_default_models():
     """Initialize the registry with default models."""
     
     # OpenAI Models
+    register_model("gpt-5", "openai", "OpenAI GPT-5")
+    register_model("gpt-5-mini", "openai", "OpenAI GPT-5-mini")
     register_model("gpt-4.1-2025-04-14", "openai", "OpenAI GPT-4.1")
     register_model("gpt-4o-2024-08-06", "openai", "OpenAI GPT-4o")
     register_model("gpt-4.1-mini-2025-04-14", "openai", "OpenAI GPT-4.1 Mini")
@@ -78,4 +80,4 @@ def _initialize_default_models():
     register_model("claude-3-5-haiku-latest", "anthropic", "Anthropic Claude 3.5 Haiku")
 
 # Initialize default models when module is imported
-_initialize_default_models() 
+_initialize_default_models() 

src/pips/models.py

@@ -124,8 +124,8 @@ class LLMModel:
 class OpenAIModel(LLMModel):
     """
     OpenAI GPT model interface.
-    
-    Supports GPT-4, GPT-4o, o3, and o4 model families with proper handling
+
+    Supports GPT-4, GPT-4o, o3, o4, and gpt-5 model families with proper handling
     of different model requirements (reasoning effort for o3/o4 models).
     """
     
@@ -188,7 +188,7 @@ class OpenAIModel(LLMModel):
         extra_args = {}
         
         # Configure parameters based on model type
-        if "o3" in self.model_name or "o4" in self.model_name:
+        if "o3" in self.model_name or "o4" in self.model_name or "gpt-5" in self.model_name:
             # Reasoning models have special parameters
             extra_args["reasoning_effort"] = "medium"
             extra_args["max_completion_tokens"] = 20000
@@ -233,7 +233,7 @@ class OpenAIModel(LLMModel):
         extra_args = {}
         
         # Configure parameters based on model type
-        if "o3" in self.model_name or "o4" in self.model_name:
+        if "o3" in self.model_name or "o4" in self.model_name or "gpt-5" in self.model_name:
             extra_args["reasoning_effort"] = "medium"
             extra_args["max_completion_tokens"] = 20000
         else: