Create/update AI Configuration

config/test.exs

@@ -160,3 +160,17 @@ config :junit_formatter,
 config :trento,
        :flaky_tests_detection,
        enabled?: System.get_env("WRITE_JUNIT") == "1"
+
+config :trento, :ai,
+  providers: [
+    provider1: [
+      models: [
+        "model1"
+      ]
+    ],
+    provider2: [
+      models: [
+        "model1"
+      ]
+    ]
+  ]

lib/trento/ai/llm_registry.ex

@@ -43,11 +43,27 @@ defmodule Trento.AI.LLMRegistry do
     end)
   end
 
+  @doc """
+  Checks if a given model is supported by a specific provider.
+  """
+  @spec model_supported_by_provider?(bitstring(), atom()) :: boolean()
+  def model_supported_by_provider?(model, provider) do
+    model in get_provider_models(provider)
+  end
+
   @doc """
   Checks if a given model is supported by any provider.
   """
   @spec model_supported?(bitstring()) :: boolean()
   def model_supported?(model), do: model in get_provider_models(:all)
 
+  @doc """
+  Checks if a given provider is supported.
+  """
+  @spec provider_supported?(atom()) :: boolean()
+  def provider_supported?(provider) when is_atom(provider), do: provider in providers()
+
+  def provider_supported?(_), do: false
+
   defp get_ai_providers_config, do: Keyword.get(ApplicationConfigLoader.load(), :providers, [])
 end

lib/trento/ai/user_configuration.ex

@@ -14,6 +14,7 @@ defmodule Trento.AI.UserConfiguration do
   schema "ai_configurations" do
     field :model, :string
     field :provider, Ecto.Enum, values: LLMRegistry.providers()
+    # field :provider, :string
     field :api_key, EncryptedBinary, redact: true
 
     belongs_to :user, Trento.Users.User, primary_key: true
@@ -22,29 +23,53 @@ defmodule Trento.AI.UserConfiguration do
   end
 
   def changeset(ai_configuration, attrs) do
-    updated_attrs = maybe_update_provider(attrs)
-
     ai_configuration
-    |> cast(updated_attrs, [:user_id, :model, :provider, :api_key])
-    |> validate_required([:user_id, :model, :api_key])
-    |> validate_change(:model, &validate_model/2)
+    |> cast(attrs, [:user_id, :model, :provider, :api_key])
+    |> validate_required([:user_id, :model, :provider, :api_key])
+    |> validate_change(:provider, &validate_provider/2)
+    |> validate_model()
     |> unique_constraint(:user_id,
       name: :ai_configurations_pkey,
       message: "User already has a configuration"
     )
     |> foreign_key_constraint(:user_id, message: "User does not exist")
   end
 
-  defp maybe_update_provider(%{model: model} = attrs),
-    do: Map.put(attrs, :provider, LLMRegistry.get_model_provider(model))
-
-  defp maybe_update_provider(attrs), do: attrs
-
-  defp validate_model(_model_field_atom, model) do
-    if LLMRegistry.model_supported?(model) do
+  defp validate_provider(_provider_field_atom, provider) do
+    if LLMRegistry.provider_supported?(provider) do
       []
     else
-      [model: {"is not supported", validation: :ai_model_validity}]
+      [provider: {"is not supported", validation: :ai_provider_validity}]
     end
   end
+
+  defp validate_model(%{errors: [provider: _]} = changeset), do: changeset
+
+  defp validate_model(changeset) do
+    provider = get_field(changeset, :provider)
+    model = get_field(changeset, :model)
+
+    changeset
+    |> force_change(:model, model)
+    |> force_change(:provider, provider)
+    |> validate_change(:model, fn _model_atom, _model ->
+      model_supported? = LLMRegistry.model_supported?(model)
+      model_supported_by_provider? = LLMRegistry.model_supported_by_provider?(model, provider)
+
+      case {model_supported?, model_supported_by_provider?} do
+        {true, true} ->
+          []
+
+        {true, false} ->
+          [
+            model:
+              {"is not supported by the specified provider",
+               validation: :ai_model_provider_mismatch}
+          ]
+
+        {false, _} ->
+          [model: {"is not supported", validation: :ai_model_validity}]
+      end
+    end)
+  end
 end

lib/trento_web/controllers/v1/ai_configuration_controller.ex

@@ -0,0 +1,75 @@
+defmodule TrentoWeb.V1.AIConfigurationController do
+  use TrentoWeb, :controller
+  use OpenApiSpex.ControllerSpecs
+
+  alias Trento.Users.User
+
+  alias Trento.AI
+
+  alias Trento.AI.UserConfiguration
+
+  alias TrentoWeb.OpenApi.V1.Schema
+
+  import Plug.Conn
+
+  plug TrentoWeb.Plugs.LoadUserPlug
+
+  plug OpenApiSpex.Plug.CastAndValidate, json_render_error_v2: true
+  action_fallback TrentoWeb.FallbackController
+
+  operation :create_ai_configuration,
+    summary: "Creates User's AI Configuration",
+    description: "Creates a new AI configuration for the currently authenticated user.",
+    tags: ["Profile"],
+    request_body:
+      {"CreateUserAIConfiguration", "application/json", Schema.AI.CreateUserConfigurationRequest},
+    responses: [
+      created:
+        {"User AI Configuration created successfully.", "application/json",
+         Schema.AI.UserConfiguration},
+      unprocessable_entity: Schema.UnprocessableEntity.response(),
+      unauthorized: Schema.Unauthorized.response(),
+      forbidden: Schema.Forbidden.response()
+    ]
+
+  def create_ai_configuration(conn, _) do
+    %User{} = current_user = Pow.Plug.current_user(conn)
+
+    creation_params = OpenApiSpex.body_params(conn)
+
+    with {:ok, %UserConfiguration{} = user_ai_config} <-
+           AI.create_user_configuration(current_user, creation_params) do
+      conn
+      |> put_status(:created)
+      |> render(:ai_configuration, %{ai_configuration: user_ai_config})
+    end
+  end
+
+  operation :update_ai_configuration,
+    summary: "Updates User's AI Configuration",
+    description: "Updates the AI configuration for the currently authenticated user.",
+    tags: ["Profile"],
+    request_body:
+      {"UpdateUserAIConfiguration", "application/json", Schema.AI.UpdateUserConfigurationRequest},
+    responses: [
+      ok:
+        {"User AI Configuration updated successfully.", "application/json",
+         Schema.AI.UserConfiguration},
+      unprocessable_entity: Schema.UnprocessableEntity.response(),
+      unauthorized: Schema.Unauthorized.response(),
+      forbidden: Schema.Forbidden.response()
+    ]
+
+  def update_ai_configuration(conn, _) do
+    %User{} = current_user = Pow.Plug.current_user(conn)
+
+    update_params = OpenApiSpex.body_params(conn)
+
+    with {:ok, %UserConfiguration{} = user_ai_config} <-
+           AI.update_user_configuration(current_user, update_params) do
+      conn
+      |> put_status(:ok)
+      |> render(:ai_configuration, %{ai_configuration: user_ai_config})
+    end
+  end
+end

lib/trento_web/controllers/v1/ai_configuration_json.ex

@@ -0,0 +1,15 @@
+defmodule TrentoWeb.V1.AIConfigurationJSON do
+  def ai_configuration(%{ai_configuration: ai_configuration}),
+    do: ai_configuration_entry(ai_configuration)
+
+  def ai_configuration_entry(%{
+        provider: provider,
+        model: model
+      }),
+      do: %{
+        provider: provider,
+        model: model
+      }
+
+  def ai_configuration_entry(_), do: nil
+end

lib/trento_web/controllers/v1/profile_json.ex

@@ -1,5 +1,9 @@
 defmodule TrentoWeb.V1.ProfileJSON do
-  alias TrentoWeb.V1.{AbilityJSON, PersonalAccessTokensJSON}
+  alias TrentoWeb.V1.{
+    AbilityJSON,
+    AIConfigurationJSON,
+    PersonalAccessTokensJSON
+  }
 
   def profile(%{
         user: %{
@@ -32,7 +36,7 @@ defmodule TrentoWeb.V1.ProfileJSON do
         created_at: created_at,
         analytics_enabled: analytics_enabled_at != nil,
         analytics_eula_accepted: analytics_eula_accepted_at != nil,
-        ai_configuration: ai_configuration(ai_configuration),
+        ai_configuration: AIConfigurationJSON.ai_configuration_entry(ai_configuration),
         idp_user: length(user_identities) > 0,
         updated_at: updated_at
       }
@@ -49,15 +53,4 @@ defmodule TrentoWeb.V1.ProfileJSON do
         }
       }),
       do: %{secret: Base.encode32(secret, padding: false), secret_qr_encoded: secret_qr_encoded}
-
-  defp ai_configuration(%{
-         provider: provider,
-         model: model
-       }),
-       do: %{
-         provider: provider,
-         model: model
-       }
-
-  defp ai_configuration(_), do: nil
 end

lib/trento_web/openapi/v1/schema/ai.ex

@@ -0,0 +1,119 @@
+defmodule TrentoWeb.OpenApi.V1.Schema.AI do
+  @moduledoc false
+
+  require OpenApiSpex
+  alias OpenApiSpex.Schema
+
+  defmodule UserConfiguration do
+    @moduledoc false
+
+    OpenApiSpex.schema(
+      %{
+        title: "AIUserConfigurationV1",
+        description: "AI configuration for a user.",
+        type: :object,
+        nullable: true,
+        additionalProperties: false,
+        properties: %{
+          provider: %Schema{
+            type: :string,
+            description: "Chosen AI provider.",
+            example: "googleai",
+            nullable: false
+          },
+          model: %Schema{
+            type: :string,
+            description: "Chosen AI model.",
+            example: "gemini-2.0-flash",
+            nullable: false
+          }
+        },
+        example: %{
+          provider: "googleai",
+          model: "gemini-2.0-flash"
+        },
+        required: [:provider, :model]
+      },
+      struct?: false
+    )
+  end
+
+  defmodule CreateUserConfigurationRequest do
+    @moduledoc false
+
+    OpenApiSpex.schema(
+      %{
+        title: "CreateUserConfigurationRequestV1",
+        description: "AI configuration request for a user.",
+        type: :object,
+        additionalProperties: false,
+        properties: %{
+          provider: %Schema{
+            type: :string,
+            description: "AI provider.",
+            nullable: false,
+            example: "googleai"
+          },
+          model: %Schema{
+            type: :string,
+            description: "AI model.",
+            nullable: false,
+            example: "gemini-2.0-flash"
+          },
+          api_key: %Schema{
+            type: :string,
+            description: "AI API key.",
+            nullable: false,
+            example: "AIza..."
+          }
+        },
+        example: %{
+          provider: "googleai",
+          model: "gemini-2.0-flash",
+          api_key: "AIza..."
+        },
+        required: [:provider, :model, :api_key]
+      },
+      struct?: false
+    )
+  end
+
+  defmodule UpdateUserConfigurationRequest do
+    @moduledoc false
+
+    OpenApiSpex.schema(
+      %{
+        title: "UpdateUserConfigurationRequestV1",
+        description: "AI configuration request for a user.",
+        type: :object,
+        additionalProperties: false,
+        minProperties: 1,
+        properties: %{
+          provider: %Schema{
+            type: :string,
+            description: "AI provider.",
+            nullable: false,
+            example: "googleai"
+          },
+          model: %Schema{
+            type: :string,
+            description: "AI model.",
+            nullable: false,
+            example: "gemini-2.0-flash"
+          },
+          api_key: %Schema{
+            type: :string,
+            description: "AI API key.",
+            nullable: false,
+            example: "AIza..."
+          }
+        },
+        example: %{
+          api_key: "AIza...",
+          model: "gemini-2.0-flash"
+        }
+      },
+      struct?: false
+    )
+  end
+end