mistral_rb 0.1.4 → 0.1.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9a420d04befbc7d822678ac18d5d251e9a3804a4f50413efe6743c05f8781c61
-  data.tar.gz: 572d1b152567b1ac54cd59ab0137eaef286732e288afcbfd79b025ced473b10d
+  metadata.gz: e8226b8266c5e6f6b99abb42bb7703b440221a2687870c65d67e5a9aa93c6e3d
+  data.tar.gz: 6ebc354b74961374e00e254e455e81cfceb06dfda60134554738017f5c2b41c6
 SHA512:
-  metadata.gz: ca18ed1fb53190fe6146c26992e2e84525dcc596189396f883871c85dfb55507bb25c399cd290bcdadf72bd781740b8429533b3a6d05249f9a2ef0fff84551b8
-  data.tar.gz: 4544e24b6fb292b3b489e5969b3334e75ef035c340d8d7118275c43ffc2ca23413feab2cc881d1310c203c567d0f9d84ceb2eadbc2035223e29963f78d99fa75
+  metadata.gz: 60759474220fa3ed9946731c64745938124f0e69a7b4f7e3dbe8bcfee178e3f87efbb5897687b056775af8e0e2a5fd928f67075a39e70ed9d4aeb95d4e6f1a75
+  data.tar.gz: eca253391a1a3c2714c00f0bd7dae098fdb580ac502b1f4d086ffdabd5b26203414598f45dcdb97a234e6791ac82671d0d7b3cc53ad5242c248bb2a8a6851c29

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/response_models.rb

@@ -1,104 +1,106 @@
-class CompletionResponse
-  attr_reader :id, :object, :created, :model, :choices, :usage
-
-  def initialize(response_hash)
-    @id = response_hash["id"]
-    @object = response_hash["object"]
-    @created = response_hash["created"]
-    @model = response_hash["model"]
-    @choices = response_hash["choices"].map { |choice| Choice.new(choice) }
-    @usage = response_hash["usage"]
+module MistralModels
+  class CompletionResponse
+    attr_reader :id, :object, :created, :model, :choices, :usage
+
+    def initialize(response_hash)
+      @id = response_hash["id"]
+      @object = response_hash["object"]
+      @created = response_hash["created"]
+      @model = response_hash["model"]
+      @choices = response_hash["choices"].map { |choice| Choice.new(choice) }
+      @usage = response_hash["usage"]
+    end
   end
-end
 
-class Choice
-  attr_reader :index, :message
+  class Choice
+    attr_reader :index, :message
 
-  def initialize(choice_hash)
-    @index = choice_hash["index"]
-    @message = Message.new(choice_hash["message"])
+    def initialize(choice_hash)
+      @index = choice_hash["index"]
+      @message = Message.new(choice_hash["message"])
+    end
   end
-end
 
-class Message
-  attr_reader :role, :content
+  class Message
+    attr_reader :role, :content
 
-  def initialize(message_hash)
-    @role = message_hash["role"]
-    @content = message_hash["content"]
+    def initialize(message_hash)
+      @role = message_hash["role"]
+      @content = message_hash["content"]
+    end
   end
-end
 
-class EmbeddingResponse
-  attr_reader :id, :object, :data, :model, :usage
+  class EmbeddingResponse
+    attr_reader :id, :object, :data, :model, :usage
 
-  def initialize(response_hash)
-    @id = response_hash["id"]
-    @object = response_hash["object"]
-    @data = response_hash["data"].map { |embedding_data| Embedding.new(embedding_data) }
-    @model = response_hash["model"]
-    @usage = response_hash["usage"]
+    def initialize(response_hash)
+      @id = response_hash["id"]
+      @object = response_hash["object"]
+      @data = response_hash["data"].map { |embedding_data| Embedding.new(embedding_data) }
+      @model = response_hash["model"]
+      @usage = response_hash["usage"]
+    end
   end
-end
 
-class Embedding
-  attr_reader :object, :embedding, :index
+  class Embedding
+    attr_reader :object, :embedding, :index
 
-  def initialize(embedding_hash)
-    @object = embedding_hash["object"]
-    @embedding = embedding_hash["embedding"]
-    @index = embedding_hash["index"]
+    def initialize(embedding_hash)
+      @object = embedding_hash["object"]
+      @embedding = embedding_hash["embedding"]
+      @index = embedding_hash["index"]
+    end
   end
-end
 
-class ModelListResponse
-  attr_reader :object, :data
+  class ModelListResponse
+    attr_reader :object, :data
 
-  def initialize(response_hash)
-    @object = response_hash["object"]
-    @data = response_hash["data"].map { |model_data| Model.new(model_data) }
+    def initialize(response_hash)
+      @object = response_hash["object"]
+      @data = response_hash["data"].map { |model_data| Model.new(model_data) }
+    end
   end
-end
 
-class Model
-  attr_reader :id, :object, :created, :owned_by, :permissions
+  class Model
+    attr_reader :id, :object, :created, :owned_by, :permissions
 
-  def initialize(model_hash)
-    @id = model_hash["id"]
-    @object = model_hash["object"]
-    @created = model_hash["created"]
-    @owned_by = model_hash["owned_by"]
-    @permissions = model_hash["permission"] # This could be further parsed into Permission objects if detailed parsing is required
+    def initialize(model_hash)
+      @id = model_hash["id"]
+      @object = model_hash["object"]
+      @created = model_hash["created"]
+      @owned_by = model_hash["owned_by"]
+      @permissions = model_hash["permission"] # This could be further parsed into Permission objects if detailed parsing is required
+    end
   end
-end
 
-class StreamedCompletionResponse
-  attr_reader :id, :object, :created, :model, :choices
+  class StreamedCompletionResponse
+    attr_reader :id, :object, :created, :model, :choices
 
-  def initialize(response_hash)
-    @id = response_hash["id"]
-    @object = response_hash["object"]
-    @created = response_hash["created"]
-    @model = response_hash["model"]
-    @choices = response_hash["choices"].map { |choice| StreamedChoice.new(choice) }
+    def initialize(response_hash)
+      @id = response_hash["id"]
+      @object = response_hash["object"]
+      @created = response_hash["created"]
+      @model = response_hash["model"]
+      @choices = response_hash["choices"].map { |choice| StreamedChoice.new(choice) }
+    end
   end
-end
 
-class StreamedChoice
-  attr_reader :index, :delta, :finish_reason
+  class StreamedChoice
+    attr_reader :index, :delta, :finish_reason
 
-  def initialize(choice_hash)
-    @index = choice_hash["index"]
-    @delta = Delta.new(choice_hash["delta"]) if choice_hash["delta"]
-    @finish_reason = choice_hash["finish_reason"]
+    def initialize(choice_hash)
+      @index = choice_hash["index"]
+      @delta = Delta.new(choice_hash["delta"]) if choice_hash["delta"]
+      @finish_reason = choice_hash["finish_reason"]
+    end
   end
-end
 
-class Delta
-  attr_reader :role, :content
+  class Delta
+    attr_reader :role, :content
 
-  def initialize(delta_hash)
-    @role = delta_hash["role"]
-    @content = delta_hash["content"]
+    def initialize(delta_hash)
+      @role = delta_hash["role"]
+      @content = delta_hash["content"]
+    end
   end
 end

data/lib/mistral_rb/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module MistralRb
-  VERSION = "0.1.4"
+  VERSION = "0.1.6"
 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"
@@ -41,7 +42,7 @@ class MistralAPI
       # Handle non-streaming response
       response = self.class.post("/chat/completions", body: body, headers: @headers)
       parsed_response = handle_response(response)
-      CompletionResponse.new(parsed_response)
+      MistralModels::CompletionResponse.new(parsed_response)
     end
   end
 
@@ -54,13 +55,13 @@ class MistralAPI
 
     response = self.class.post("/embeddings", body: body, headers: @headers)
     parsed_response = handle_response(response)
-    EmbeddingResponse.new(parsed_response)
+    MistralModels::EmbeddingResponse.new(parsed_response)
   end
 
   def list_available_models
     response = self.class.get("/models", headers: @headers)
     parsed_response = handle_response(response)
-    ModelListResponse.new(parsed_response)
+    MistralModels::ModelListResponse.new(parsed_response)
   end
 
   private
@@ -82,7 +83,7 @@ class MistralAPI
       begin
         # Only parse the JSON content if it's not the end-of-stream indicator
         json_content = JSON.parse(data_content)
-        StreamedCompletionResponse.new(json_content)
+        MistralModels::StreamedCompletionResponse.new(json_content)
       rescue JSON::ParserError => e
         puts "Error parsing JSON: #{e.message}"
       end

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.4
+  version: 0.1.6
 platform: ruby
 authors:
 - Franck Stephane Ndzomga
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-12-26 00:00:00.000000000 Z
+date: 2024-01-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler