Add safety handling to google model

elpassion · Feb 24, 2024 · 6f09ce9 · 6f09ce9
1 parent c7b24bd
commit 6f09ce9
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/apps/api/lib/buildel/langchain/chat_google_ai.ex b/apps/api/lib/buildel/langchain/chat_google_ai.ex
@@ -474,6 +474,17 @@ defmodule Buildel.LangChain.ChatModels.ChatGoogleAI do
     )
   end
 
+  def do_process_response(%{"finishReason" => "SAFETY", "index" => index}, message_type) do
+    do_process_response(
+      %{
+        "finishReason" => "DOESNOTMATTER",
+        "content" => %{"role" => "model", "parts" => [%{"text" => "Safety"}]},
+        "index" => index
+      },
+      message_type
+    )
+  end
+
   def do_process_response(%{"error" => %{"message" => reason}}, _) do
     Logger.error("Received error from API: #{inspect(reason)}")
     {:error, reason}