Update src/macg/llm_openai.py

src/macg/llm_openai.py

@@ -1,40 +1,31 @@
 from __future__ import annotations
 
-import os
 from openai import OpenAI
-
 from macg.llm_base import LLMClient
 
 
 class OpenAIResponsesLLM(LLMClient):
     def __init__(
         self,
+        api_key: str,
         model: str = "gpt-5",
-        api_key: str | None = None,
-        max_output_tokens: int = 900,
+        base_url: str = "https://api.openai.com/v1",
         temperature: float = 0.2,
-        base_url: str | None = None,
+        max_output_tokens: int = 900,
     ) -> None:
+        if not api_key:
+            raise ValueError("OpenAI api_key is required.")
         self.model = model
-        self.api_key = api_key or os.getenv("OPENAI_API_KEY")
-        if not self.api_key:
-            raise ValueError("OPENAI_API_KEY is not set.")
-
-        self.max_output_tokens = max_output_tokens
         self.temperature = temperature
-
-        if base_url:
-            self.client = OpenAI(api_key=self.api_key, base_url=base_url)
-        else:
-            self.client = OpenAI(api_key=self.api_key)
+        self.max_output_tokens = max_output_tokens
+        self.client = OpenAI(api_key=api_key, base_url=base_url)
 
     def complete(self, system: str, prompt: str) -> str:
-        # Uses the Responses API
         resp = self.client.responses.create(
             model=self.model,
             instructions=system,
             input=prompt,
-            max_output_tokens=self.max_output_tokens,
             temperature=self.temperature,
+            max_output_tokens=self.max_output_tokens,
         )
         return resp.output_text