Update src/macg/llm_openai.py

src/macg/llm_openai.py

@@ -10,7 +10,7 @@ class OpenAIResponsesLLM(LLMClient):
         api_key: str,
         model: str = "gpt-5",
         base_url: str = "https://api.openai.com/v1",
-        temperature: float = 0.2,
+        temperature: float | None = 0.2,
         max_output_tokens: int = 900,
     ) -> None:
         if not api_key:
@@ -21,25 +21,25 @@ class OpenAIResponsesLLM(LLMClient):
         self.client = OpenAI(api_key=api_key, base_url=base_url)
 
     def complete(self, system: str, prompt: str) -> str:
-    params = dict(
-        model=self.model,
-        instructions=system,
-        input=prompt,
-        max_output_tokens=self.max_output_tokens,
-    )
+        params = {
+            "model": self.model,
+            "instructions": system,
+            "input": prompt,
+            "max_output_tokens": self.max_output_tokens,
+        }
 
-    # try sending temperature (some models accept it)
-    if self.temperature is not None:
-        params["temperature"] = self.temperature
+        # Some models reject temperature; include it only if set
+        if self.temperature is not None:
+            params["temperature"] = self.temperature
 
-    try:
-        resp = self.client.responses.create(**params)
-        return resp.output_text
-    except Exception as e:
-        msg = str(e)
-        # If the model doesn't support temperature, retry without it
-        if "Unsupported parameter" in msg and "temperature" in msg:
-            params.pop("temperature", None)
+        try:
             resp = self.client.responses.create(**params)
             return resp.output_text
-        raise
+        except Exception as e:
+            msg = str(e)
+            # If the model doesn't support temperature, retry without it
+            if "Unsupported parameter" in msg and "temperature" in msg:
+                params.pop("temperature", None)
+                resp = self.client.responses.create(**params)
+                return resp.output_text
+            raise