Update list of language models, Assistant model belongs_to LanguageMo…

…del, implement the "best" naming
AllYourBot · May 13, 2024 · de6c887 · de6c887
1 parent 77dcefb
commit de6c887
Show file tree

Hide file tree

Showing 5 changed files with 18 additions and 9 deletions.
diff --git a/app/models/assistant.rb b/app/models/assistant.rb
@@ -7,6 +7,8 @@ class Assistant < ApplicationRecord
   has_many :steps, dependent: :destroy
   has_many :messages # TODO: What should happen if an assistant is deleted?
 
+  belongs_to :language_model, primary_key: :name, foreign_key: :model
+
   validates :tools, presence: true, allow_blank: true
 
   MAX_LIST_DISPLAY = 5

diff --git a/app/models/language_model.rb b/app/models/language_model.rb
@@ -2,4 +2,10 @@
 class LanguageModel < ApplicationRecord
    def readonly?() = !new_record?
    def before_destroy() = raise ActiveRecord::ReadOnlyRecord
+
+   PROVIDER_ID_MAP = {'gpt-best': 'gpt-4-turbo',
+     'claude-best': 'claude-3-opus-20240229'}             
+   def provider_id
+     PROVIDER_ID_MAP[self.name.to_sym] || self.name unless self.name =~ /^best/
+   end
 end
diff --git a/app/services/ai_backends/anthropic.rb b/app/services/ai_backends/anthropic.rb
@@ -49,7 +49,7 @@ def get_next_chat_message(&chunk_received_handler)
 
     begin
       response = @client.messages(
-        model: @assistant.model,
+        model: @assistant.model.provider_id,
         system: @assistant.instructions,
         messages: preceding_messages,
         parameters: {

diff --git a/app/services/ai_backends/open_ai.rb b/app/services/ai_backends/open_ai.rb
@@ -49,7 +49,7 @@ def get_next_chat_message(&chunk_received_handler)
 
     begin
       response = @client.chat(parameters: {
-        model: @assistant.model,
+        model: @assistant.language_model.provider_id,
         messages: system_message + preceding_messages,
         stream: response_handler,
         max_tokens: 2000, # we should really set this dynamically, based on the model, to the max

diff --git a/db/migrate/20240512163300_create_language_models.rb b/db/migrate/20240512163300_create_language_models.rb
@@ -10,22 +10,23 @@ def change
     # Current models
     up_only do
       {
-        'open_ai_best': 'Best OpenAI Model',
-        'claude_best': 'Best Anthropic Model',
+        'gpt-best': 'Best OpenAI Model',
+        'claude-best': 'Best Anthropic Model',
 
         'gpt-4': 'ChatGPT 4',
-        'gpt-4-turbo': 'ChatGPT 4 Turbo with Vision (may update)',
+        'gpt-4-turbo': 'ChatGPT 4 Turbo with Vision (may update in future)',
         'gpt-4-turbo-2024-04-09': 'ChatGPT-4 Turbo with Vision (2024-04-09)',
         'gpt-4-turbo-preview': 'ChatGPT 4 Turbo Preview',
         'gpt-4-0125-preview': 'ChatGPT 4 Turbo Preview (2024-01-25)',
         'gpt-4-1106-preview': 'ChatGPT 4 Turbo Preview (2023-11-06)',
-        'gpt-4-vision-preview': 'ChatGPT 4 Turbo Model preview with the ability to understand images',
-        'gpt-4-1106-vision-preview': 'ChatGPT 4 Turbo Model preview with the ability to understand images (2023-11-06)',
+        'gpt-4-vision-preview': 'ChatGPT 4 Turbo Model preview with the ability to understand images (2023-11-06)',
+        'gpt-4-1106-vision-preview': 'ChatGPT 4 Turbo preview with the ability to understand images (2023-11-06)',
         'gpt-4-0613': 'ChatGPT 4 Snapshot improved function calling (2023-06-13)',
 
         'gpt-3.5-turbo': 'ChatGPT 3.5 Turbo',
-        'gpt-3.5-turbo-16k-0613': 'ChatGPT 3.5 Turbo (2023-06-13)',
-        'gpt-3.5-turbo-0125': 'ChatGPT 3.5 Turbo (2023-01-25)',
+        'gpt-3.5-turbo-16k-0613': 'ChatGPT 3.5 Turbo (2022-06-13)',
+        'gpt-3.5-turbo-0125': 'ChatGPT 3.5 Turbo (2022-01-25)',
+        'gpt-3.5-turbo-1106': 'ChatGPT 3.5 Turbo (2022-11-06)',
         'gpt-3.5-turbo-instruct': 'ChatGPT 3.5 Turbo Instruct',
 
         'claude-3-opus-20240229': 'Claude 3 Opus (2024-02-29)',