Create src/macg/llm_openai.py

src/macg/llm_openai.py

@@ -0,0 +1,40 @@
+from __future__ import annotations
+
+import os
+from openai import OpenAI
+
+from macg.llm_base import LLMClient
+
+
+class OpenAIResponsesLLM(LLMClient):
+    def __init__(
+        self,
+        model: str = "gpt-5",
+        api_key: str | None = None,
+        max_output_tokens: int = 900,
+        temperature: float = 0.2,
+        base_url: str | None = None,
+    ) -> None:
+        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)
+
+    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,
+        )
+        return resp.output_text