only do with_indifferent_access on hash response

OlympiaAI · May 30, 2024 · 86fcf13 · 86fcf13
1 parent b9f5422
commit 86fcf13
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/Gemfile.lock b/Gemfile.lock
@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    open_router (0.3.1)
+    open_router (0.3.2)
       activesupport (>= 6.0)
       dotenv (>= 2)
       faraday (>= 1)

diff --git a/lib/open_router/client.rb b/lib/open_router/client.rb
@@ -44,7 +44,9 @@ def complete(messages, model: "openrouter/auto", providers: [], transforms: [],
       post(path: "/chat/completions", parameters:).tap do |response|
         raise ServerError, response.dig("error", "message") if response.presence&.dig("error", "message").present?
         raise ServerError, "Empty response from OpenRouter. Might be worth retrying once or twice." if stream.blank? && response.blank?
-      end.with_indifferent_access
+
+        return response.with_indifferent_access if response.is_a?(Hash)
+      end
     end
 
     # Fetches the list of available models from the OpenRouter API.

diff --git a/lib/open_router/version.rb b/lib/open_router/version.rb
@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module OpenRouter
-  VERSION = "0.3.1"
+  VERSION = "0.3.2"
 end