Merge pull request #4019 from nulib/4581-update-model-name

Update the OpenSearch ingest pipeline to use the correct model name
nulib · Jun 14, 2024 · 1f3b7d8 · 1f3b7d8
2 parents 44fdd2d + a62f37c
commit 1f3b7d8
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/app/lib/meadow/search/index.ex b/app/lib/meadow/search/index.ex
@@ -137,8 +137,11 @@ defmodule Meadow.Search.Index do
 
  defp embedding_model_name(model_id) do
  case HTTP.get(["_plugins", "_ml", "models", model_id]) do
- {:ok, %{status_code: 200, body: %{"name" => name}}} -> name
- _ -> nil
+ {:ok, %{status_code: 200, body: %{"name" => name}}} ->
+ Regex.replace(~r[^.+/huggingface/], name, "")
+
+ _ ->
+ nil
  end
  end