lm test tag

config/config.exs

@@ -1,5 +1,3 @@
 import Config
 
-config :agens,
-  registry: Agens.Registry,
-  real_llm: false
+config :agens, registry: Agens.Registry

mix.exs

@@ -14,7 +14,12 @@ defmodule Agens.MixProject do
       description:
         "Create multi-agent workflows with AI and Language Models using OTP components for reliable and scalable automation.",
       package: package(),
-      docs: docs()
+      docs: docs(),
+      aliases: aliases(),
+      preferred_cli_env: [
+        "test.all": :test,
+        "test.lm": :test
+      ]
     ]
   end
 
@@ -92,4 +97,11 @@ defmodule Agens.MixProject do
 
   defp elixirc_paths(:test), do: ["lib", "test/support"]
   defp elixirc_paths(_), do: ["lib"]
+
+  defp aliases do
+    [
+      "test.all": ["test --include lm"],
+      "test.lm": ["test --only lm"]
+    ]
+  end
 end

test/agens/job_test.exs

@@ -224,4 +224,12 @@ defmodule Agens.JobTest do
       assert_receive {:job_ended, ^name, :complete}
     end
   end
+
+  describe "lm" do
+
+    @tag :lm
+    test "run job" do
+      assert true
+    end
+  end
 end

test/support/serving.ex

@@ -41,7 +41,4 @@ defmodule Test.Support.Serving do
       {:reply, "STUB RUN", state}
     end
   end
-
-  def get(true), do: LLM.get()
-  def get(false), do: Stub.get()
 end

test/test_helper.exs

@@ -1,5 +1,3 @@
-real_llm? = Application.get_env(:agens, :real_llm, false)
-
 Supervisor.start_link(
   [
     {Agens.Supervisor, name: Agens.Supervisor}
@@ -9,10 +7,10 @@ Supervisor.start_link(
 
 %Agens.Serving.Config{
   name: :text_generation,
-  serving: Test.Support.Serving.get(real_llm?)
+  serving: Test.Support.Serving.Stub
 }
 |> Agens.Serving.start()
 
 :meck.new(Agent, [:passthrough])
 
-ExUnit.start()
+ExUnit.start(exclude: [:lm])