ollama-rb 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: bc1be94d75a4db6787115a0cfbdc7b6aa149418a59612e0012f845512f361ec0
+  data.tar.gz: d1a565383613490c54c45a534e5fff1a9060803423b6dd9146f036eb769c15ef
+SHA512:
+  metadata.gz: 6a09fcb88ec6f2a4e41706fad85ca3def0e75b0a7df87d60fa7c167cccb19abc24163a5dc3d9c1bee1a8a88eca06fa0dbeefa8dd9b79c9215d2db564d28ea5e8
+  data.tar.gz: c8ec22503cb81e3dd376b0775930047d685c0ead3c6d699249449a08544d1de2435b18fb4f2a92d2819cd7f25e320ee2a57d23e82e6386a7760bd0074137de6d

data/.rubocop.yml

@@ -0,0 +1,13 @@
+AllCops:
+  TargetRubyVersion: 3.0
+
+Style/StringLiterals:
+  Enabled: true
+  EnforcedStyle: double_quotes
+
+Style/StringLiteralsInInterpolation:
+  Enabled: true
+  EnforcedStyle: double_quotes
+
+Layout/LineLength:
+  Max: 120

data/CHANGELOG.md

@@ -0,0 +1,5 @@
+## [Unreleased]
+
+## [0.1.0] - 2024-03-19
+
+- Initial release

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2024 songji.zeng
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/README.md

@@ -0,0 +1,301 @@
+# Ollama Ruby Library
+
+The Ollama Ruby library provides the easiest way to integrate your Ruby project with [Ollama](https://github.com/jmorganca/ollama).
+
+## Index
+
+- [Ollama Ruby Library](#ollama-ruby-library)
+  - [Installation](#installation)
+  - [Usage](#usage)
+    - [Create a Client](#create-a-client)
+    - [Generate a chat completion](#generate-a-chat-completion)
+    - [Generate a completion](#generate-a-completion)
+    - [Create a Model](#create-a-model)
+    - [List Local Models](#list-local-models)
+    - [Show Model Information](#show-model-information)
+    - [Copy a Model](#copy-a-model)
+    - [Delete a Model](#delete-a-model)
+    - [Pull a Model](#pull-a-model)
+    - [Push a Model](#push-a-model)
+    - [Generate Embeddings](#generate-embeddings)
+  - [Development](#development)
+  - [Contributing](#contributing)
+  - [License](#license)
+
+## Installation
+
+Install the gem and add to the application's Gemfile by executing:
+
+```sh
+bundle add ollama-rb
+```
+
+If bundler is not being used to manage dependencies, install the gem by executing:
+
+```sh
+gem install ollama-rb
+```
+
+## Usage
+
+### Create a Client
+
+```ruby
+require "ollama"
+
+ollama = Ollama::Client.new
+
+# Specify url, default to http://localhost:11434
+ollama = Ollama::Client.new(base_url: "http://localhost:11434")
+```
+
+### Generate a chat completion
+
+```ruby
+response = ollama.chat.create(
+  model: "llama2",
+  messages: [
+    { role: "user", content: "Why is the sky blue?" }
+  ]
+)
+# => #<Ollama::Response:0x000000011fa1a840...
+response.ok?
+# => true
+response.result
+# =>
+# {"model"=>"llama2",
+#  "created_at"=>"2024-03-20T06:53:18.298807078Z",
+#  "message"=>
+#   {"role"=>"assistant",
+#    "content"=> ...
+```
+
+**Streaming response**
+
+```ruby
+response = ollama.chat.create(
+  model: "llama2",
+  messages: [
+    { role: "user", content: "Why is the sky blue?" }
+  ]
+) do |chunk|
+  puts chunk
+end
+# =>
+#{"model"=>"llama2", "created_at"=>"2024-03-20T06:57:57.513159464Z", "message"=>{"role"=>"assistant", "content"=>"\n"}, "done"=>false}
+#{"model"=>"llama2", "created_at"=>"2024-03-20T06:57:57.616592691Z", "message"=>{"role"=>"assistant", "content"=>"The"}, "done"=>false}
+#{"model"=>"llama2", "created_at"=>"2024-03-20T06:57:57.70737176Z", "message"=>{"role"=>"assistant", "content"=>" sky"}, "done"=>false}
+#{"model"=>"llama2", "created_at"=>"2024-03-20T06:57:57.796324471Z", "message"=>{"role"=>"assistant", "content"=>" appears"}, "done"=>false}
+#{"model"=>"llama2", "created_at"=>"2024-03-20T06:57:57.884097322Z", "message"=>{"role"=>"assistant", "content"=>" blue"}, "done"=>false}
+# ...
+```
+
+### Generate a completion
+
+```ruby
+response = ollama.completion.create(model: "llama2", prompt: "hello!")
+
+response.result
+# =>
+# {"model"=>"llama2",
+#  "created_at"=>"2024-03-20T08:03:27.910169204Z",
+#  "response"=>"Hello there! It's nice to meet you. Is there something I can help you with or would you like to chat?",
+#  "done"=>true,
+#  "context"=>
+#   [518,
+#    25580,
+#    29962,
+#    3532,
+#    ...
+#    13563,
+#    29973],
+#  "total_duration"=>6212545461,
+#  "load_duration"=>2024921059,
+#  "prompt_eval_count"=>22,
+#  "prompt_eval_duration"=>1815255000,
+#  "eval_count"=>27,
+#  "eval_duration"=>2371725000}
+```
+
+**Streaming response**
+
+```ruby
+ollama.completion.create(model: "llama2", prompt: "hello!") do |chunk|
+  puts chunk
+end
+# =>
+# {"model"=>"llama2", "created_at"=>"2024-03-20T08:08:01.196291424Z", "response"=>"Hello", "done"=>false}
+# {"model"=>"llama2", "created_at"=>"2024-03-20T08:08:01.285639365Z", "response"=>"!", "done"=>false}
+# {"model"=>"llama2", "created_at"=>"2024-03-20T08:08:01.3753276Z", "response"=>" It", "done"=>false}
+# {"model"=>"llama2", "created_at"=>"2024-03-20T08:08:01.464252328Z", "response"=>"'", "done"=>false}
+# {"model"=>"llama2", "created_at"=>"2024-03-20T08:08:01.552918803Z", "response"=>"s", "done"=>false}
+# {"model"=>"llama2", "created_at"=>"2024-03-20T08:08:01.641877239Z", "response"=>" nice", "done"=>false}
+# {"model"=>"llama2", "created_at"=>"2024-03-20T08:08:01.730397754Z", "response"=>" to", "done"=>false}
+# {"model"=>"llama2", "created_at"=>"2024-03-20T08:08:01.819209813Z", "response"=>" meet", "done"=>false}
+# {"model"=>"llama2", "created_at"=>"2024-03-20T08:08:01.907875913Z", "response"=>" you", "done"=>false}
+# {"model"=>"llama2", "created_at"=>"2024-03-20T08:08:01.996684973Z", "response"=>".", "done"=>false}
+# {"model"=>"llama2", "created_at"=>"2024-03-20T08:08:02.085516116Z", "response"=>" Is", "done"=>false}
+# {"model"=>"llama2", "created_at"=>"2024-03-20T08:08:02.1781973Z", "response"=>" there", "done"=>false}
+# {"model"=>"llama2", "created_at"=>"2024-03-20T08:08:02.267609408Z", "response"=>" something", "done"=>false}
+# {"model"=>"llama2", "created_at"=>"2024-03-20T08:08:02.357217892Z", "response"=>" I", "done"=>false}
+# {"model"=>"llama2", "created_at"=>"2024-03-20T08:08:02.446981087Z", "response"=>" can", "done"=>false}
+# {"model"=>"llama2", "created_at"=>"2024-03-20T08:08:02.536783282Z", "response"=>" help", "done"=>false}
+# {"model"=>"llama2", "created_at"=>"2024-03-20T08:08:02.645166548Z", "response"=>" you", "done"=>false}
+# {"model"=>"llama2", "created_at"=>"2024-03-20T08:08:02.737494769Z", "response"=>" with", "done"=>false}
+# {"model"=>"llama2", "created_at"=>"2024-03-20T08:08:02.82763751Z", "response"=>" or", "done"=>false}
+# {"model"=>"llama2", "created_at"=>"2024-03-20T08:08:02.917220827Z", "response"=>" would", "done"=>false}
+# {"model"=>"llama2", "created_at"=>"2024-03-20T08:08:03.006731978Z", "response"=>" you", "done"=>false}
+# {"model"=>"llama2", "created_at"=>"2024-03-20T08:08:03.098463277Z", "response"=>" like", "done"=>false}
+# {"model"=>"llama2", "created_at"=>"2024-03-20T08:08:03.18839214Z", "response"=>" to", "done"=>false}
+# {"model"=>"llama2", "created_at"=>"2024-03-20T08:08:03.277780622Z", "response"=>" chat", "done"=>false}
+# {"model"=>"llama2", "created_at"=>"2024-03-20T08:08:03.367252189Z", "response"=>"?", "done"=>false}
+# {"model"=>"llama2", "created_at"=>"2024-03-20T08:08:03.457281303Z", "response"=>"", "done"=>true, "context"=>[518, 25580, 29962, 3532, 14816, 29903, 29958, 5299, 829, 14816, 29903, 6778, 13, 13, 12199, 29991, 518, 29914, 25580, 29962, 13, 10994, 29991, 739, 29915, 29879, 7575, 304, 5870, 366, 29889, 1317, 727, 1554, 306, 508, 1371, 366, 411, 470, 723, 366, 763, 304, 13563, 29973], "total_duration"=>2432818562, "load_duration"=>1470557, "prompt_eval_duration"=>167895000, "eval_count"=>26, "eval_duration"=>2260824000}
+```
+
+### Create a Model
+
+```ruby
+ollama.models.create(name: "mario", modelfile: "FROM llama2\nSYSTEM You are mario from Super Mario Bros.") do |chunk|
+  puts chunk
+end
+# =>
+# {"status"=>"reading model metadata"}
+# {"status"=>"creating system layer"}
+# {"status"=>"using already created layer sha256:8934d96d3f08982e95922b2b7a2c626a1fe873d7c3b06e8e56d7bc0a1fef9246"}
+# {"status"=>"using already created layer sha256:8c17c2ebb0ea011be9981cc3922db8ca8fa61e828c5d3f44cb6ae342bf80460b"}
+# {"status"=>"using already created layer sha256:7c23fb36d80141c4ab8cdbb61ee4790102ebd2bf7aeff414453177d4f2110e5d"}
+# {"status"=>"using already created layer sha256:2e0493f67d0c8c9c68a8aeacdf6a38a2151cb3c4c1d42accf296e19810527988"}
+# {"status"=>"using already created layer sha256:fa304d6750612c207b8705aca35391761f29492534e90b30575e4980d6ca82f6"}
+# {"status"=>"writing layer sha256:1741cf59ce26ff01ac614d31efc700e21e44dd96aed60a7c91ab3f47e440ef94"}
+# {"status"=>"writing layer sha256:786d77d232711e91aafad74df1bacc01e630525d8e83d57a758693725f08d511"}
+# {"status"=>"writing manifest"}
+# {"status"=>"success"}
+```
+
+### List Local Models
+
+```ruby
+response = ollama.models.list
+response.result
+# =>
+# {"models"=>
+#   [{"name"=>"llama2:latest",
+#     "model"=>"llama2:latest",
+#     "modified_at"=>"2024-03-19T10:37:39.212281917Z",
+#     "size"=>3826793677,
+#     "digest"=>"78e26419b4469263f75331927a00a0284ef6544c1975b826b15abdaef17bb962",
+#     "details"=>{"parent_model"=>"", "format"=>"gguf", "family"=>"llama", "families"=>["llama"], "parameter_size"=>"7B", "quantization_level"=>"Q4_0"}},
+#    {"name"=>"mario:latest",
+#     "model"=>"mario:latest",
+#     "modified_at"=>"2024-03-20T08:21:20.316403821Z",
+#     "size"=>3826793787,
+#     "digest"=>"291f46d2fa687dfaff45de96a8cb6e32707bc16ec1e1dfe8d65e9634c34c660c",
+#     "details"=>{"parent_model"=>"", "format"=>"gguf", "family"=>"llama", "families"=>["llama"], "parameter_size"=>"7B", "quantization_level"=>"Q4_0"}}]}
+```
+
+### Show Model Information
+
+```ruby
+response = ollama.models.show("mario")
+response.result
+=>
+{"license"=>
+  "LLAMA 2 COMMUNITY LICENSE AGREEMENT\t\nLlama 2 Version Release Date: July 18, 2023\n\n\"Agreement\" means the terms and conditions for use, reproduction, distribution and \nmodification of the Llama Materials set forth herein.\n\n\"...",
+  "modelfile"=>
+  "# Modelfile generated by \"ollama show\"\n# To build a new Modelfile based on this one, replace the FROM line with:\n# FROM mario:latest\n\nFROM llama2:latest\nTEMPLATE \
+"\"\"[INST] <<SYS>>{{ .System }}<</SYS>>\n\n{{ .Prompt }} [/INST]\n\"\"\"\nSYSTEM \"\"\"You are mario from Super Mario Bros.\"\"\"\nPARAMETER stop \"[INST]\"\nPARAMETER stop
+ \"[/INST]\"\nPARAMETER stop \"<<SYS>>\"\nPARAMETER stop \"<</SYS>>\"",
+ "parameters"=>
+  "stop                           \"[INST]\"\nstop                           \"[/INST]\"\nstop                           \"<<SYS>>\"\nstop                           \"<</SYS
+>>\"",
+ "template"=>"[INST] <<SYS>>{{ .System }}<</SYS>>\n\n{{ .Prompt }} [/INST]\n",
+ "system"=>"You are mario from Super Mario Bros.",
+ "details"=>{"parent_model"=>"llama2:latest", "format"=>"gguf", "family"=>"llama", "families"=>["llama"], "parameter_size"=>"7B", "quantization_level"=>"Q4_0"}}
+```
+
+### Copy a Model
+
+```ruby
+response = ollama.models.copy(source: "llama2")
+# Same as
+response = ollama.models.copy(source: "llama2", destination: "llama2-backup")
+response.ok?
+# => true
+
+response = ollama.models.copy(source: "non-existence")
+response.ok?
+# => false
+response.result
+# => {"error"=>"model 'non-existence' not found"}
+```
+
+### Delete a Model
+
+```ruby
+response = ollama.models.delete("llama2-backup")
+response.ok?
+# => true
+
+response = ollama.models.delete("non-existence")
+response.ok?
+# => false
+response.result
+# => {"error"=>"model 'non-existence' not found"}
+```
+
+### Pull a Model
+
+```ruby
+ollama.models.pull(name: "tinyllama") do |chunk|
+  puts chunk
+end
+# =>
+# {"status"=>"pulling manifest"}
+# {"status"=>"pulling 2af3b81862c6", "digest"=>"sha256:2af3b81862c6be03c769683af18efdadb2c33f60ff32ab6f83e42c043d6c7816", "total"=>637699456, "completed"=>637699456}
+# {"status"=>"pulling af0ddbdaaa26", "digest"=>"sha256:af0ddbdaaa26f30d54d727f9dd944b76bdb926fdaf9a58f63f78c532f57c191f", "total"=>70, "completed"=>70}
+# {"status"=>"pulling c8472cd9daed", "digest"=>"sha256:c8472cd9daed5e7c20aa53689e441e10620a002aacd58686aeac2cb188addb5c", "total"=>31, "completed"=>31}
+# {"status"=>"pulling fa956ab37b8c", "digest"=>"sha256:fa956ab37b8c21152f975a7fcdd095c4fee8754674b21d9b44d710435697a00d", "total"=>98, "completed"=>98}
+# {"status"=>"pulling 6331358be52a", "digest"=>"sha256:6331358be52a6ebc2fd0755a51ad1175734fd17a628ab5ea6897109396245362", "total"=>483, "completed"=>483}
+# {"status"=>"verifying sha256 digest"}
+# {"status"=>"writing manifest"}
+# {"status"=>"removing any unused layers"}
+# {"status"=>"success"}
+```
+
+### Push a Model
+
+You need to create an account at https://ollama.ai and add your Public Key at https://ollama.ai/settings/keys to allow you push models to your namespace.
+
+```ruby
+ollama.models.copy(source: "mario", destination: "your-namespace/mario")
+ollama.models.push(name: "your-namespace/mario") do |chunk|
+  puts chunk
+end
+```
+
+### Generate Embeddings
+
+```ruby
+response = ollama.embeddings.create(model: "llama2", prompt: "Hello!")
+response.result
+# =>
+{"embedding"=>
+  [1.3464512825012207,
+   -1.0983257293701172,
+   ...
+   -2.2046988010406494, 0.3163630962371826] }
+```
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake test` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and the created tag, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/songjiz/ollama-rb.
+
+## License
+
+The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).

data/Rakefile

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+require "bundler/gem_tasks"
+require "minitest/test_task"
+
+Minitest::TestTask.create
+
+require "rubocop/rake_task"
+
+RuboCop::RakeTask.new
+
+task default: %i[test rubocop]

data/lib/ollama/api/blobs.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+module Ollama
+  module API
+    class Blobs
+      def initialize(client:)
+        @client = client
+      end
+
+      def create(digest)
+        @client.post "/api/blobs/#{digest}"
+      end
+
+      def exists?(digest)
+        response = @client.head "/api/blobs/#{digest}"
+        response.ok?
+      end
+    end
+  end
+end

data/lib/ollama/api/chat.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+module Ollama
+  module API
+    class Chat
+      def initialize(client:)
+        @client = client
+      end
+
+      def create(model:, messages:, **options, &block)
+        json = {
+          model: model,
+          messages: messages
+        }.merge(options).compact.transform_keys(&:to_sym)
+
+        json[:stream] = block_given?
+
+        @client.post "/api/chat", json: json, &block
+      end
+    end
+  end
+end

data/lib/ollama/api/completion.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+module Ollama
+  module API
+    class Completion
+      def initialize(client:)
+        @client = client
+      end
+
+      def create(model:, prompt:, images: [], **options, &block)
+        json = {
+          model: model,
+          prompt: prompt,
+          images: images
+        }.merge(options).compact.transform_keys(&:to_sym)
+
+        json[:stream] = block_given?
+
+        @client.post "/api/generate", json: json, &block
+      end
+
+      alias generate create
+    end
+  end
+end

data/lib/ollama/api/embeddings.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module Ollama
+  module API
+    class Embeddings
+      def initialize(client:)
+        @client = client
+      end
+
+      def create(model:, prompt:, **options)
+        @client.post "/api/embeddings", json: { model: model, prompt: prompt }.merge(options).compact
+      end
+    end
+  end
+end

data/lib/ollama/api/models.rb

@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+
+module Ollama
+  module API
+    class Models
+      def initialize(client:)
+        @client = client
+      end
+
+      def list
+        @client.get "/api/tags"
+      end
+
+      def create(name:, modelfile: nil, path: nil, &block)
+        @client.post "/api/create", json: { name: name, modelfile: modelfile, path: path, stream: block_given? }.compact, &block
+      end
+
+      def show(name)
+        @client.post "/api/show", json: { name: name }
+      end
+
+      def copy(source:, destination: nil)
+        @client.post "/api/copy", json: { source: source, destination: destination || "#{source}-backup" }
+      end
+
+      def delete(name)
+        @client.delete "/api/delete", json: { name: name }
+      end
+
+      def pull(name:, insecure: nil, &block)
+        @client.post "/api/pull", json: { name: name, insecure: insecure, stream: block_given? }.compact, &block
+      end
+
+      def push(name:, insecure: nil, &block)
+        @client.post "/api/push", json: { name: name, insecure: insecure, stream: block_given? }.compact, &block
+      end
+    end
+  end
+end

data/lib/ollama/client.rb

@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+require "forwardable"
+require_relative "connection"
+require_relative "api/blobs"
+require_relative "api/chat"
+require_relative "api/completion"
+require_relative "api/embeddings"
+require_relative "api/models"
+
+module Ollama
+  class Client
+    extend Forwardable
+
+    def_delegators :@connection, :post, :get, :delete, :head
+
+    def initialize(base_url: "http://localhost:11434", logger: nil, **options)
+      @connection = Connection.new(base_url, logger: logger, **options)
+    end
+
+    def chat
+      API::Chat.new client: self
+    end
+
+    def completion
+      API::Completion.new client: self
+    end
+
+    def models
+      API::Models.new client: self
+    end
+
+    def blobs
+      API::Blobs.new client: self
+    end
+
+    def embeddings
+      API::Embeddings.new client: self
+    end
+  end
+end

data/lib/ollama/connection.rb

@@ -0,0 +1,84 @@
+# frozen_string_literal: true
+
+require "net/https"
+require "json"
+require "uri"
+require_relative "response"
+
+module Ollama
+  class Connection
+    def initialize(base_url, logger: nil, **options)
+      @base_uri = URI.parse(base_url.to_s)
+      @logger   = logger
+      @options  = options
+    end
+
+    def post(path, json: nil, headers: {}, **options)
+      http    = new_http
+      request = Net::HTTP::Post.new(path)
+      set_request_headers request, default_headers.merge(headers)
+      request.body = JSON.generate(json) if json
+
+      return Response.new(http.request(request)) unless block_given?
+
+      result = []
+      response = http.request(request) do |http_response|
+        # Raises if response is not successful
+        # http_response.value
+        http_response.read_body do |chunk|
+          parsed = JSON.parse(chunk)
+          result << parsed
+          yield parsed
+        end
+      end
+
+      Response.new(response).tap { |resp| resp.result = result }
+    end
+
+    def get(path, params: {}, headers: {}, **options)
+      http    = new_http
+      query   = URI.encode_www_form(params)
+      uri     = URI.parse(path).tap { |uri| uri.query = query if query.length > 0 }
+      request = Net::HTTP::Get.new(uri.to_s)
+      set_request_headers request, default_headers.merge(headers)
+      Response.new http.request(request)
+    end
+
+    def delete(path, json: nil, headers: {}, **options)
+      http    = new_http
+      request = Net::HTTP::Delete.new(path)
+      set_request_headers request, default_headers.merge(headers)
+      request.body = JSON.generate(json) if json
+
+      Response.new http.request(request)
+    end
+
+    def head(path, headers: {}, **options)
+      http    = new_http
+      request = Net::HTTP::Head.new(path)
+      set_request_headers request, default_headers.merge(headers)
+      Response.new http.request(request)
+    end
+
+    private
+      def new_http
+        Net::HTTP.new(@base_uri.host, @base_uri.port).tap do |http|
+          http.set_debug_output @logger if @logger
+        end
+      end
+
+      def default_headers
+        {
+          "User-Agent" => USER_AGENT
+        }
+      end
+
+      def set_request_headers(request, headers)
+        request.tap do
+          headers.each do |key, value|
+            request[key] = value
+          end
+        end
+      end
+  end
+end

data/lib/ollama/response.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+require "forwardable"
+require "json"
+
+module Ollama
+  class Response
+    extend Forwardable
+    def_delegators :@http_response, :code, :body
+
+    def initialize(http_response)
+      @http_response = http_response
+    end
+
+    def ok?
+      code == "200"
+    end
+
+    def error?
+      !ok?
+    end
+
+    def result=(value)
+      @result = value
+    end
+
+    def result
+      @result ||= JSON.parse(body) rescue {}
+    end
+  end
+end

data/lib/ollama/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module Ollama
+  VERSION = "0.1.0"
+end

data/lib/ollama.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+require_relative "ollama/version"
+require_relative "ollama/client"
+
+module Ollama
+  USER_AGENT = "ollama-rb/#{VERSION}"
+  class Error < StandardError; end
+end

metadata

@@ -0,0 +1,61 @@
+--- !ruby/object:Gem::Specification
+name: ollama-rb
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- songji.zeng
+autorequire:
+bindir: exe
+cert_chain: []
+date: 2024-03-20 00:00:00.000000000 Z
+dependencies: []
+description: The Ollama Ruby library provides the easiest way to integrate your Ruby
+  project with Ollama.
+email:
+- songji.zeng@outlook.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".rubocop.yml"
+- CHANGELOG.md
+- LICENSE.txt
+- README.md
+- Rakefile
+- lib/ollama.rb
+- lib/ollama/api/blobs.rb
+- lib/ollama/api/chat.rb
+- lib/ollama/api/completion.rb
+- lib/ollama/api/embeddings.rb
+- lib/ollama/api/models.rb
+- lib/ollama/client.rb
+- lib/ollama/connection.rb
+- lib/ollama/response.rb
+- lib/ollama/version.rb
+homepage: https://github.com/songjiz/ollama-rb
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/songjiz/ollama-rb
+  source_code_uri: https://github.com/songjiz/ollama-rb
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.0.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.5.6
+signing_key:
+specification_version: 4
+summary: Ollama Ruby library
+test_files: []