smollama 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f9f331024c63381cad8781105f25f9c86f8d1745f89ee4ec6ce230e65646500b
-  data.tar.gz: 14b790b0158ac0cd54a9520e37d68eab471cbc5951e9a38274bbc3da51d9df93
+  metadata.gz: 5eb20d070bc8386684a46b046d3b31bcb91c48c444bffb9366c9fdc9e8c1a34c
+  data.tar.gz: 9d0df8aa3966ae405d9df737fd2cc82799eeee7f2c8bb983f4c583f28f578fd5
 SHA512:
-  metadata.gz: 7f9e09789dcabfc5282332fda7742f7a054547e9c0df57b349282db5ab7d4dba7ead148da52b0a5345aa921b94d07d611feec48f34e20672473933776c6cfff0
-  data.tar.gz: e81e862f8534d92d9b929dfe6ee0901b3f39030cb55610572d4a2dbf1f6f5e4c29a90866b9fd26c74226b494795a612ce6df5adf4fb49458d806ab3eb9b9706b
+  metadata.gz: 67f0073ebf5dc0bda3b5851ab85c9354dbb1e63fa722f3120230974045b7b4a930d6fc0b0aebc692aff6267853471a6fa2c9ef715e68bf55d62637ce79f0f769
+  data.tar.gz: 0efde51d756f704a4b0570728a836affbf22676bb22c8490e41a545d5c2050e7e5fc545da5cb5fe5c5cfe336f6d6e2b328c112e80f51033bc33fb716846ac7b9

data/README.md

@@ -85,6 +85,47 @@ special_client = Smollama::Client.new(model: 'llama2')
 response = special_client.ask("Hello!")
 ```
 
+### Vision - Chat with Images
+
+Vision models can accept images alongside text to describe, classify, and answer questions about what they see.
+
+```ruby
+# Use a vision-capable model
+client = Smollama::Client.new(model: 'gemma3')
+
+# With a local file path
+response = client.chat(
+  "What is in this image?",
+  images: ["./cat.jpg"]
+)
+puts response[:content]
+
+# With a URL
+response = client.chat(
+  "Describe this image",
+  images: ["https://example.com/image.jpg"]
+)
+
+# With multiple images
+response = client.chat(
+  "Compare these images",
+  images: ["./image1.jpg", "./image2.jpg"]
+)
+
+# With base64 encoded image data
+img_data = Base64.strict_encode64(File.read("./image.jpg"))
+response = client.chat(
+  "What do you see?",
+  images: [img_data]
+)
+```
+
+The `images` parameter accepts:
+- File paths (e.g., `"./image.jpg"`)
+- URLs (e.g., `"https://example.com/image.jpg"`)
+- Base64 encoded strings
+- An array of any combination of the above
+
 ### Server Health Check
 
 ```ruby

data/lib/smollama/client.rb

@@ -1,5 +1,6 @@
 require 'excon'
 require 'json'
+require 'base64'
 
 module Smollama
   class Client
@@ -33,8 +34,8 @@ module Smollama
     end
 
     # Main chat method with configurable parameters
-    def chat(message, temperature: nil, top_p: nil, max_tokens: nil, stream: false)
-      messages = build_messages(message)
+    def chat(message, temperature: nil, top_p: nil, max_tokens: nil, stream: false, images: nil)
+      messages = build_messages(message, images: images)
 
       payload = {
         model: @model,
@@ -105,8 +106,8 @@ module Smollama
 
     private
 
-    def build_messages(input)
-      case input
+    def build_messages(input, images: nil)
+      messages = case input
       when String
         [{ role: 'user', content: input }]
       when Hash
@@ -116,6 +117,35 @@ module Smollama
       else
         raise "Invalid message format"
       end
+
+      # Add images to the last user message if provided
+      if images && !images.empty?
+        encoded_images = encode_images(images)
+        messages.last[:images] = encoded_images
+      end
+
+      messages
+    end
+
+    def encode_images(images)
+      Array(images).map do |image|
+        case image
+        when String
+          if image.start_with?('http://', 'https://')
+            # URL - Ollama can handle URLs directly, but we'll fetch and encode
+            require 'open-uri'
+            Base64.strict_encode64(URI.open(image).read)
+          elsif File.exist?(image)
+            # File path
+            Base64.strict_encode64(File.read(image))
+          else
+            # Assume it's already base64 encoded
+            image
+          end
+        else
+          raise "Invalid image format: #{image.class}"
+        end
+      end
     end
 
     def send_request(payload)

data/lib/smollama/version.rb

@@ -1,3 +1,3 @@
 module Smollama
-  VERSION = "0.1.0"
-end
+  VERSION = "0.2.0"
+end

data/smollama.gemspec

@@ -25,6 +25,8 @@ Gem::Specification.new do |spec|
 
   # Runtime dependencies
   spec.add_dependency "excon", "~> 0.100"
+  spec.add_dependency "base64"
+  spec.add_dependency "logger"
 
   # Development dependencies
   spec.add_development_dependency "bundler", "~> 2.0"

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: smollama
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - makevoid
 bindir: exe
 cert_chain: []
-date: 2025-08-29 00:00:00.000000000 Z
+date: 2025-11-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: excon
@@ -23,6 +23,34 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '0.100'
+- !ruby/object:Gem::Dependency
+  name: base64
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: logger
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement