abliteration 0.1.0 → 0.1.2

data/lib/abliteration/models/author.rb

@@ -15,6 +15,9 @@ require 'time'
 
 module Abliteration
   class Author < ApiModelBase
+    # Same value as `author`; added for consistency with the other single-entity payloads.
+    attr_accessor :id
+
     attr_accessor :author
 
     attr_accessor :total_models
@@ -28,6 +31,7 @@ module Abliteration
     # Attribute mapping from ruby-style variable name to JSON key.
     def self.attribute_map
       {
+        :'id' => :'id',
         :'author' => :'author',
         :'total_models' => :'total_models',
         :'total_downloads' => :'total_downloads',
@@ -49,6 +53,7 @@ module Abliteration
     # Attribute type mapping.
     def self.openapi_types
       {
+        :'id' => :'String',
         :'author' => :'String',
         :'total_models' => :'Integer',
         :'total_downloads' => :'Integer',
@@ -80,6 +85,10 @@ module Abliteration
         h[k.to_sym] = v
       }
 
+      if attributes.key?(:'id')
+        self.id = attributes[:'id']
+      end
+
       if attributes.key?(:'author')
         self.author = attributes[:'author']
       end
@@ -121,6 +130,7 @@ module Abliteration
     def ==(o)
       return true if self.equal?(o)
       self.class == o.class &&
+          id == o.id &&
           author == o.author &&
           total_models == o.total_models &&
           total_downloads == o.total_downloads &&
@@ -137,7 +147,7 @@ module Abliteration
     # Calculates hash code according to all attributes.
     # @return [Integer] Hash code
     def hash
-      [author, total_models, total_downloads, total_likes, first_seen_at].hash
+      [id, author, total_models, total_downloads, total_likes, first_seen_at].hash
     end
 
     # Builds the object from hash

data/lib/abliteration/models/author_detail.rb

@@ -15,6 +15,9 @@ require 'time'
 
 module Abliteration
   class AuthorDetail < ApiModelBase
+    # Same value as `author`; added for consistency with the other single-entity payloads.
+    attr_accessor :id
+
     attr_accessor :author
 
     attr_accessor :total_models
@@ -30,6 +33,7 @@ module Abliteration
     # Attribute mapping from ruby-style variable name to JSON key.
     def self.attribute_map
       {
+        :'id' => :'id',
         :'author' => :'author',
         :'total_models' => :'total_models',
         :'total_downloads' => :'total_downloads',
@@ -52,6 +56,7 @@ module Abliteration
     # Attribute type mapping.
     def self.openapi_types
       {
+        :'id' => :'String',
         :'author' => :'String',
         :'total_models' => :'Integer',
         :'total_downloads' => :'Integer',
@@ -90,6 +95,10 @@ module Abliteration
         h[k.to_sym] = v
       }
 
+      if attributes.key?(:'id')
+        self.id = attributes[:'id']
+      end
+
       if attributes.key?(:'author')
         self.author = attributes[:'author']
       end
@@ -137,6 +146,7 @@ module Abliteration
     def ==(o)
       return true if self.equal?(o)
       self.class == o.class &&
+          id == o.id &&
           author == o.author &&
           total_models == o.total_models &&
           total_downloads == o.total_downloads &&
@@ -154,7 +164,7 @@ module Abliteration
     # Calculates hash code according to all attributes.
     # @return [Integer] Hash code
     def hash
-      [author, total_models, total_downloads, total_likes, first_seen_at, models].hash
+      [id, author, total_models, total_downloads, total_likes, first_seen_at, models].hash
     end
 
     # Builds the object from hash

data/lib/abliteration/models/dataset.rb

@@ -25,16 +25,44 @@ module Abliteration
 
     attr_accessor :license
 
+    attr_accessor :language
+
+    # Recent-window downloads reported by the Hub.
     attr_accessor :downloads
 
+    # All-time downloads reported by the Hub.
     attr_accessor :downloads_all
 
     attr_accessor :likes
 
+    attr_accessor :trending_score
+
+    # Server returns 0/1 (SQLite bool convention), not a JSON boolean.
     attr_accessor :gated
 
+    attr_accessor :main_size_bytes
+
+    # Human-formatted size (e.g. \"90.4 MB\").
+    attr_accessor :size_display
+
+    attr_accessor :age_days
+
+    # Server returns 0/1, not a JSON boolean.
+    attr_accessor :is_trending
+
+    # Server returns 0/1, not a JSON boolean.
+    attr_accessor :just_added
+
+    attr_accessor :description
+
+    attr_accessor :tags
+
+    attr_accessor :hf_url
+
     attr_accessor :created_at
 
+    attr_accessor :last_modified_at
+
     attr_accessor :first_seen_at
 
     # Attribute mapping from ruby-style variable name to JSON key.
@@ -45,11 +73,22 @@ module Abliteration
         :'slug' => :'slug',
         :'category' => :'category',
         :'license' => :'license',
+        :'language' => :'language',
         :'downloads' => :'downloads',
         :'downloads_all' => :'downloads_all',
         :'likes' => :'likes',
+        :'trending_score' => :'trending_score',
         :'gated' => :'gated',
+        :'main_size_bytes' => :'main_size_bytes',
+        :'size_display' => :'size_display',
+        :'age_days' => :'age_days',
+        :'is_trending' => :'is_trending',
+        :'just_added' => :'just_added',
+        :'description' => :'description',
+        :'tags' => :'tags',
+        :'hf_url' => :'hf_url',
         :'created_at' => :'created_at',
+        :'last_modified_at' => :'last_modified_at',
         :'first_seen_at' => :'first_seen_at'
       }
     end
@@ -72,11 +111,22 @@ module Abliteration
         :'slug' => :'String',
         :'category' => :'String',
         :'license' => :'String',
+        :'language' => :'String',
         :'downloads' => :'Integer',
         :'downloads_all' => :'Integer',
         :'likes' => :'Integer',
+        :'trending_score' => :'Integer',
         :'gated' => :'Integer',
+        :'main_size_bytes' => :'Integer',
+        :'size_display' => :'String',
+        :'age_days' => :'Integer',
+        :'is_trending' => :'Integer',
+        :'just_added' => :'Integer',
+        :'description' => :'String',
+        :'tags' => :'Array<String>',
+        :'hf_url' => :'String',
         :'created_at' => :'String',
+        :'last_modified_at' => :'String',
         :'first_seen_at' => :'String'
       }
     end
@@ -88,11 +138,21 @@ module Abliteration
         :'slug',
         :'category',
         :'license',
+        :'language',
         :'downloads',
         :'downloads_all',
         :'likes',
+        :'trending_score',
         :'gated',
+        :'main_size_bytes',
+        :'size_display',
+        :'age_days',
+        :'is_trending',
+        :'just_added',
+        :'description',
+        :'hf_url',
         :'created_at',
+        :'last_modified_at',
         :'first_seen_at'
       ])
     end
@@ -133,6 +193,10 @@ module Abliteration
         self.license = attributes[:'license']
       end
 
+      if attributes.key?(:'language')
+        self.language = attributes[:'language']
+      end
+
       if attributes.key?(:'downloads')
         self.downloads = attributes[:'downloads']
       end
@@ -145,14 +209,56 @@ module Abliteration
         self.likes = attributes[:'likes']
       end
 
+      if attributes.key?(:'trending_score')
+        self.trending_score = attributes[:'trending_score']
+      end
+
       if attributes.key?(:'gated')
         self.gated = attributes[:'gated']
       end
 
+      if attributes.key?(:'main_size_bytes')
+        self.main_size_bytes = attributes[:'main_size_bytes']
+      end
+
+      if attributes.key?(:'size_display')
+        self.size_display = attributes[:'size_display']
+      end
+
+      if attributes.key?(:'age_days')
+        self.age_days = attributes[:'age_days']
+      end
+
+      if attributes.key?(:'is_trending')
+        self.is_trending = attributes[:'is_trending']
+      end
+
+      if attributes.key?(:'just_added')
+        self.just_added = attributes[:'just_added']
+      end
+
+      if attributes.key?(:'description')
+        self.description = attributes[:'description']
+      end
+
+      if attributes.key?(:'tags')
+        if (value = attributes[:'tags']).is_a?(Array)
+          self.tags = value
+        end
+      end
+
+      if attributes.key?(:'hf_url')
+        self.hf_url = attributes[:'hf_url']
+      end
+
       if attributes.key?(:'created_at')
         self.created_at = attributes[:'created_at']
       end
 
+      if attributes.key?(:'last_modified_at')
+        self.last_modified_at = attributes[:'last_modified_at']
+      end
+
       if attributes.key?(:'first_seen_at')
         self.first_seen_at = attributes[:'first_seen_at']
       end
@@ -183,11 +289,22 @@ module Abliteration
           slug == o.slug &&
           category == o.category &&
           license == o.license &&
+          language == o.language &&
           downloads == o.downloads &&
           downloads_all == o.downloads_all &&
           likes == o.likes &&
+          trending_score == o.trending_score &&
           gated == o.gated &&
+          main_size_bytes == o.main_size_bytes &&
+          size_display == o.size_display &&
+          age_days == o.age_days &&
+          is_trending == o.is_trending &&
+          just_added == o.just_added &&
+          description == o.description &&
+          tags == o.tags &&
+          hf_url == o.hf_url &&
           created_at == o.created_at &&
+          last_modified_at == o.last_modified_at &&
           first_seen_at == o.first_seen_at
     end
 
@@ -200,7 +317,7 @@ module Abliteration
     # Calculates hash code according to all attributes.
     # @return [Integer] Hash code
     def hash
-      [id, author, slug, category, license, downloads, downloads_all, likes, gated, created_at, first_seen_at].hash
+      [id, author, slug, category, license, language, downloads, downloads_all, likes, trending_score, gated, main_size_bytes, size_display, age_days, is_trending, just_added, description, tags, hf_url, created_at, last_modified_at, first_seen_at].hash
     end
 
     # Builds the object from hash

data/lib/abliteration/models/dataset_detail.rb

@@ -14,10 +14,58 @@ require 'date'
 require 'time'
 
 module Abliteration
-  # The full dataset record, plus files and the passport when one exists.
   class DatasetDetail < ApiModelBase
     attr_accessor :id
 
+    attr_accessor :author
+
+    attr_accessor :slug
+
+    attr_accessor :category
+
+    attr_accessor :license
+
+    attr_accessor :language
+
+    # Recent-window downloads reported by the Hub.
+    attr_accessor :downloads
+
+    # All-time downloads reported by the Hub.
+    attr_accessor :downloads_all
+
+    attr_accessor :likes
+
+    attr_accessor :trending_score
+
+    # Server returns 0/1 (SQLite bool convention), not a JSON boolean.
+    attr_accessor :gated
+
+    attr_accessor :main_size_bytes
+
+    # Human-formatted size (e.g. \"90.4 MB\").
+    attr_accessor :size_display
+
+    attr_accessor :age_days
+
+    # Server returns 0/1, not a JSON boolean.
+    attr_accessor :is_trending
+
+    # Server returns 0/1, not a JSON boolean.
+    attr_accessor :just_added
+
+    attr_accessor :description
+
+    attr_accessor :tags
+
+    attr_accessor :hf_url
+
+    attr_accessor :created_at
+
+    attr_accessor :last_modified_at
+
+    attr_accessor :first_seen_at
+
+    # Per-file entries: path, type, size_bytes, is_lfs.
     attr_accessor :files
 
     attr_accessor :passport
@@ -26,6 +74,27 @@ module Abliteration
     def self.attribute_map
       {
         :'id' => :'id',
+        :'author' => :'author',
+        :'slug' => :'slug',
+        :'category' => :'category',
+        :'license' => :'license',
+        :'language' => :'language',
+        :'downloads' => :'downloads',
+        :'downloads_all' => :'downloads_all',
+        :'likes' => :'likes',
+        :'trending_score' => :'trending_score',
+        :'gated' => :'gated',
+        :'main_size_bytes' => :'main_size_bytes',
+        :'size_display' => :'size_display',
+        :'age_days' => :'age_days',
+        :'is_trending' => :'is_trending',
+        :'just_added' => :'just_added',
+        :'description' => :'description',
+        :'tags' => :'tags',
+        :'hf_url' => :'hf_url',
+        :'created_at' => :'created_at',
+        :'last_modified_at' => :'last_modified_at',
+        :'first_seen_at' => :'first_seen_at',
         :'files' => :'files',
         :'passport' => :'passport'
       }
@@ -45,6 +114,27 @@ module Abliteration
     def self.openapi_types
       {
         :'id' => :'String',
+        :'author' => :'String',
+        :'slug' => :'String',
+        :'category' => :'String',
+        :'license' => :'String',
+        :'language' => :'String',
+        :'downloads' => :'Integer',
+        :'downloads_all' => :'Integer',
+        :'likes' => :'Integer',
+        :'trending_score' => :'Integer',
+        :'gated' => :'Integer',
+        :'main_size_bytes' => :'Integer',
+        :'size_display' => :'String',
+        :'age_days' => :'Integer',
+        :'is_trending' => :'Integer',
+        :'just_added' => :'Integer',
+        :'description' => :'String',
+        :'tags' => :'Array<String>',
+        :'hf_url' => :'String',
+        :'created_at' => :'String',
+        :'last_modified_at' => :'String',
+        :'first_seen_at' => :'String',
         :'files' => :'Array<Object>',
         :'passport' => :'Passport'
       }
@@ -57,6 +147,13 @@ module Abliteration
       ])
     end
 
+    # List of class defined in allOf (OpenAPI v3)
+    def self.openapi_all_of
+      [
+      :'Dataset'
+      ]
+    end
+
     # Initializes the object
     # @param [Hash] attributes Model attributes in the form of hash
     def initialize(attributes = {})
@@ -77,6 +174,92 @@ module Abliteration
         self.id = attributes[:'id']
       end
 
+      if attributes.key?(:'author')
+        self.author = attributes[:'author']
+      end
+
+      if attributes.key?(:'slug')
+        self.slug = attributes[:'slug']
+      end
+
+      if attributes.key?(:'category')
+        self.category = attributes[:'category']
+      end
+
+      if attributes.key?(:'license')
+        self.license = attributes[:'license']
+      end
+
+      if attributes.key?(:'language')
+        self.language = attributes[:'language']
+      end
+
+      if attributes.key?(:'downloads')
+        self.downloads = attributes[:'downloads']
+      end
+
+      if attributes.key?(:'downloads_all')
+        self.downloads_all = attributes[:'downloads_all']
+      end
+
+      if attributes.key?(:'likes')
+        self.likes = attributes[:'likes']
+      end
+
+      if attributes.key?(:'trending_score')
+        self.trending_score = attributes[:'trending_score']
+      end
+
+      if attributes.key?(:'gated')
+        self.gated = attributes[:'gated']
+      end
+
+      if attributes.key?(:'main_size_bytes')
+        self.main_size_bytes = attributes[:'main_size_bytes']
+      end
+
+      if attributes.key?(:'size_display')
+        self.size_display = attributes[:'size_display']
+      end
+
+      if attributes.key?(:'age_days')
+        self.age_days = attributes[:'age_days']
+      end
+
+      if attributes.key?(:'is_trending')
+        self.is_trending = attributes[:'is_trending']
+      end
+
+      if attributes.key?(:'just_added')
+        self.just_added = attributes[:'just_added']
+      end
+
+      if attributes.key?(:'description')
+        self.description = attributes[:'description']
+      end
+
+      if attributes.key?(:'tags')
+        if (value = attributes[:'tags']).is_a?(Array)
+          self.tags = value
+        end
+      end
+
+      if attributes.key?(:'hf_url')
+        self.hf_url = attributes[:'hf_url']
+      end
+
+      if attributes.key?(:'created_at')
+        self.created_at = attributes[:'created_at']
+      end
+
+      if attributes.key?(:'last_modified_at')
+        self.last_modified_at = attributes[:'last_modified_at']
+      end
+
+      if attributes.key?(:'first_seen_at')
+        self.first_seen_at = attributes[:'first_seen_at']
+      end
+
       if attributes.key?(:'files')
         if (value = attributes[:'files']).is_a?(Array)
           self.files = value
@@ -109,6 +292,27 @@ module Abliteration
       return true if self.equal?(o)
       self.class == o.class &&
           id == o.id &&
+          author == o.author &&
+          slug == o.slug &&
+          category == o.category &&
+          license == o.license &&
+          language == o.language &&
+          downloads == o.downloads &&
+          downloads_all == o.downloads_all &&
+          likes == o.likes &&
+          trending_score == o.trending_score &&
+          gated == o.gated &&
+          main_size_bytes == o.main_size_bytes &&
+          size_display == o.size_display &&
+          age_days == o.age_days &&
+          is_trending == o.is_trending &&
+          just_added == o.just_added &&
+          description == o.description &&
+          tags == o.tags &&
+          hf_url == o.hf_url &&
+          created_at == o.created_at &&
+          last_modified_at == o.last_modified_at &&
+          first_seen_at == o.first_seen_at &&
           files == o.files &&
           passport == o.passport
     end
@@ -122,7 +326,7 @@ module Abliteration
     # Calculates hash code according to all attributes.
     # @return [Integer] Hash code
     def hash
-      [id, files, passport].hash
+      [id, author, slug, category, license, language, downloads, downloads_all, likes, trending_score, gated, main_size_bytes, size_display, age_days, is_trending, just_added, description, tags, hf_url, created_at, last_modified_at, first_seen_at, files, passport].hash
     end
 
     # Builds the object from hash