fixes related to ratelimit

fastrepl · Oct 30, 2024 · 71f4535 · 71f4535
1 parent e571ca5
commit 71f4535
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 10 deletions.
diff --git a/core/config/config.exs b/core/config/config.exs
@@ -58,10 +58,10 @@ config :canary, Oban,
   engine: Oban.Engines.Basic,
   queues: [
     default: 10,
-    github_processor: 2,
+    github_processor: 1,
     webpage_processor: 2,
     email: 10,
-    trieve: 50,
+    trieve: 5,
     tinybird: 1
   ],
   repo: Canary.Repo,

diff --git a/core/lib/canary/index/trieve/client.ex b/core/lib/canary/index/trieve/client.ex
@@ -62,7 +62,7 @@ defmodule Canary.Index.Trieve.Actual do
            }
          ) do
       {:ok, %{status: 200, body: _}} -> :ok
-      {:ok, %{status: status, body: error}} when status in 400..499 -> {:error, error}
+      {:ok, %{status: status, body: error}} when status in 400..599 -> {:error, error}
       {:error, error} -> {:error, error}
     end
   end
@@ -75,7 +75,7 @@ defmodule Canary.Index.Trieve.Actual do
            headers: [{"TR-Dataset", tracking_id}]
          ) do
       {:ok, %{status: 204}} -> :ok
-      {:ok, %{status: status, body: error}} when status in 400..499 -> {:error, error}
+      {:ok, %{status: status, body: error}} when status in 400..599 -> {:error, error}
       {:error, error} -> {:error, error}
     end
   end
@@ -94,7 +94,7 @@ defmodule Canary.Index.Trieve.Actual do
     # https://docs.trieve.ai/api-reference/chunk-group/create-or-upsert-group-or-groups
     case client |> Req.post(url: "/chunk_group", json: data) do
       {:ok, %{status: 200, body: _}} -> :ok
-      {:ok, %{status: status, body: error}} when status in 400..499 -> {:error, error}
+      {:ok, %{status: status, body: error}} when status in 400..599 -> {:error, error}
       {:error, error} -> {:error, error}
     end
   end
@@ -107,7 +107,7 @@ defmodule Canary.Index.Trieve.Actual do
            params: [delete_chunks: true]
          ) do
       {:ok, %{status: 204}} -> :ok
-      {:ok, %{status: status, body: error}} when status in 400..499 -> {:error, error}
+      {:ok, %{status: status, body: error}} when status in 400..599 -> {:error, error}
       {:error, error} -> {:error, error}
     end
   end
@@ -185,7 +185,7 @@ defmodule Canary.Index.Trieve.Actual do
         {:ok, %{status: 200}} ->
           {:cont, :ok}
 
-        {:ok, %{status: status, body: error}} when status in 400..499 ->
+        {:ok, %{status: status, body: error}} when status in 400..599 ->
           {:halt, {:error, error}}
 
         {:error, error} ->
@@ -200,7 +200,7 @@ defmodule Canary.Index.Trieve.Actual do
       {:ok, %{status: 200}} ->
         :ok
 
-      {:ok, %{status: status, body: error}} when status in 400..499 ->
+      {:ok, %{status: status, body: error}} when status in 400..599 ->
         {:error, error}
 
       {:error, error} ->
@@ -336,7 +336,7 @@ defmodule Canary.Index.Trieve.Actual do
       {:ok, %{status: 200, body: %{"results" => results}}} ->
         {:ok, results}
 
-      {:ok, %{status: status, body: error}} when status in 400..499 ->
+      {:ok, %{status: status, body: error}} when status in 400..599 ->
         if error["message"] =~ "Should have at least one value for match" do
           {:ok, []}
         else
@@ -365,7 +365,7 @@ defmodule Canary.Index.Trieve.Actual do
           # https://docs.trieve.ai/api-reference/chunk-group/get-chunks-in-group-by-tracking-id
           case client |> Req.get(url: "/chunk_group/tracking_id/#{group_tracking_id}/#{page}") do
             {:ok, %{status: 200, body: result}} -> {:ok, result}
-            {:ok, %{status: status, body: error}} when status in 400..499 -> {:error, error}
+            {:ok, %{status: status, body: error}} when status in 400..599 -> {:error, error}
             {:error, error} -> {:error, error}
           end
         end)