Packages

ash_ai

0.2.14
0.7.3 0.7.2 0.7.1 0.7.0 0.6.1 0.6.0 0.5.0 0.4.0 0.3.0 0.2.14 0.2.13 0.2.12 0.2.11 0.2.10 0.2.9 0.2.8 0.2.7 0.2.6 0.2.5 0.2.4 0.2.3 0.2.2 0.2.1 0.2.0 0.1.11 0.1.10 0.1.9 0.1.8 0.1.7 0.1.6 0.1.5 0.1.4 0.1.3 0.1.1 0.1.0

Integrated LLM features for your Ash application.

HexDocs

Current section

Files

Jump to
Readme
35 Versions
11 Dependencies
Dependants
Files
Activity
ash_ai lib ash_ai mcp dev.ex
Raw

lib/ash_ai/mcp/dev.ex 

# SPDX-FileCopyrightText: 2024 ash_ai contributors <https://github.com/ash-project/ash_ai/graphs.contributors>

#

# SPDX-License-Identifier: MIT



defmodule AshAi.Mcp.Dev do

  @moduledoc """

  Place in your endpoint's code_reloading section to expose Ash dev MCP"



  Default path is `/ash_ai/mcp`

  """

  @behaviour Plug



  @impl true

  def init(opts) do

    path =

      opts

      |> Keyword.get(:path, "/ash_ai/mcp")

      |> String.split("/")

      |> Enum.reject(&(&1 == ""))



    opts =

      opts

      |> Keyword.put(:tools, :ash_dev_tools)

      |> Keyword.put(:path, path)



    AshAi.Mcp.Router.init(opts)

  end



  @impl true

  def call(%Plug.Conn{path_info: path_info} = conn, opts) do

    expected_path = Keyword.get(opts, :path)



    case Enum.split(path_info, length(expected_path)) do

      {^expected_path, rest} ->

        conn

        |> Plug.forward(rest, AshAi.Mcp.Router, opts)

        |> Plug.Conn.halt()



      _ ->

        conn

    end

  end

end