Merge pull request #1 from upmaru/feature/create-agentic-message

Setup agentic namespace and agentic message creation

Author: zacksiri

lib/tama_ex/messaging/response.ex renamed to lib/tama_ex/agentic.ex

@@ -1,7 +1,10 @@
-defmodule TamaEx.Messaging.Response do
-  def create(client, body, options \\ []) do
-    with {:ok, validated_client} <- TamaEx.validate_client(client, ["api"]) do
-      path = "/chat/completions"
+defmodule TamaEx.Agentic do
+  alias TamaEx.Message.Params, as: MessageParams
+
+  def create_message(client, body, options \\ []) do
+    with {:ok, validated_client} <- TamaEx.validate_client(client, ["agentic"]),
+         {:ok, message_params} <- MessageParams.validate(body) do
+      path = "/messages"
       timeout = Keyword.get(options, :timeout) || 300_000
       headers = Keyword.get(options, :headers) || []
       stream? = Map.get(body, "stream") || Map.get(body, :stream)
@@ -29,7 +32,7 @@ defmodule TamaEx.Messaging.Response do
         Req.merge(validated_client,
           method: :post,
           url: path,
-          json: body,
+          json: %{message: message_params},
           receive_timeout: timeout,
           headers: headers
         )

lib/tama_ex/agentic/message.ex

@@ -0,0 +1,23 @@
+defmodule TamaEx.Agentic.Message do
+  use Ecto.Schema
+  import Ecto.Changeset
+
+  @primary_key false
+  embedded_schema do
+    field :id, :binary_id
+    field :identifier, :string
+    field :current_state, :string
+  end
+
+  def changeset(message, attrs) do
+    message
+    |> cast(attrs, [:id, :identifier, :current_state])
+    |> validate_required([:id, :identifier, :current_state])
+  end
+
+  def parse!(attrs) do
+    %__MODULE__{}
+    |> changeset(attrs)
+    |> apply_action!(:insert)
+  end
+end

lib/tama_ex/agentic/message/author.ex

@@ -0,0 +1,19 @@
+defmodule TamaEx.Agentic.Message.Author do
+  use Ecto.Schema
+  import Ecto.Changeset
+
+  @derive Jason.Encoder
+
+  @primary_key false
+  embedded_schema do
+    field :class, :string, default: "actor"
+    field :identifier, :string
+    field :source, :string
+  end
+
+  def changeset(author, attrs) do
+    author
+    |> cast(attrs, [:class, :identifier, :source])
+    |> validate_required([:class, :identifier, :source])
+  end
+end

lib/tama_ex/agentic/message/params.ex

@@ -0,0 +1,38 @@
+defmodule TamaEx.Agentic.Message.Params do
+  use Ecto.Schema
+  import Ecto.Changeset
+
+  @derive Jason.Encoder
+
+  alias TamaEx.Agentic.Message.Author
+  alias TamaEx.Agentic.Message.Thread
+
+  @valid_attrs [
+    :recipient,
+    :class,
+    :identifier,
+    :content,
+    :index
+  ]
+
+  @primary_key false
+  embedded_schema do
+    field :recipient, :string
+    field :class, :string, default: "user-message"
+    field :identifier, :string
+
+    embeds_one :author, Author
+    embeds_one :thread, Thread
+
+    field :content, :string
+    field :index, :integer
+  end
+
+  def changeset(params, attrs) do
+    params
+    |> cast(attrs, @valid_attrs)
+    |> validate_required(@valid_attrs)
+    |> cast_embed(:author, required: true)
+    |> cast_embed(:thread, required: true)
+  end
+end

lib/tama_ex/agentic/message/thread.ex

@@ -0,0 +1,18 @@
+defmodule TamaEx.Agentic.Message.Thread do
+  use Ecto.Schema
+  import Ecto.Changeset
+
+  @derive Jason.Encoder
+
+  @primary_key false
+  embedded_schema do
+    field :class, :string, default: "thread"
+    field :identifier, :string
+  end
+
+  def changeset(thread, attrs) do
+    thread
+    |> cast(attrs, [:class, :identifier])
+    |> validate_required([:identifier])
+  end
+end

lib/tama_ex/message/params.ex

@@ -0,0 +1,51 @@
+defmodule TamaEx.Message.Params do
+  use Ecto.Schema
+  import Ecto.Changeset
+
+  @derive Jason.Encoder
+
+  alias TamaEx.Message.Params.Author
+  alias TamaEx.Message.Params.Thread
+
+  @valid_attrs [
+    :recipient,
+    :class,
+    :identifier,
+    :content,
+    :index,
+    :stream
+  ]
+
+  @primary_key false
+  embedded_schema do
+    field :recipient, :string
+    field :class, :string, default: "user-message"
+    field :identifier, :string
+    field :content, :string
+    field :index, :integer
+    field :stream, :boolean, default: false
+
+    embeds_one :author, Author
+    embeds_one :thread, Thread
+  end
+
+  def changeset(params, attrs) do
+    params
+    |> cast(attrs, @valid_attrs)
+    |> validate_required([:recipient, :identifier, :content, :index])
+    |> cast_embed(:author, required: true)
+    |> cast_embed(:thread, required: true)
+  end
+
+  def validate(attrs) when is_map(attrs) do
+    %__MODULE__{}
+    |> changeset(attrs)
+    |> case do
+      %Ecto.Changeset{valid?: true} = changeset ->
+        {:ok, apply_changes(changeset)}
+
+      %Ecto.Changeset{valid?: false} = changeset ->
+        raise Ecto.InvalidChangesetError, action: :validate, changeset: changeset
+    end
+  end
+end

lib/tama_ex/message/params/author.ex

@@ -0,0 +1,19 @@
+defmodule TamaEx.Message.Params.Author do
+  use Ecto.Schema
+  import Ecto.Changeset
+
+  @derive Jason.Encoder
+
+  @primary_key false
+  embedded_schema do
+    field :class, :string, default: "actor"
+    field :identifier, :string
+    field :source, :string
+  end
+
+  def changeset(author, attrs) do
+    author
+    |> cast(attrs, [:class, :identifier, :source])
+    |> validate_required([:identifier, :source])
+  end
+end

lib/tama_ex/message/params/thread.ex

@@ -0,0 +1,18 @@
+defmodule TamaEx.Message.Params.Thread do
+  use Ecto.Schema
+  import Ecto.Changeset
+
+  @derive Jason.Encoder
+
+  @primary_key false
+  embedded_schema do
+    field :class, :string, default: "thread"
+    field :identifier, :string
+  end
+
+  def changeset(thread, attrs) do
+    thread
+    |> cast(attrs, [:class, :identifier])
+    |> validate_required([:identifier])
+  end
+end