foobara-ai 0.0.12 → 0.0.14

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a43321389b2cc813a204d4f7340b75c8a7c7bdd131220fae16682602258caab9
-  data.tar.gz: f5cf356dc3f0411bf0e2ffb50772828ed3cd8f729552ca16d385c55f01a00e83
+  metadata.gz: 9c915b80c804ceddc6311400052b2a7cce0011bfcfda4d80dc200f63e50382e5
+  data.tar.gz: 3e50a758f4b65631c5c535fcce51aca9013170cdcd856760c956bb3605b1acff
 SHA512:
-  metadata.gz: 672cac03fe0c5b0ac8f2dfa2182f37ebdd1399a71721994f2699a9f54cc3a75522cfa8dcdea03442569fdecd4df7cc6b1e040fda1d62720ddd8c8fd1ccfa2c19
-  data.tar.gz: 16364632be158b69c7ae22d48b0d98b1bd1c3c88aad2e2b57a648b2bd5d2b8185fe866302280c769bfee0fb20bebe3edb0c25fc7ec6c9c1dba6da504e5a00f08
+  metadata.gz: 1eb53a355a90be12c04da34da4449c22ef8808c4b9bb98e422285a363089d5284fbf916138bc60dcfab525ade24016fdda5878649b9310d9c662857326d9ea0a
+  data.tar.gz: 6d825776dc6a23720bf90d72c9f61aed29665f19625d8c36dcd03388f751ccae3b798c38cab8c388c0f15b3b69f7fe17e42d7c52a61d7e6145e6cf3ce103d30c

data/CHANGELOG.md

@@ -1,3 +1,11 @@
+## [0.0.14] - 2025-07-05
+
+- o1 models don't support temperature so force it to 1
+
+## [0.0.13] - 2025-06-27
+
+- Add GenerateChatCompletion command
+
 ## [0.0.12] - 2025-05-27
 
 - Do not load service models in parallel due to undiagnosed race condition somewhere

data/src/foobara/ai/answer_bot/ask.rb

@@ -43,7 +43,10 @@ module Foobara
             inputs[:instructions] = instructions
           end
 
-          self.answer = run_mapped_subcommand!(ai_command, inputs)
+          inputs = domain_map!(inputs, to: ai_command)
+          answer = run_subcommand!(ai_command, inputs)
+
+          self.answer = domain_map!(answer, to: :string)
         end
       end
     end

data/src/foobara/ai/answer_bot/domain_mappers/anthropic_api/chat_to_create_message.rb

@@ -0,0 +1,56 @@
+require_relative "../../types/model_enum"
+
+module Foobara
+  module Ai
+    module AnswerBot
+      module DomainMappers
+        module AnthropicApi
+          class ChatToCreateMessage < Foobara::DomainMapper
+            from do
+              chat Types::Chat, :required
+              temperature :float
+              model :model_enum
+            end
+            to Foobara::Ai::AnthropicApi::CreateMessage
+
+            def map
+              system_messages, non_system_messages = chat.messages.partition(&:system?)
+
+              inputs = {
+                messages: non_system_messages.map do |message|
+                  { role: message.role, content: message.content }
+                end
+              }
+
+              unless system_messages.empty?
+                inputs[:system] = system_messages.map(&:content).join("\n")
+              end
+
+              if temperature
+                inputs[:temperature] = temperature
+              end
+
+              if model
+                inputs[:model] = model
+              end
+
+              inputs
+            end
+
+            def chat
+              from[:chat]
+            end
+
+            def temperature
+              from[:temperature]
+            end
+
+            def model
+              from[:model]
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/src/foobara/ai/answer_bot/domain_mappers/anthropic_api/message_result_to_message.rb

@@ -0,0 +1,23 @@
+module Foobara
+  module Ai
+    module AnswerBot
+      module DomainMappers
+        module AnthropicApi
+          class MessageResultToMessage < Foobara::DomainMapper
+            from Foobara::Ai::AnthropicApi::Types::MessageResult
+            to Types::Message
+
+            def map
+              {
+                content: message_result.content.last.text,
+                role: message_result.role
+              }
+            end
+
+            alias message_result from
+          end
+        end
+      end
+    end
+  end
+end

data/src/foobara/ai/answer_bot/domain_mappers/ollama_api/chat_completion_result_to_message.rb

@@ -0,0 +1,25 @@
+module Foobara
+  module Ai
+    module AnswerBot
+      module DomainMappers
+        module OllamaApi
+          class ChatCompletionResultToMessage < Foobara::DomainMapper
+            from Foobara::Ai::OllamaApi::Types::ChatCompletion
+            to Types::Message
+
+            def map
+              message = chat_completion.message
+
+              {
+                content: message.content,
+                role: message.role
+              }
+            end
+
+            alias chat_completion from
+          end
+        end
+      end
+    end
+  end
+end

data/src/foobara/ai/answer_bot/domain_mappers/ollama_api/chat_to_generate_chat_completion.rb

@@ -0,0 +1,52 @@
+require_relative "../../types/model_enum"
+
+module Foobara
+  module Ai
+    module AnswerBot
+      module DomainMappers
+        module OllamaApi
+          class ChatToGenerateChatCompletion < Foobara::DomainMapper
+            from do
+              chat Types::Chat, :required
+              temperature :float
+              model :model_enum
+            end
+            to Foobara::Ai::OllamaApi::GenerateChatCompletion
+
+            def map
+              inputs = {
+                messages: chat.messages.map do |message|
+                  { role: message.role, content: message.content }
+                end
+              }
+
+              if temperature
+                inputs[:options] = {
+                  temperature:
+                }
+              end
+
+              if model
+                inputs[:model] = model
+              end
+
+              inputs
+            end
+
+            def chat
+              from[:chat]
+            end
+
+            def temperature
+              from[:temperature]
+            end
+
+            def model
+              from[:model]
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/src/foobara/ai/answer_bot/domain_mappers/open_ai_api/chat_completion_result_to_message.rb

@@ -0,0 +1,25 @@
+module Foobara
+  module Ai
+    module AnswerBot
+      module DomainMappers
+        module OpenAiApi
+          class ChatCompletionResultToMessage < Foobara::DomainMapper
+            from Foobara::Ai::OpenAiApi::Types::ChatCompletion
+            to Types::Message
+
+            def map
+              message = chat_completion.choices.first.message
+
+              {
+                content: message.content,
+                role: message.role
+              }
+            end
+
+            alias chat_completion from
+          end
+        end
+      end
+    end
+  end
+end

data/src/foobara/ai/answer_bot/domain_mappers/open_ai_api/chat_to_generate_chat_completion.rb

@@ -0,0 +1,56 @@
+require_relative "../../types/model_enum"
+
+module Foobara
+  module Ai
+    module AnswerBot
+      module DomainMappers
+        module OpenAiApi
+          class ChatToGenerateChatCompletion < Foobara::DomainMapper
+            from do
+              chat Types::Chat, :required
+              temperature :float
+              model :model_enum
+            end
+            to Foobara::Ai::OpenAiApi::CreateChatCompletion
+
+            def map
+              inputs = {
+                messages: chat.messages.map do |message|
+                  { role: message.role, content: message.content }
+                end
+              }
+
+              if temperature
+                # NOTE: some models don't support temperature, like o1. Instead of possibly receiving an error
+                # we will set the temperature to 1 which prevents a 400. Maybe print a warning instead?
+                inputs[:temperature] = model_supports_temperature? ? temperature : 1
+              end
+
+              if model
+                inputs[:model] = model
+              end
+
+              inputs
+            end
+
+            def chat
+              from[:chat]
+            end
+
+            def temperature
+              from[:temperature]
+            end
+
+            def model
+              from[:model]
+            end
+
+            def model_supports_temperature?
+              model && model.to_s !~ /\Ao1(-|$)/
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/src/foobara/ai/answer_bot/generate_next_message.rb

@@ -0,0 +1,49 @@
+module Foobara
+  module Ai
+    module AnswerBot
+      class GenerateNextMessage < Foobara::Command
+        description "Takes a chat and generates the next message using the model of your choice."
+
+        inputs do
+          chat Types::Chat, :required
+          service :service_enum, "If two services expose the same model, you can specify which one to use."
+          model :model_enum, default: "claude-3-7-sonnet-20250219", description: "The model to use."
+          temperature :float
+        end
+
+        result Types::Message
+
+        AI_SERVICES.each_key do |service|
+          command = Foobara::Ai::AnswerBot::DomainMappers::ServiceToChatCompletionCommand.map!(service)
+          depends_on command
+        end
+
+        def execute
+          determine_ai_service
+          determine_ai_command
+          run_ai_service
+
+          next_message
+        end
+
+        attr_accessor :ai_command, :response, :ai_service, :next_message
+
+        def determine_ai_service
+          self.ai_service = service || domain_map!(model, from: :model_enum)
+        end
+
+        def determine_ai_command
+          self.ai_command = Foobara::Ai::AnswerBot::DomainMappers::ServiceToChatCompletionCommand.map!(ai_service)
+        end
+
+        def run_ai_service
+          inputs = { chat:, temperature:, model: }
+          inputs = domain_map!(inputs, to: ai_command)
+          next_message = run_subcommand!(ai_command, inputs)
+
+          self.next_message = domain_map!(next_message, to: Types::Message)
+        end
+      end
+    end
+  end
+end

data/src/foobara/ai/answer_bot/types/chat.rb

@@ -0,0 +1,13 @@
+module Foobara
+  module Ai
+    module AnswerBot
+      module Types
+        class Chat < Foobara::Model
+          attributes do
+            messages [Message]
+          end
+        end
+      end
+    end
+  end
+end

data/src/foobara/ai/answer_bot/types/message.rb

@@ -0,0 +1,18 @@
+module Foobara
+  module Ai
+    module AnswerBot
+      module Types
+        class Message < Foobara::Model
+          attributes do
+            role :symbol, one_of: [:assistant, :user, :system]
+            content :string
+          end
+
+          def system?
+            role == :system
+          end
+        end
+      end
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: foobara-ai
 version: !ruby/object:Gem::Version
-  version: 0.0.12
+  version: 0.0.14
 platform: ruby
 authors:
 - Miles Georgi
@@ -21,22 +21,31 @@ files:
 - README.md
 - lib/foobara/ai.rb
 - src/foobara/ai/answer_bot/ask.rb
+- src/foobara/ai/answer_bot/domain_mappers/anthropic_api/chat_to_create_message.rb
 - src/foobara/ai/answer_bot/domain_mappers/anthropic_api/message_result_to_answer.rb
+- src/foobara/ai/answer_bot/domain_mappers/anthropic_api/message_result_to_message.rb
 - src/foobara/ai/answer_bot/domain_mappers/anthropic_api/model_to_foobara_model.rb
 - src/foobara/ai/answer_bot/domain_mappers/anthropic_api/model_to_model_enum_string.rb
 - src/foobara/ai/answer_bot/domain_mappers/anthropic_api/question_to_create_message.rb
 - src/foobara/ai/answer_bot/domain_mappers/model_to_ai_service.rb
+- src/foobara/ai/answer_bot/domain_mappers/ollama_api/chat_completion_result_to_message.rb
 - src/foobara/ai/answer_bot/domain_mappers/ollama_api/chat_completion_to_answer.rb
+- src/foobara/ai/answer_bot/domain_mappers/ollama_api/chat_to_generate_chat_completion.rb
 - src/foobara/ai/answer_bot/domain_mappers/ollama_api/model_to_foobara_model.rb
 - src/foobara/ai/answer_bot/domain_mappers/ollama_api/model_to_model_enum_string.rb
 - src/foobara/ai/answer_bot/domain_mappers/ollama_api/question_to_generate_chat_completion.rb
+- src/foobara/ai/answer_bot/domain_mappers/open_ai_api/chat_completion_result_to_message.rb
 - src/foobara/ai/answer_bot/domain_mappers/open_ai_api/chat_completion_to_answer.rb
+- src/foobara/ai/answer_bot/domain_mappers/open_ai_api/chat_to_generate_chat_completion.rb
 - src/foobara/ai/answer_bot/domain_mappers/open_ai_api/model_to_foobara_model.rb
 - src/foobara/ai/answer_bot/domain_mappers/open_ai_api/model_to_model_enum_string.rb
 - src/foobara/ai/answer_bot/domain_mappers/open_ai_api/question_to_create_chat_completion.rb
 - src/foobara/ai/answer_bot/domain_mappers/service_to_command.rb
 - src/foobara/ai/answer_bot/domain_mappers/service_to_list_models_command.rb
+- src/foobara/ai/answer_bot/generate_next_message.rb
 - src/foobara/ai/answer_bot/list_models.rb
+- src/foobara/ai/answer_bot/types/chat.rb
+- src/foobara/ai/answer_bot/types/message.rb
 - src/foobara/ai/answer_bot/types/model.rb
 - src/foobara/ai/answer_bot/types/model_enum.rb
 - src/foobara/ai/answer_bot/types/service_enum.rb
@@ -62,7 +71,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.6.7
+rubygems_version: 3.6.9
 specification_version: 4
 summary: No description. Add one.
 test_files: []