Cyoa

.github/workflows/main.yml

@@ -1,14 +1,14 @@
 name: Fly Deploy
 on:
   push:
-    branches:
-      - main
+
 jobs:
   deploy:
     name: Deploy app
     runs-on: ubuntu-latest
+    concurrency: deploy-group    # ensure only one action runs at a time
     steps:
-      - uses: actions/checkout@v3
+      - uses: actions/checkout@v4
       - uses: superfly/flyctl-actions/setup-flyctl@master
       - run: flyctl deploy --remote-only
         env:

.gitignore

@@ -36,4 +36,5 @@ npm-debug.log
 /assets/node_modules/
 
 .env
-chatbots*.db*
+chatbots*.db*
+priv/static/images/*

config/dev.exs

@@ -9,7 +9,7 @@ import Config
 config :chat_bots, ChatBotsWeb.Endpoint,
   # Binding to loopback ipv4 address prevents access from other machines.
   # Change to `ip: {0, 0, 0, 0}` to allow access from other machines.
-  http: [ip: {127, 0, 0, 1}, port: 4000],
+  http: [ip: {127, 0, 0, 1}, port: 4001],
   check_origin: false,
   code_reloader: true,
   debug_errors: true,
@@ -53,7 +53,8 @@ config :chat_bots, ChatBots.Repo,
 config :chat_bots, ChatBotsWeb.Endpoint,
   live_reload: [
     patterns: [
-      ~r"priv/static/.*(js|css|png|jpeg|jpg|gif|svg)$",
+      # ~r"priv/static/.*(js|css|png|jpeg|jpg|gif|svg)$",
+      ~r"priv/static/.*(js|css)$",
       ~r"lib/chat_bots_web/(controllers|live|components)/.*(ex|heex)$"
     ]
   ]
@@ -70,3 +71,6 @@ config :phoenix, :stacktrace_depth, 20
 
 # Initialize plugs at runtime for faster development compilation
 config :phoenix, :plug_init_mode, :runtime
+
+# path for downloading images
+config :chat_bots, download_path: "priv/static/images/generated"

config/runtime.exs

@@ -33,6 +33,9 @@ config :openai,
   organization_key: env!("OPENAI_ORG_KEY"),
   http_options: [recv_timeout: 30_000]
 
+# Stability.ai configuration
+config :chat_bots, stability_ai_api_key: env!("STABILITY_AI_API_KEY")
+
 if config_env() == :prod do
   # The secret key base is used to sign/encrypt cookies and other secrets.
   # A default value is used in config/dev.exs and config/test.exs but you
@@ -69,6 +72,9 @@ if config_env() == :prod do
   # get password from env (will raise if not set)
   config :chat_bots, :auth, password: System.fetch_env!("USER_PASSWORD")
 
+  # path for downloading generated images
+  config :chat_bots, download_path: "/data/generated-images"
+
   # ## SSL Support
   #
   # To get SSL working, you will need to add the `https` key

config/test.exs

@@ -18,3 +18,6 @@ config :logger, level: :warning
 
 # Initialize plugs at runtime for faster test compilation
 config :phoenix, :plug_init_mode, :runtime
+
+# set download path for generated images to /tmp
+config :chat_bots, :download_path, "/tmp"

lib/chat_bots/application.ex

@@ -28,8 +28,9 @@ defmodule ChatBots.Application do
     opts = [strategy: :one_for_one, name: ChatBots.Supervisor]
     result = Supervisor.start_link(children, opts)
 
-    # Temporarily reset the database on start during development
-    ChatBots.Seeder.reset()
+    # Update the bots
+    ChatBots.Seeder.update_bots()
+
     result
   end
 

lib/chat_bots/bots.ex

@@ -18,4 +18,16 @@ defmodule ChatBots.Bots do
     |> where([b], b.id == ^id)
     |> Repo.one()
   end
+
+  @doc """
+  create_bot/1 creates a new bot with the given attributes.
+  """
+  def create_bot(attrs) do
+    %Bot{}
+    |> Bot.changeset(attrs)
+    |> Repo.insert(
+      on_conflict: {:replace_all_except, [:id, :inserted_at]},
+      conflict_target: :name
+    )
+  end
 end

lib/chat_bots/bots/bot.ex

@@ -5,14 +5,16 @@ defmodule ChatBots.Bots.Bot do
   schema "bots" do
     field(:directive, :string)
     field(:name, :string)
+    field(:json_mode, :boolean, default: false)
 
     timestamps()
   end
 
   @doc false
   def changeset(bot, attrs) do
     bot
-    |> cast(attrs, [:name, :directive])
+    |> cast(attrs, [:name, :directive, :json_mode])
     |> validate_required([:name, :directive])
+    |> unique_constraint(:name)
   end
 end

lib/chat_bots/chats.ex

@@ -1,21 +1,81 @@
 defmodule ChatBots.Chats do
-  alias ChatBots.Chats.{Chat, Message}
   alias ChatBots.Bots
+  alias ChatBots.Chats.Chat
+  alias ChatBots.Chats.Message
+  alias ChatBots.Repo
+
+  import Ecto.Changeset
+  import Ecto.Query
+
+  @doc """
+  Creates a new chat for the given bot_id.
+  Adds a message with the bot's system prompt.
+  """
+  def create_chat(bot) do
+    %Chat{}
+    |> change(bot_id: bot.id)
+    |> put_assoc(:messages, [%Message{role: "system", content: bot.directive}])
+    |> Repo.insert!()
+  end
+
+  @doc """
+  Retrieves a chat by id
+  """
+  def get_chat!(id) do
+    Repo.get!(Chat, id)
+    |> Repo.preload([:messages, :bot])
+  end
+
+  @doc """
+  Creates a new message for the given chat
+  """
+  def create_message(chat, attrs) do
+    %Message{}
+    |> cast(attrs, [:role, :content])
+    |> put_assoc(:chat, chat)
+    |> Repo.insert()
+  end
 
   @doc """
   Creates a new chat with the given bot_id.
   """
   def new_chat(bot_id) do
     bot = Bots.get_bot(bot_id)
     system_prompt = %Message{role: "system", content: bot.directive}
-    %Chat{bot_id: bot_id, messages: [system_prompt]}
+    [system_prompt]
   end
 
   @doc """
   Adds a message to the chat.
   """
-  # TODO: enforce that it only accepts a %Message struct
-  def add_message(chat, message) do
-    %Chat{chat | messages: chat.messages ++ [message]}
+  def add_message(messages, message) do
+    messages ++ [message]
+  end
+
+  @doc """
+  Lists all chats
+  Preloads messages
+  """
+  def list_chats do
+    preload_query = preload_latest_message_query()
+
+    from(c in Chat,
+      order_by: [desc: :inserted_at],
+      preload: [:bot, :messages, latest_message: ^preload_query]
+    )
+    |> Repo.all()
+  end
+
+  defp preload_latest_message_query do
+    ranking_query =
+      from(m in Message,
+        select: %{id: m.id, row_number: over(row_number(), :message_partition)},
+        windows: [message_partition: [partition_by: :chat_id, order_by: [desc: m.inserted_at]]]
+      )
+
+    from(m in Message,
+      join: r in subquery(ranking_query),
+      on: m.id == r.id and r.row_number == 1
+    )
   end
 end

lib/chat_bots/chats/bubble.ex

@@ -0,0 +1,3 @@
+defmodule ChatBots.Chats.Bubble do
+  defstruct [:type, :text]
+end

lib/chat_bots/chats/chat.ex

@@ -1,3 +1,14 @@
 defmodule ChatBots.Chats.Chat do
-  defstruct [:bot_id, :messages]
+  use Ecto.Schema
+  alias ChatBots.Bots.Bot
+  alias ChatBots.Chats.Message
+
+  @primary_key {:id, :binary_id, autogenerate: true}
+  schema "chats" do
+    belongs_to(:bot, Bot)
+    has_many(:messages, Message)
+    has_one(:latest_message, Message)
+
+    timestamps()
+  end
 end

lib/chat_bots/chats/choice.ex

@@ -0,0 +1,3 @@
+defmodule ChatBots.Chats.Choice do
+  defstruct [:options]
+end

lib/chat_bots/chats/image.ex

@@ -0,0 +1,3 @@
+defmodule ChatBots.Chats.Image do
+  defstruct [:file, :prompt]
+end

lib/chat_bots/chats/image_request.ex

@@ -0,0 +1,3 @@
+defmodule ChatBots.Chats.ImageRequest do
+  defstruct [:prompt]
+end

lib/chat_bots/chats/message.ex

@@ -1,3 +1,22 @@
 defmodule ChatBots.Chats.Message do
-  defstruct [:role, :content]
+  use Ecto.Schema
+  import Ecto.Changeset
+
+  @primary_key {:id, :binary_id, autogenerate: true}
+  @foreign_key_type :binary_id
+  schema "messages" do
+    field(:role, :string)
+    field(:content, :string)
+
+    belongs_to(:chat, ChatBots.Chats.Chat)
+
+    timestamps()
+  end
+
+  @doc false
+  def changeset(message, attrs) do
+    message
+    |> cast(attrs, [:role, :content])
+    |> validate_required([:role, :content])
+  end
 end

lib/chat_bots/open_ai/api.ex

@@ -0,0 +1,41 @@
+defmodule ChatBots.OpenAi.Api do
+  alias ChatBots.OpenAi.Client
+
+  @model "gpt-4o-mini"
+
+  @doc """
+  Sends a message to the chat bot and returns the updated chat.
+  """
+  def send_message(bot, messages) do
+    params = prepare_params(bot, messages)
+
+    case Client.chat_completion(params) do
+      {:ok, %{choices: [choice | _]}} ->
+        {:ok, choice["message"]}
+
+      {:error, :timeout} ->
+        {:error, %{"message" => "Your request timed out"}}
+
+      {:error, error} ->
+        {:error, error["error"]}
+    end
+  end
+
+  defp prepare_params(bot, messages) do
+    [messages: messages]
+    |> add_model()
+    |> maybe_add_json_mode(bot)
+  end
+
+  defp add_model(params) do
+    params ++ [model: @model]
+  end
+
+  defp maybe_add_json_mode(params, bot) do
+    if bot.json_mode do
+      params ++ [response_format: %{type: "json_object"}]
+    else
+      params
+    end
+  end
+end

lib/chat_bots/open_ai/client.ex

@@ -8,8 +8,8 @@ defmodule ChatBots.OpenAi.Client do
   @callback chat_completion(model: String.t(), messages: [Message.t()]) ::
               {:ok, map()} | {:error, map()}
 
-  def chat_completion(model: model, messages: messages) do
-    client().chat_completion(model: model, messages: messages)
+  def chat_completion(options) do
+    client().chat_completion(options)
   end
 
   defp client do

lib/chat_bots/open_ai/http_client.ex

@@ -5,7 +5,7 @@ defmodule ChatBots.OpenAi.HttpClient do
   @behaviour ChatBots.OpenAi.Client
 
   @impl true
-  def chat_completion(model: model, messages: messages) do
-    OpenAI.chat_completion(model: model, messages: messages)
+  def chat_completion(params) do
+    OpenAI.chat_completion(params)
   end
 end