foobara-open-ai-api 0.0.9 → 0.0.11

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5e6cb9bdbfee03bc629c3c4b2641363f846dec53e14317860cda02c7be0faa22
-  data.tar.gz: 60f99581e848a20defcbfbc5f9ba5b01fa4b30a61a49bb8399c557eb8cfd2921
+  metadata.gz: 8c61e466a2f1f4ce7908a205c89f8518e0c0bae5f05f507011352b546e796aa5
+  data.tar.gz: 0afacf44c3eaf6ff404bcb4e171fc74ecc833d8b97110fac1967ce650045d949
 SHA512:
-  metadata.gz: f79050ea578166386b43af65ed4efe7249e92aba4d3c87eced62d5e8859a65634c2e852fd18a304fccfd8d904a4a7a40d7c4a092ab030b89fe9c7cd0c684cb2f
-  data.tar.gz: 3ae087e54109610caa2ce6d065d8658587b751612bf7ddc26e0a7b00b4f69a4ce5db72b1a85073979ca2f47d9ae951d6ce57e4e4c4a453e5eccb207961fb8d91
+  metadata.gz: 1dd6e5433e7d018fd9fb30a45f7fe4af6bcaf899ef2323ec1175bbe4d7383361f4abdf8eed6cff746a5964283bc2f269c43d2e3f0dac301b0b467c53a13c8531
+  data.tar.gz: db2e7fe5d1f4f8d013d444f164cd727f6c49d5a95d276899545d5a26f144233a0eae3c708dbebff4da44906dbd6f7a108a9d42f4493b79f32ce03a965fb8ad5c

data/CHANGELOG.md

@@ -1,3 +1,12 @@
+## [0.0.11] - 2025-02-19
+
+- Cache the list of models in ListModels
+- Automatically construct the list of valid model names for model_enum at startup
+
+## [0.0.10] - 2025-02-18
+
+- Add a ModelEnum
+
 ## [0.0.8] - 2025-02-17
 
 - Add support for some newly added fields

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/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/list_models.rb

@@ -1,12 +1,16 @@
+require_relative "base_command"
+require_relative "types/model"
+require "foobara/cached_command"
+
 module Foobara
   module Ai
     module OpenAiApi
       class ListModels < BaseCommand
-        path "/models"
+        include CachedCommand
 
-        inputs do
-          api_token :string
-        end
+        self.foobara_cache_expiry = 24 * 60 * 60
+
+        path "/models"
 
         result [Types::Model]
 

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

@@ -0,0 +1,15 @@
+require_relative "../list_models"
+
+module Foobara
+  module Ai
+    module OpenAiApi
+      module Types
+        model_list = ListModels.run!.map(&:id).sort
+
+        ModelEnum = Foobara::Enumerated.make_module(model_list)
+
+        OpenAiApi.foobara_register_type(:model, :string, one_of: ModelEnum)
+      end
+    end
+  end
+end

metadata

@@ -1,14 +1,28 @@
 --- !ruby/object:Gem::Specification
 name: foobara-open-ai-api
 version: !ruby/object:Gem::Version
-  version: 0.0.9
+  version: 0.0.11
 platform: ruby
 authors:
 - Miles Georgi
 bindir: bin
 cert_chain: []
-date: 2025-02-17 00:00:00.000000000 Z
+date: 2025-02-19 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: foobara-cached-command
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: foobara-http-api-command
   requirement: !ruby/object:Gem::Requirement
@@ -44,6 +58,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