tiny_ollama 1.0.0 → 2.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 598de0585c4437aec61f9963b09e6f623d7c3988f039440cfb1c6bd6303b7a0b
-  data.tar.gz: e5b5425dee8e809ba501fa0f8bc6951470191e41afd7d3a1c263d00e6ca238aa
+  metadata.gz: ca4fe9ba0bb0854333e7c1e62b6f0deb0e1b944e5f406086c7148524743521b7
+  data.tar.gz: 47f4a7b71302ea587ad84fd37b9c9ae2cd3d9514ee8b20eb2fe01128f96e42ab
 SHA512:
-  metadata.gz: ead24d28f0395d2431403b83401cd7bd0ac59e7df1861c8691be42932283c2f53612d25843b5382652ce30a255ae0a193250a15184e562ec2cab106f616137aa
-  data.tar.gz: e71614054eeecd5512b77748f69c9dddb3cf882d9e259b0cdce1baf47be372ff3c8993b5054d96977059779b93323d64bec3b90fac53500097fcdc3f32afa53d
+  metadata.gz: dab2b666565a058ff3e48080b9de089316fd0cb5be6b7dc764f6ac6ef3e30de351f7bf2de6fb0de0de14a7a221ca81a3a0e1c713d74b6a851bb74cfc008d9e90
+  data.tar.gz: e765aee90aef074d6b5cffd9b71da0a90fe20b443feabc305909fd739c07192446bcde7fb46bfab9f888489a632f76e15933ac5c9eb59db6dd02e209182ec35f

data/lib/tiny_ollama.rb

@@ -25,30 +25,6 @@ class TinyOllama
     @format = format
   end
 
-  # sends a request to POST /api/generate
-  def generate(prompt)
-    request_body = {
-      model: @model,
-      prompt: prompt,
-      stream: @stream,
-      keep_alive: @keep_alive,
-      options: {
-        num_ctx: @context_size,
-      }.merge(@format ? { format: @format } : {})
-    }.to_json
-
-    uri = URI("http://#{@host}:#{@port}/api/generate")
-    headers = { 'Content-Type' => 'application/json' }
-    response = Net::HTTP.post(uri, request_body, headers)
-
-    # Handle potential errors (e.g., non-200 responses)
-    unless response.is_a?(Net::HTTPSuccess)
-      raise TinyOllamaModelError.new("Ollama API Error: #{response.code} - #{response.body}")
-    end
-
-    JSON.parse(response.body)['response']
-  end
-
   # sends a request to POST /api/chat
   #
   # messages: an array of hashes in the following format:
@@ -73,7 +49,7 @@ class TinyOllama
   #
   # NOTE: the messages parameter needs to include a system message if you want
   # to override the model's default instructions
-  def chat(messages)
+  def prompt(messages)
     request_body = {
       model: @model,
       messages: messages,
@@ -91,11 +67,11 @@ class TinyOllama
 
     # Handle potential errors (e.g., non-200 responses)
     unless response.is_a?(Net::HTTPSuccess)
-      raise TinyOllamaModelError.new("Ollama API Error: #{response.code} - #{response.body}")
+      raise TinyOllamaModelError, "Ollama API Error: #{response.code} - #{response.body}"
     end
 
     JSON.parse(response.body)['message']['content']
   end
 end
 
-class TinyOllamaModelError; end
+class TinyOllamaModelError < StandardError; end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: tiny_ollama
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 2.0.0
 platform: ruby
 authors:
 - Jeff Lunt
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-06-26 00:00:00.000000000 Z
+date: 2024-07-10 00:00:00.000000000 Z
 dependencies: []
 description: a tiny gem for using ollama's generate and chat endpoints
 email: jefflunt@gmail.com