ruby_llm 0.1.0.pre28 → 0.1.0.pre30

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 467d0db634f70567417e20d87791372e39276e18fa4cd145cc065925b7379e7f
-  data.tar.gz: 70f3e92280366104d53446891ae645183706df5d8d384799c83299a43142e2eb
+  metadata.gz: 33b602ae544211e79207cf51bbfad5f1ed9c78cd1d9710d4f57916c95377bb60
+  data.tar.gz: 3fac0d6aed252218d7f7bd8e460f7b281c0306ca066e0194193e11a44ac8400d
 SHA512:
-  metadata.gz: 06e6021199954eace1c88e984f052d1d2644a9df3905280f9903fbd7ecfdf3b3bf904d3078c34982aeed43e15a1871946e1a9ad174ba621c7dcf3ce64ab190b5
-  data.tar.gz: 6697cd297ed9805114c433bf7610d4d608dce919b5ee3d99d1cec135d329640d607f64d3282ace206cb176f30390defc9c14ced6842f659033d96c04e7853dbe
+  metadata.gz: fa21a4d89b3704f384c257a7bcca71e42b0442b2f660ddd2c93d108f77bcf547b5d1c05101d85269abc5aab8fbe37e3bec3da8a4fd09d77f8184ed923fea0986
+  data.tar.gz: 5295d0c1ec5660e4ce9e7388836bbc37d82eae0ccce77a4f601f59ed91933cb2dc31b84f79f5aab3271d5b8b318aa190ca0224926a28c69a0e409ed9035db4c8

data/lib/ruby_llm/model_capabilities/anthropic.rb

@@ -54,6 +54,10 @@ module RubyLLM
         end
       end
 
+      def model_type(_)
+        'chat'
+      end
+
       PRICES = {
         claude35_sonnet: { input: 3.0, output: 15.0 }, # $3.00/$15.00 per million tokens
         claude35_haiku: { input: 0.80, output: 4.0 }, # $0.80/$4.00 per million tokens

data/lib/ruby_llm/model_capabilities/openai.rb

@@ -175,7 +175,7 @@ module RubyLLM
         name
           .gsub(/(\d{4}) (\d{2}) (\d{2})/, '\1\2\3')
           .gsub(/^Gpt /, 'GPT-')
-          .gsub(/^O[13] /, 'O\1-')
+          .gsub(/^O([13]) /, 'O\1-')
           .gsub(/^Chatgpt /, 'ChatGPT-')
           .gsub(/^Tts /, 'TTS-')
           .gsub(/^Dall E /, 'DALL-E-')

data/lib/ruby_llm/models.json

@@ -82,7 +82,7 @@
     "provider": "anthropic",
     "context_window": 100000,
     "max_tokens": 4096,
-    "type": "model",
+    "type": "chat",
     "family": "claude2",
     "supports_vision": false,
     "supports_functions": false,
@@ -98,7 +98,7 @@
     "provider": "anthropic",
     "context_window": 100000,
     "max_tokens": 4096,
-    "type": "model",
+    "type": "chat",
     "family": "claude2",
     "supports_vision": false,
     "supports_functions": false,
@@ -114,7 +114,7 @@
     "provider": "anthropic",
     "context_window": 200000,
     "max_tokens": 8192,
-    "type": "model",
+    "type": "chat",
     "family": "claude35_haiku",
     "supports_vision": false,
     "supports_functions": true,
@@ -130,7 +130,7 @@
     "provider": "anthropic",
     "context_window": 200000,
     "max_tokens": 8192,
-    "type": "model",
+    "type": "chat",
     "family": "claude35_sonnet",
     "supports_vision": true,
     "supports_functions": true,
@@ -146,7 +146,7 @@
     "provider": "anthropic",
     "context_window": 200000,
     "max_tokens": 8192,
-    "type": "model",
+    "type": "chat",
     "family": "claude35_sonnet",
     "supports_vision": true,
     "supports_functions": true,
@@ -162,7 +162,7 @@
     "provider": "anthropic",
     "context_window": 200000,
     "max_tokens": 4096,
-    "type": "model",
+    "type": "chat",
     "family": "claude3_haiku",
     "supports_vision": true,
     "supports_functions": true,
@@ -178,7 +178,7 @@
     "provider": "anthropic",
     "context_window": 200000,
     "max_tokens": 4096,
-    "type": "model",
+    "type": "chat",
     "family": "claude3_opus",
     "supports_vision": true,
     "supports_functions": true,
@@ -194,7 +194,7 @@
     "provider": "anthropic",
     "context_window": 200000,
     "max_tokens": 4096,
-    "type": "model",
+    "type": "chat",
     "family": "claude3_sonnet",
     "supports_vision": true,
     "supports_functions": true,
@@ -1707,7 +1707,7 @@
   {
     "id": "o1-mini",
     "created_at": "2024-09-06T20:56:48+02:00",
-    "display_name": "O-Mini",
+    "display_name": "O1-Mini",
     "provider": "openai",
     "context_window": 200000,
     "max_tokens": 4096,
@@ -1726,7 +1726,7 @@
   {
     "id": "o1-mini-2024-09-12",
     "created_at": "2024-09-06T20:56:19+02:00",
-    "display_name": "O-Mini 20240912",
+    "display_name": "O1-Mini 20240912",
     "provider": "openai",
     "context_window": 200000,
     "max_tokens": 65536,
@@ -1745,7 +1745,7 @@
   {
     "id": "o1-preview",
     "created_at": "2024-09-06T20:54:57+02:00",
-    "display_name": "O-Preview",
+    "display_name": "O1-Preview",
     "provider": "openai",
     "context_window": 4096,
     "max_tokens": 4096,
@@ -1764,7 +1764,7 @@
   {
     "id": "o1-preview-2024-09-12",
     "created_at": "2024-09-06T20:54:25+02:00",
-    "display_name": "O-Preview 20240912",
+    "display_name": "O1-Preview 20240912",
     "provider": "openai",
     "context_window": 4096,
     "max_tokens": 4096,

data/lib/ruby_llm/providers/anthropic.rb

@@ -88,7 +88,7 @@ module RubyLLM
             created_at: Time.parse(model['created_at']),
             display_name: model['display_name'],
             provider: slug,
-            type: model['type'],
+            type: capabilities.model_type(model['id']),
             family: capabilities.model_family(model['id']),
             context_window: capabilities.determine_context_window(model['id']),
             max_tokens: capabilities.determine_max_tokens(model['id']),

data/lib/ruby_llm/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RubyLLM
-  VERSION = '0.1.0.pre28'
+  VERSION = '0.1.0.pre30'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm
 version: !ruby/object:Gem::Version
-  version: 0.1.0.pre28
+  version: 0.1.0.pre30
 platform: ruby
 authors:
 - Carmine Paolino