mistral_rb 0.1.5 → 0.1.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1c775af119701a9e79b9aed5e79049ffcb5d3b434d8730d51e857443d1cdeb84
-  data.tar.gz: 43141631a0f888a09e85259832a2d30a8c3c02c8659327ff97bc859317795340
+  metadata.gz: d6a738ce2d5cdd375d86ff8b3573d10f88e75fa50b4320107c19af9764e57f68
+  data.tar.gz: 51be8eb88f3d1c17433cbd26e647abdacd6b5998496a451545bb5f49b7f012b9
 SHA512:
-  metadata.gz: 2f8513858962d88312898f29be0d2fb786d0c0e693877e404aca529c5589d0e077b267066f297b5b2b238f77507c1a3eb93e5703d63513f31bccbff386000441
-  data.tar.gz: 155f73296b4de0a124531c3e2c26babba0636b61ae6a39c328b815dd44b47f7124cf64ae46dc1dedfa6fc58feb099925224af1df33bb590a8587913e32d2528f
+  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/embedding_engines/mistral_embeddings.rb

@@ -9,6 +9,7 @@ class MistralEmbeddingCreator
     @chunker = chunker
     @model = model
     @api_key = api_key || ENV['MISTRAL_API_KEY']
+    raise 'API key not found. Please set the MISTRAL_API_KEY environment variable.' if api_key.nil? || api_key.empty?
 
     if @api_key
       @llm = MistralAPI.new(api_key: @api_key)

data/lib/embedding_engines/openai_embeddings.rb

@@ -9,6 +9,7 @@ class OpenaiEmbeddingCreator
     @chunker = chunker
     @model = model
     @api_key = api_key || ENV['OPENAI_API_KEY']
+    raise 'API key not found. Please set the OPENAI_API_KEY environment variable.' if api_key.nil? || api_key.empty?
 
     if @api_key
       @llm = OpenAI::Client.new(access_token: @api_key)

data/lib/mistral_rb/version.rb

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

data/lib/mistral_rb.rb

@@ -12,6 +12,7 @@ class MistralAPI
   include HTTParty
 
   def initialize(api_key: ENV["MISTRAL_API_KEY"], base_uri: "https://api.mistral.ai/v1")
+    raise 'API key not found. Please set the MISTRAL_API_KEY environment variable.' if api_key.nil? || api_key.empty?
     @headers = {
       "Authorization" => "Bearer #{api_key}",
       "Content-Type" => "application/json"
@@ -19,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,
@@ -27,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
 

data/lib/vector_stores/pinecone.rb

@@ -7,6 +7,8 @@ Dotenv.load()
 class PineconeService
   attr_reader :index
   def initialize(pinecone_key: ENV['PINECONE_API_KEY'], pinecone_env: ENV['PINECONE_ENV'], index_name:)
+    raise 'API key not found. Please set the PINECONE_API_KEY environment variable.' if pinecone_key.nil? || pinecone_key.empty?
+    raise 'ENV not found. Please set the PINECONE_ENV environment variable.' if pinecone_env.nil? || pinecone_env.empty?
     @pinecone_key = pinecone_key
     @pinecone_env = pinecone_env
     @index_name = index_name

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: mistral_rb
 version: !ruby/object:Gem::Version
-  version: 0.1.5
+  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