tiny_gemini 1.0.1 → 1.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ab7e00903ada0199cff54253ad7f1461818b6c55947980bc67cde4aa7d641da5
-  data.tar.gz: cfa8b05416999c1b5d5095ae8f06d800ca709e9ad7d93f478f50c52199824894
+  metadata.gz: 1bbd394b0cc2b9b63634b60360e8827c8fe5bb1e482dd2a3b05eb99f7bdef6d0
+  data.tar.gz: da9fe333b0e14e5cfe0167600a23e547e09b99d981ec7205528a657de326d86d
 SHA512:
-  metadata.gz: 955d8e3bc0b12f934bfa44a66022ed345530a5b6f9302eeef858e8599f7bf7ed8ebaec84fba0345427bccf62045f3f075b8be1dbf6dd57ea53dc8b023b89cc85
-  data.tar.gz: 0e7961437e74dc645bb35f88e8fa41930d6e2a9a3ac91583ed08e5e2a0dcdfaeec5bb4f5d15f0d150e08c753294d4ed0bad22038824a91c11aa9ed8ba6fb7ac2
+  metadata.gz: 7ec88cdc3b50604ed4fff7bb61704324d07dbf7d0b2016312e98e8ed440d9a6672fd73253007854b4c2a3aa70ff671d8165d10a89275400838613bbe542b53f0
+  data.tar.gz: 9c6f41d624e1bf930097636c19dc8cb4613ed15255db44b54fe255f9089289e1f397de73c429c1f065fa0786044c1936efd6f236eef7a7e08f2e00d0ad712b91

data/lib/tiny_gemini.rb

@@ -59,6 +59,15 @@ class TinyGemini
   def chat(messages)
     body = { contents: messages }
     body.merge!(system_instruction: { parts: { text: @system_instruction } }) if @system_instruction
+    body.merge!({
+        safetySettings: [
+          {
+              category: "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+              threshold: "BLOCK_NONE"
+          }
+      ]
+    })
+
     request_body = body.to_json
 
     uri = URI("https://#{@host}#{@path}/#{@model}:#{@action}?key=#{@api_key}")
@@ -70,7 +79,17 @@ class TinyGemini
       raise TinyGeminiModelError, "Gemini API Error: #{response.code}\n#{JSON.pretty_generate(response.body)}"
     end
 
-    JSON.parse(response.body)['candidates'].first.dig('content', 'parts').first['text']
+    # response and error handling
+    parsed_response = JSON.parse(response.body)
+    raise(TinyGeminiModelError, "No condidates in Gemini response") unless parsed_response['candidates']
+
+    first_candidate_response = parsed_response['candidates'].first
+    raise(TinyGeminiModelError, "No first candidate response in Gemini response") unless first_candidate_response
+
+    text_response = first_candidate_response&.dig('content', 'parts')&.first['text']
+    raise(TinyGeminiModelError, "Text response is nil or empty: `#{text_response.inspect}`") unless text_response.present?
+
+    text_response
   end
 end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: tiny_gemini
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 1.0.2
 platform: ruby
 authors:
 - Jeff Lunt