foobara-open-ai-api 0.0.8 → 0.0.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b1134ded66f3c63f8f0def005ed37383f5c69522899e51fe3316dce02c1210f8
-  data.tar.gz: 0170e54de38d1b4ac99b8d7c110d67ce67dbb2da53bebca8fdf94dda7c0f0027
+  metadata.gz: c561a0e79f029a99ffad41fe5c090ddfab2c7e008c7fd0f16b6d21e4d7901085
+  data.tar.gz: 284fdd95baef9a2bdbd33fd89aeed1dc1644017a93d36e5045ce17cb58c471f6
 SHA512:
-  metadata.gz: be8f41c655f0a08a2c3b44a2c13ad9f5bf2f39fb055b2119b537a3459e344fb476bb672087e2887ae5e3d779bca81b93b0f9a2c0771a56f98b24a72fe26c8f9f
-  data.tar.gz: 2f091240994a101906585d51cd6915d0153dd7d2eb1f011f63d6cfa8f7e8593144be01ce1877ca47e8096870eef4b91cfa31a8966d617a8d24fce695d0675b39
+  metadata.gz: 1e7bd6525404df690f81f902e01bc7e611e23ff5ba71b5d61df07a3743239effb8ae7a96438a52463ef89bee7e3afa6c1941846d66574342b039e996c2e5ea78
+  data.tar.gz: 4a13093a70ab33d82a6e2854c622fc2ebae81e447f17c0532251ef84ad5dcfda1ef1e6997e3e5ebd21d14a940749fcbc0b3365b4eb19d74d9145530b64e013ca

data/CHANGELOG.md

@@ -1,9 +1,14 @@
-## [0.0.7] - 2025-02-17
+## [0.0.9] - 2025-02-18
+
+- Add a ModelEnum
+
+## [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/lib/foobara/open_ai_api.rb

@@ -1,4 +1,5 @@
-# TODO: consider moving this to src/ and deleting lib/
 require "foobara/all"
 
+require_relative "../../src/foobara/ai/open_ai_api"
+
 Foobara::Util.require_directory "#{__dir__}/../../src"

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

@@ -4,18 +4,24 @@ 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
-          # 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
         end
 
         def api_token
-          inputs[:api_token] || ENV.fetch("OPENAI_API_KEY", nil)
+          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

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

@@ -8,7 +8,7 @@ module Foobara
         http_method :post
 
         add_inputs do
-          model :string, default: "gpt-3.5-turbo", one_of: %w[gpt-3.5-turbo gpt-4]
+          model :model, default: Types::ModelEnum::GPT_3_5_TURBO
           messages :array do
             role :string, one_of: %w[system user assistant]
             content :string

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

@@ -0,0 +1,73 @@
+module Foobara
+  module Ai
+    module OpenAiApi
+      module Types
+        # Can grab an updated list by running something like:
+        #
+        # irb(main):001> require "foobara/open_ai_api"
+        # irb(main):002> Foobara::Ai::OpenAiApi.api_token = File.read("tokens/open_ai_token.txt") and nil
+        # irb(main):003> Foobara::Ai::OpenAiApi::ListModels.run!.map(&:id).sort
+        # =>
+        # ["babbage-002",
+        #  "chatgpt-4o-latest",
+        #  "dall-e-2",
+        #  "dall-e-3",
+        #  "davinci-002",
+        #  "gpt-3.5-turbo",
+        # ...
+        ModelEnum = Foobara::Enumerated.make_module(
+          "babbage-002",
+          "chatgpt-4o-latest",
+          "dall-e-2",
+          "dall-e-3",
+          "davinci-002",
+          "gpt-3.5-turbo",
+          "gpt-3.5-turbo-0125",
+          "gpt-3.5-turbo-1106",
+          "gpt-3.5-turbo-16k",
+          "gpt-3.5-turbo-instruct",
+          "gpt-3.5-turbo-instruct-0914",
+          "gpt-4",
+          "gpt-4-0125-preview",
+          "gpt-4-0613",
+          "gpt-4-1106-preview",
+          "gpt-4-turbo",
+          "gpt-4-turbo-2024-04-09",
+          "gpt-4-turbo-preview",
+          "gpt-4o",
+          "gpt-4o-2024-05-13",
+          "gpt-4o-2024-08-06",
+          "gpt-4o-2024-11-20",
+          "gpt-4o-audio-preview",
+          "gpt-4o-audio-preview-2024-10-01",
+          "gpt-4o-audio-preview-2024-12-17",
+          "gpt-4o-mini",
+          "gpt-4o-mini-2024-07-18",
+          "gpt-4o-mini-audio-preview",
+          "gpt-4o-mini-audio-preview-2024-12-17",
+          "gpt-4o-mini-realtime-preview",
+          "gpt-4o-mini-realtime-preview-2024-12-17",
+          "gpt-4o-realtime-preview",
+          "gpt-4o-realtime-preview-2024-10-01",
+          "gpt-4o-realtime-preview-2024-12-17",
+          "o1-mini",
+          "o1-mini-2024-09-12",
+          "o1-preview",
+          "o1-preview-2024-09-12",
+          "omni-moderation-2024-09-26",
+          "omni-moderation-latest",
+          "text-embedding-3-large",
+          "text-embedding-3-small",
+          "text-embedding-ada-002",
+          "tts-1",
+          "tts-1-1106",
+          "tts-1-hd",
+          "tts-1-hd-1106",
+          "whisper-1"
+        )
+
+        OpenAiApi.foobara_register_type(:model, :string, one_of: ModelEnum)
+      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,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: foobara-open-ai-api
 version: !ruby/object:Gem::Version
-  version: 0.0.8
+  version: 0.0.10
 platform: ruby
 authors:
 - Miles Georgi
 bindir: bin
 cert_chain: []
-date: 2025-02-17 00:00:00.000000000 Z
+date: 2025-02-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: foobara-http-api-command
@@ -44,6 +44,7 @@ files:
 - 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
+- src/foobara/ai/open_ai_api/types/model_enum.rb
 homepage: https://github.com/foobara/open-ai-api
 licenses:
 - Apache-2.0