mistral_rb 0.1.6 → 0.1.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e8226b8266c5e6f6b99abb42bb7703b440221a2687870c65d67e5a9aa93c6e3d
-  data.tar.gz: 6ebc354b74961374e00e254e455e81cfceb06dfda60134554738017f5c2b41c6
+  metadata.gz: d6a738ce2d5cdd375d86ff8b3573d10f88e75fa50b4320107c19af9764e57f68
+  data.tar.gz: 51be8eb88f3d1c17433cbd26e647abdacd6b5998496a451545bb5f49b7f012b9
 SHA512:
-  metadata.gz: 60759474220fa3ed9946731c64745938124f0e69a7b4f7e3dbe8bcfee178e3f87efbb5897687b056775af8e0e2a5fd928f67075a39e70ed9d4aeb95d4e6f1a75
-  data.tar.gz: eca253391a1a3c2714c00f0bd7dae098fdb580ac502b1f4d086ffdabd5b26203414598f45dcdb97a234e6791ac82671d0d7b3cc53ad5242c248bb2a8a6851c29
+  metadata.gz: c7909af91f1f7f1be58f71f7b22ef98628d34359aa9e599fa7c43b837a6401f08bcf161c4553057bc195b241ca45d4c86720bfa0f5dda71525b9685b4ad24ae3
+  data.tar.gz: a7acf05f8b18515490162f7e0fb7664bdb6aee75b95a1bc3288f12ec36a102c6857dc31e06212c81b6cdeb3de02d3c916a876631e860d9a085bd7f03e5bac813

data/lib/content_extractors/text_content_extractor.rb

@@ -42,7 +42,7 @@ class TxtContentExtractor
       pages = split_into_pages(sanitized_content)
       @page_count = pages.size
 
-      pages
+      [pages, content]
     end
   end
 

data/lib/mistral_rb/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module MistralRb
-  VERSION = "0.1.6"
+  VERSION = "0.1.7"
 end

data/lib/mistral_rb.rb

@@ -20,7 +20,7 @@ class MistralAPI
     self.class.base_uri base_uri
   end
 
-  def create_chat_completion(model:, messages:, temperature: 0.7, top_p: 1, max_tokens: nil, stream: false, safe_mode: false, random_seed: nil)
+  def create_chat_completion(model:, messages:, temperature: 0.7, top_p: 1, max_tokens: nil, stream: false, safe_prompt: false, random_seed: nil)
     body = {
       model: model,
       messages: messages,
@@ -28,7 +28,7 @@ class MistralAPI
       top_p: top_p,
       max_tokens: max_tokens,
       stream: stream,
-      safe_mode: safe_mode,
+      safe_prompt: safe_prompt,
       random_seed: random_seed
     }.compact.to_json
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: mistral_rb
 version: !ruby/object:Gem::Version
-  version: 0.1.6
+  version: 0.1.7
 platform: ruby
 authors:
 - Franck Stephane Ndzomga
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-01-02 00:00:00.000000000 Z
+date: 2024-01-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler