new plugin: openai gpt

3 files changed, 138 insertions, 0 deletions

diff --git a/lib/couch.ex b/lib/couch.ex
new file mode 100644
index 0000000..fdd8579
--- /dev/null
+++ b/lib/couch.ex
@@ -0,0 +1,18 @@
+defmodule Couch do
+  def get(db, doc) do
+    config = Application.get_env(:lsg, :couch)
+    url = [Keyword.get(config, :url), db, doc] |> Enum.join("/")
+    user = Keyword.get(config, :user)
+    pass = Keyword.get(config, :pass)
+    client_options = Keyword.get(config, :client_options, [])
+    headers = [{"accept", "application/json"}, {"user-agent", "beautte"}]
+    options = [hackney: [:insecure, {:basic_auth, {user, pass}}]] ++ client_options
+    case HTTPoison.get(url, headers, options) do
+      {:ok, %HTTPoison.Response{status_code: 200, body: body}} ->
+        {:ok, Poison.decode!(body)}
+      {:ok, %HTTPoison.Response{status_code: 404}} ->
+        {:error, :not_found}
+      error -> {:error, {:couchdb_error, error}}
+    end
+  end
+end
diff --git a/lib/lsg_irc/gpt_plugin.ex b/lib/lsg_irc/gpt_plugin.ex
new file mode 100644
index 0000000..f628f8d
--- /dev/null
+++ b/lib/lsg_irc/gpt_plugin.ex
@@ -0,0 +1,103 @@
+defmodule LSG.IRC.GptPlugin do
+  require Logger
+
+  def irc_doc() do
+    """
+    # OpenAI GPT
+
+    * **!gpt** list GPT tasks
+    * **!gpt `[task]` `<task args>`** run a task
+    * **?offensive `<content>`** is content offensive
+    """
+  end
+
+  @couch_db "bot-plugin-openai-prompts"
+  @trigger "gpt"
+
+  def start_link() do
+    GenServer.start_link(__MODULE__, [], name: __MODULE__)
+  end
+
+  def init(_) do
+    regopts = [plugin: __MODULE__]
+    {:ok, _} = Registry.register(IRC.PubSub, "trigger:#{@trigger}", regopts)
+    {:ok, _} = Registry.register(IRC.PubSub, "trigger:offensive", regopts)
+    {:ok, nil}
+  end
+
+  def handle_info({:irc, :trigger, @trigger, m = %IRC.Message{trigger: %IRC.Trigger{type: :bang, args: [task | args]}}}, state) do
+    case Couch.get(@couch_db, task) do
+      {:ok, task} -> task(m, task, Enum.join(args, " "))
+      {:error, :not_found} -> m.replyfun.("gpt: no such task: #{task}")
+      error ->
+        Logger.info("gpt: task load error: #{inspect error}")
+        m.replyfun.("gpt: database error")
+    end
+    {:noreply, state}
+  end
+
+  def handle_info({:irc, :trigger, @trigger, m = %IRC.Message{trigger: %IRC.Trigger{type: :bang, args: []}}}, state) do
+    case Couch.get(@couch_db, "_all_docs") do
+      {:ok, %{"rows" => []}} -> m.replyfun.("gpt: no tasks available")
+      {:ok, %{"rows" => tasks}} ->
+        tasks = tasks |> Enum.map(fn(task) -> Map.get(task, "id") end) |> Enum.join(", ")
+        m.replyfun.("gpt: tasks: #{tasks}")
+      error ->
+        Logger.info("gpt: task load error: #{inspect error}")
+        m.replyfun.("gpt: database error")
+    end
+    {:noreply, state}
+  end
+
+  def handle_info({:irc, :trigger, "offensive", m = %IRC.Message{trigger: %IRC.Trigger{type: :query, args: text}}}, state) do
+    text = Enum.join(text, " ")
+    {moderate?, moderation} = moderation(text, m.account.id)
+    reply = cond do
+      moderate? -> "⚠️ #{Enum.join(moderation, ", ")}"
+      !moderate? && moderation -> "👍"
+      !moderate? -> "☠️ error"
+    end
+    m.replyfun.(reply)
+    {:noreply, state}
+  end
+
+  def handle_info(_, state) do
+    {:noreply, state}
+  end
+
+  defp task(msg, task = %{"type" => "completions", "prompt" => prompt}, content) do
+    prompt = Tmpl.render(prompt, msg, %{"content" => content})
+    args = Map.get(task, "openai_params")
+    |> Map.put("prompt", prompt)
+    |> Map.put("user", msg.account.id)
+    {moderate?, moderation} = moderation(content, msg.account.id)
+    if moderate?, do: msg.replyfun.("⚠️ offensive input: #{Enum.join(moderation, ", ")}")
+    Logger.debug("GPT: request #{inspect args}")
+    case OpenAi.post("/v1/completions", args) do
+      {:ok, %{"choices" => [%{"text" => text} | _]}} ->
+        {moderate?, moderation} = moderation(text, msg.account.id)
+        if moderate?, do: msg.replyfun.("🚨 offensive output: #{Enum.join(moderation, ", ")}")
+        msg.replyfun.(String.trim(text))
+      error ->
+        Logger.error("gpt error: #{inspect error}")
+        msg.replyfun.("gpt: ☠️ ")
+    end
+  end
+
+  defp moderation(content, user_id) do
+    case OpenAi.post("/v1/moderations", %{"input" => content, "user" => user_id}) do
+      {:ok, %{"results" => [%{"flagged" => true, "categories" => categories} | _]}} ->
+        cat = categories
+        |> Enum.filter(fn({_key, value}) -> value end)
+        |> Enum.map(fn({key, _}) -> key end)
+        {true, cat}
+      {:ok, moderation} ->
+        Logger.debug("gpt: moderation: not flagged, #{inspect moderation}")
+        {false, true}
+      error ->
+        Logger.error("gpt: moderation error: #{inspect error}")
+        {false, false}
+    end
+  end
+
+end
diff --git a/lib/open_ai.ex b/lib/open_ai.ex
new file mode 100644
index 0000000..9feb9a4
--- /dev/null
+++ b/lib/open_ai.ex
@@ -0,0 +1,17 @@
+defmodule OpenAi do
+
+  def post(path, data, options \\ []) do
+    config = Application.get_env(:lsg, :openai, [])
+    url = "https://api.openai.com#{path}"
+    headers = [{"user-agent", "internal private experiment bot, href@random.sh"},
+               {"content-type", "application/json"},
+               {"authorization", "Bearer " <> Keyword.get(config, :key, "unset-api-key")}]
+    options = options ++ [timeout: :timer.seconds(180), recv_timeout: :timer.seconds(180)]
+    with {:ok, json} <- Poison.encode(data),
+         {:ok, %HTTPoison.Response{status_code: 200, body: body}} <- HTTPoison.post(url, json, headers, options),
+         {:ok, data} <- Poison.decode(body) do
+      {:ok, data}
+    end
+  end
+
+end