foobara-open-ai-api 0.0.7 → 0.0.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ef8553ce176e7f0b8d4059b06c7606ff4670381d404f51db536812dec5eb6469
-  data.tar.gz: 6736eb9ce73a9b6f1f6b7e56a09e347d80cec7c3dfe81f69f757c21415837657
+  metadata.gz: 5e6cb9bdbfee03bc629c3c4b2641363f846dec53e14317860cda02c7be0faa22
+  data.tar.gz: 60f99581e848a20defcbfbc5f9ba5b01fa4b30a61a49bb8399c557eb8cfd2921
 SHA512:
-  metadata.gz: 0c73216f70e9cbf0ef4ef224c2c8db9be55a39102f19bd799edbd86d5a583d43dd450a5d36b6fd761eaaa4267a22144094ecbcda564feea1c82390f90e587a44
-  data.tar.gz: b80ee1a7820d1ced6bf690d1b6b5bb517a14c789f76aaab0cf75573913ca12003a3c5395c893f28af620580130acc77fe3e9e280248ce9743246c3203bea54eb
+  metadata.gz: f79050ea578166386b43af65ed4efe7249e92aba4d3c87eced62d5e8859a65634c2e852fd18a304fccfd8d904a4a7a40d7c4a092ab030b89fe9c7cd0c684cb2f
+  data.tar.gz: 3ae087e54109610caa2ce6d065d8658587b751612bf7ddc26e0a7b00b4f69a4ce5db72b1a85073979ca2f47d9ae951d6ce57e4e4c4a453e5eccb207961fb8d91

data/CHANGELOG.md

@@ -1,8 +1,10 @@
-## [0.0.7] - 2025-02-17
+## [0.0.8] - 2025-02-17
 
 - Add support for some newly added fields
 - Make sure our models don't break when OpenAI adds new attributes to their responses
   before we have a chance to update them
+- Implement ListModels
+- Provide a way to set api token on project module
 
 ## [0.0.6] - 2025-02-16
 

data/src/foobara/ai/open_ai_api/base_command.rb

@@ -0,0 +1,33 @@
+require "foobara/http_api_command"
+
+module Foobara
+  module Ai
+    module OpenAiApi
+      class BaseCommand < Foobara::Command
+        class << self
+          attr_accessor :api_token
+        end
+
+        include HttpApiCommand
+
+        base_url "https://api.openai.com/v1"
+
+        inputs do
+          api_token :string
+        end
+
+        def api_token
+          inputs[:api_token] || BaseCommand.api_token || api_token_from_env
+        end
+
+        def api_token_from_env
+          ENV.fetch("OPENAI_API_KEY", nil)
+        end
+
+        def build_request_headers
+          self.request_headers = { "Content-Type" => "application/json", "Authorization" => "Bearer #{api_token}" }
+        end
+      end
+    end
+  end
+end

data/src/foobara/ai/open_ai_api/create_chat_completion.rb

@@ -1,13 +1,13 @@
+require_relative "base_command"
+
 module Foobara
   module Ai
     module OpenAiApi
-      class CreateChatCompletion < Foobara::Command
-        URL = "https://api.openai.com/v1/chat/completions".freeze
+      class CreateChatCompletion < BaseCommand
+        path "/chat/completions"
+        http_method :post
 
-        inputs do
-          # TODO: come up with some kind of execution context object to wrap all of these calls upon which
-          # data wrapping a "request" can live instead of being passed around?
-          api_token :string
+        add_inputs do
           model :string, default: "gpt-3.5-turbo", one_of: %w[gpt-3.5-turbo gpt-4]
           messages :array do
             role :string, one_of: %w[system user assistant]
@@ -17,50 +17,12 @@ module Foobara
 
         result Types::ChatCompletion
 
-        def execute
-          build_request_body
-          build_request_headers
-          issue_http_request
-          parse_response
-          build_completion
-
-          completion
-        end
-
-        attr_accessor :request_body, :request_headers, :response, :response_body, :completion
-
         def build_request_body
           self.request_body = { model:, messages: }
         end
 
-        def build_request_headers
-          self.request_headers = { "Content-Type" => "application/json", "Authorization" => "Bearer #{api_token}" }
-        end
-
-        def api_token
-          inputs[:api_token] || ENV.fetch("OPENAI_API_KEY", nil)
-        end
-
-        def issue_http_request
-          url = URI.parse(URL)
-          self.response = Net::HTTP.post(url, JSON.generate(request_body), request_headers)
-        end
-
-        def parse_response
-          completion_json = if response.is_a?(Net::HTTPSuccess)
-                              response.body
-                            else
-                              # :nocov:
-                              raise "Could not get completion from #{URL}: " \
-                                    "#{response.code} #{response.message}"
-                              # :nocov:
-                            end
-
-          self.response_body = JSON.parse(completion_json)
-        end
-
-        def build_completion
-          self.completion = Types::ChatCompletion.new(response_body)
+        def build_result
+          Types::ChatCompletion.new(response_body, ignore_unexpected_attributes: true)
         end
       end
     end

data/src/foobara/ai/open_ai_api/list_models.rb

@@ -0,0 +1,19 @@
+module Foobara
+  module Ai
+    module OpenAiApi
+      class ListModels < BaseCommand
+        path "/models"
+
+        inputs do
+          api_token :string
+        end
+
+        result [Types::Model]
+
+        def build_result
+          response_body["data"]
+        end
+      end
+    end
+  end
+end

data/src/foobara/ai/open_ai_api/types/model.rb

@@ -0,0 +1,16 @@
+module Foobara
+  module Ai
+    module OpenAiApi
+      module Types
+        class Model < Foobara::Model
+          attributes do
+            id :string
+            created :datetime
+            object :string
+            owned_by :string
+          end
+        end
+      end
+    end
+  end
+end

data/src/foobara/ai/open_ai_api.rb

@@ -5,6 +5,16 @@ module Foobara
   module Ai
     module OpenAiApi
       foobara_domain!
+
+      class << self
+        def api_token=(token)
+          BaseCommand.api_token = token
+        end
+
+        def api_token
+          BaseCommand.api_token
+        end
+      end
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: foobara-open-ai-api
 version: !ruby/object:Gem::Version
-  version: 0.0.7
+  version: 0.0.9
 platform: ruby
 authors:
 - Miles Georgi
@@ -10,7 +10,7 @@ cert_chain: []
 date: 2025-02-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: foobara
+  name: foobara-http-api-command
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -36,11 +36,14 @@ files:
 - README.md
 - lib/foobara/open_ai_api.rb
 - src/foobara/ai/open_ai_api.rb
+- src/foobara/ai/open_ai_api/base_command.rb
 - src/foobara/ai/open_ai_api/create_chat_completion.rb
+- src/foobara/ai/open_ai_api/list_models.rb
 - src/foobara/ai/open_ai_api/types/chat_completion.rb
 - src/foobara/ai/open_ai_api/types/chat_completion/choice.rb
 - src/foobara/ai/open_ai_api/types/chat_completion/message.rb
 - src/foobara/ai/open_ai_api/types/chat_completion/usage.rb
+- src/foobara/ai/open_ai_api/types/model.rb
 homepage: https://github.com/foobara/open-ai-api
 licenses:
 - Apache-2.0