mistral_rb 0.1.4 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9a420d04befbc7d822678ac18d5d251e9a3804a4f50413efe6743c05f8781c61
-  data.tar.gz: 572d1b152567b1ac54cd59ab0137eaef286732e288afcbfd79b025ced473b10d
+  metadata.gz: 1c775af119701a9e79b9aed5e79049ffcb5d3b434d8730d51e857443d1cdeb84
+  data.tar.gz: 43141631a0f888a09e85259832a2d30a8c3c02c8659327ff97bc859317795340
 SHA512:
-  metadata.gz: ca18ed1fb53190fe6146c26992e2e84525dcc596189396f883871c85dfb55507bb25c399cd290bcdadf72bd781740b8429533b3a6d05249f9a2ef0fff84551b8
-  data.tar.gz: 4544e24b6fb292b3b489e5969b3334e75ef035c340d8d7118275c43ffc2ca23413feab2cc881d1310c203c567d0f9d84ceb2eadbc2035223e29963f78d99fa75
+  metadata.gz: 2f8513858962d88312898f29be0d2fb786d0c0e693877e404aca529c5589d0e077b267066f297b5b2b238f77507c1a3eb93e5703d63513f31bccbff386000441
+  data.tar.gz: 155f73296b4de0a124531c3e2c26babba0636b61ae6a39c328b815dd44b47f7124cf64ae46dc1dedfa6fc58feb099925224af1df33bb590a8587913e32d2528f

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.5"
 end

data/lib/mistral_rb.rb

@@ -41,7 +41,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 +54,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 +82,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

metadata

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