phi_4 chat template support fix after merge

brainlid · Jan 13, 2025 · 3630129 · 3630129
1 parent 674a13f
commit 3630129
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lib/utils/chat_templates.ex b/lib/utils/chat_templates.ex
@@ -79,6 +79,11 @@ defmodule LangChain.Utils.ChatTemplates do
 
   Note: The `:zephyr` format supports specific system messages.
 
+
+  ### `:phi_4`
+
+  The `:phi_4` template format is also supported.
+
   ## Template callback
 
   It's possible to pass a callback as a template.
@@ -88,7 +93,7 @@ defmodule LangChain.Utils.ChatTemplates do
   alias LangChain.Message
 
   @type template_callback :: ([Message.t()], Keyword.t() -> String.t())
-  @type chat_format :: :inst | :im_start | :llama_2 | :llama_3 | :zephyr | template_callback()
+  @type chat_format :: :inst | :im_start | :llama_2 | :llama_3 | :phi_4 | :zephyr | template_callback()
 
   # Option:
   # - `add_generation_prompt`: boolean. Defaults to False.