Initial commit

site/social_ai/ollama.py

@@ -0,0 +1,63 @@
+"""Ollama client for drafting social posts."""
+
+from __future__ import annotations
+
+import logging
+
+import requests
+from django.conf import settings
+
+logger = logging.getLogger(__name__)
+
+
+class OllamaError(RuntimeError):
+    pass
+
+
+def generate_social_post(
+    prompt: str,
+    *,
+    platform: str = "",
+    tone: str = "professional, warm, on-brand",
+) -> str:
+    """
+    Call the LAN Ollama endpoint (default http://10.0.0.128:11434) to draft copy.
+
+    Uses the standard Ollama /api/generate HTTP API.
+    """
+    base = (settings.OLLAMA_BASE_URL or "").rstrip("/")
+    if not base:
+        raise OllamaError("OLLAMA_BASE_URL is not configured")
+
+    system = (
+        "You write short social media posts for a local business. "
+        f"Tone: {tone}. "
+        "Return only the post text, no preamble."
+    )
+    if platform:
+        system += f" Optimize for {platform}."
+
+    full_prompt = f"{system}\n\nUser request:\n{prompt}"
+
+    url = f"{base}/api/generate"
+    payload = {
+        "model": settings.OLLAMA_MODEL,
+        "prompt": full_prompt,
+        "stream": False,
+    }
+    try:
+        response = requests.post(
+            url,
+            json=payload,
+            timeout=settings.OLLAMA_TIMEOUT_SECONDS,
+        )
+        response.raise_for_status()
+    except requests.RequestException as exc:
+        logger.exception("Ollama request failed")
+        raise OllamaError(f"Ollama unreachable at {url}: {exc}") from exc
+
+    data = response.json()
+    text = (data.get("response") or "").strip()
+    if not text:
+        raise OllamaError("Ollama returned empty response")
+    return text