Create chat.py

app/lib/chat.py

@@ -0,0 +1,28 @@
+from typing import List, Dict, Any
+from openai import OpenAI
+from ..config import settings
+
+def get_client() -> OpenAI:
+    if not settings.openai_api_key:
+        raise RuntimeError("OPENAI_API_KEY is not set in the Space (Settings → Variables).")
+    return OpenAI(api_key=settings.openai_api_key)
+
+def chat_completion(messages: List[Dict[str, str]], model: str | None = None, temperature: float = 0.7) -> Dict[str, Any]:
+    client = get_client()
+    used_model = model or settings.openai_model
+
+    resp = client.chat.completions.create(
+        model=used_model,
+        messages=messages,
+        temperature=temperature,
+    )
+    choice = resp.choices[0]
+    return {
+        "model": used_model,
+        "finish_reason": choice.finish_reason,
+        "message": {
+            "role": choice.message.role,
+            "content": choice.message.content,
+        },
+        "usage": resp.usage.model_dump() if hasattr(resp, "usage") and resp.usage else None
+    }