foobara-ollama-api 0.0.4 → 1.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6cf7fb0a1be355ba8ab6aa6f841a06743fc835f41c58070bd918f5b05ceeecc2
-  data.tar.gz: ad0dbd6b18b7517887e691d352820b920a79c57a9734d696d5dbf1a6b32f9921
+  metadata.gz: ee43f55f3a81ee1c728084bdb60c9ef4cad53bbea1e955d65c7f7367f08fc964
+  data.tar.gz: 5beae1213df18cba068631697a61946e7ec1536fe5f0f10f4d8e251707e887ce
 SHA512:
-  metadata.gz: a4d2ff4007bee4b9b7a9a4d97b108d21d8b93d6f61530a13bb8e00987048f61755ffb395b3a4124cb79b79b0154dce45e4a22f453aab5b2256e4ac03d5f2a42e
-  data.tar.gz: 3b916be0b92c2b9857b89ed61b3da721c6263be9f8e5c26d9939c11a30857fb7553b9c75b089925fa9c2e530959397afd05726287591a569357fc3ea84a3c610
+  metadata.gz: eeb814ee8d48bfc49dda34ed149bbfaca26d5ba37118e024b5456bc589c53d197ed0203654fb69ab254edb029da0075ab2209ec9ba59fe5f12110aa3a99ee392
+  data.tar.gz: a34c800318aacdfbfd77c6a3f29fb2d0079cfd906d0fa5f30ed51b6b8c05396adb6d49eddb5829ece72d8eea52b19cb38132a529dfcbd49772b1eb739eada04f

data/CHANGELOG.md

@@ -1,3 +1,11 @@
+## [1.0.0] - 2025-07-09
+
+- Add OllamaApi.default_llm_model
+
+## [0.0.5] - 2025-06-27
+
+- Add temperature input to GenerateChatCompletion
+
 ## [0.0.4] - 2025-03-01
 
 - Rename :model to :model_enum to avoid collision with Model in filesystems that are case-insensitive

data/lib/foobara/ollama_api.rb

@@ -7,3 +7,15 @@ require "foobara/http_api_command"
 require_relative "../../src/foobara/ai/ollama_api"
 
 Foobara::Util.require_directory "#{__dir__}/../../src"
+
+module Foobara
+  module Ai
+    module OllamaApi
+      class << self
+        def reset_all
+          remove_instance_variable(:@default_llm_model) if instance_variable_defined?(:@default_llm_model)
+        end
+      end
+    end
+  end
+end

data/src/foobara/ai/ollama_api/generate_chat_completion.rb

@@ -7,6 +7,9 @@ module Foobara
         inputs do
           model :model_enum, :required
           messages [Types::Message], :required
+          options do
+            temperature :float
+          end
         end
 
         result Types::ChatCompletion
@@ -16,6 +19,16 @@ module Foobara
         http_timeout 600
 
         def build_request_body
+          inputs = {
+            model:,
+            messages:,
+            stream: false
+          }
+
+          if options && !options.empty?
+            inputs[:options] = options
+          end
+
           self.request_body = inputs.merge(stream: false)
         end
       end

data/src/foobara/ai/ollama_api.rb

@@ -4,6 +4,30 @@ module Foobara
 
     module OllamaApi
       foobara_domain!
+
+      class << self
+        def default_llm_model
+          return @default_llm_model if @default_llm_model
+
+          require_relative "ollama_api/types/model_enum"
+
+          ["qwen3", "deepseek-r1", "llama3.3", "gemma3"].each do |model|
+            candidates = Types::ModelEnum.all_values.select do |model_name|
+              model_name.start_with?(model)
+            end
+
+            next if candidates.empty?
+
+            @default_llm_model = candidates.max_by do |candidate|
+              candidate[/:([\d.]+)([mkbt])\z/, 1].to_f
+            end
+
+            return @default_llm_model
+          end
+
+          Types::ModelEnum.all_values.first
+        end
+      end
     end
   end
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: foobara-ollama-api
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 1.0.0
 platform: ruby
 authors:
 - Miles Georgi
 bindir: bin
 cert_chain: []
-date: 2025-03-01 00:00:00.000000000 Z
+date: 1980-01-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: foobara-http-api-command
@@ -61,14 +61,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 3.4.2
+      version: 3.4.3
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.6.3
+rubygems_version: 3.6.9
 specification_version: 4
 summary: Ruby Ollama API client using foobara commands
 test_files: []