everyai 0.0.0 → 0.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 84a4a9deb6315f3817a631aa86f93a8c4be163b9b05508383907a32aabb42d89
-  data.tar.gz: 69e82ad21f67ebab33f563e52c5ff8ff8d86940d84ff76681a96107ca3dd3af2
+  metadata.gz: cfb5db5a2396f0ba3be48382a13bf6be48ccdf9baf0dae4dacff56ec91d69ca0
+  data.tar.gz: bc289476d96b29e2034ff4a6522cb5fa5f435f597a69fbd3f38fe320ffc8abbe
 SHA512:
-  metadata.gz: e667fccccf7eeb8b303224946b169ac6fb4d7617de4f0fbf803525469c123646e865c19a2f27483084ed50851dfe35f45bf8ac311952720aea47f7f20703722f
-  data.tar.gz: 4aca7a987e8a214979a3a8f7cec8acd9f331933b993a9f1069f7c388e50ec3a5d7944ed5f19a7849809d59a954b9fe1fb762549a2feea0b4fcfbf1bb95220081
+  metadata.gz: 0f23773f74789802b9891d0ca030209574da59e56a930e421a5935e9fa1897c8bf75477486c19ee004747e6ddf7631d0c9ba90c9c6ff292bada55987fddf7a48
+  data.tar.gz: '082951e08d8bb58b21c6eff2175849d4a2d9fd1ad4c403ea8f1416d4f62881cae4c5ac4144a938bfb9f00da83640311e96734edd36d820b059eaabc8f92fa43e'

data/lib/everyai.rb

@@ -1,18 +1,36 @@
 require "ollama-ai"
 
 class Everyai
+  attr_reader :client
+  def initialize
+    @client = Ollama.new(
+      credentials: { address: 'http://localhost:11434' },
+      opiions: { server_sent_events: true }
+    )
+  end
+
+  def generate(prompt, model)
+    result = client.generate(
+      { model: model,
+        prompt: prompt }
+    )
+    result.map { |r| r['response']}.join
+  rescue Ollama::Errors::RequestError
+    puts "This Ollama model is not installed. Type y to install or any key to continue"
+    answer = gets.chomp.downcase
+    `ollama run #{model}` if answer == "y"
+  end
+
   class << self
     def generate(prompt, model: "llama3.1")
-      client = Ollama.new(
-        credentials: { address: 'http://localhost:11434' },
-        opiions: { server_sent_events: true }
-      )
+      @ai ||= new
+      @ai.generate(prompt, model)
+    end
 
-      result = client.generate(
-        { model: model,
-          prompt: prompt }
-      )
-      result.map { |r| r['response']}.join
+    def ls
+      puts "Listing Ollama AI Models"
+      str = `ollama list`
+      puts str.split("\n")
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: everyai
 version: !ruby/object:Gem::Version
-  version: 0.0.0
+  version: 0.0.1
 platform: ruby
 authors:
 - Enoch Tamulonis