rasti-ai 1.0.0 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e5f39be89f25780d7f9b33adf06b12109c9ddb482c2aa85220d400cfb945c000
-  data.tar.gz: e4ce69118debdf636aab53cf8fa25cfd233633ac5c499f9f0bc1a4b5bc05294f
+  metadata.gz: 3bd63fcf9d5f5af5cd58627893322ee4f09c3a5d45c489865ff3cebe9c11aaf3
+  data.tar.gz: ed2992eabeb5a0c0fe429d8c9dbaa96036658cf3c27540b4069e80e5ee6b83df
 SHA512:
-  metadata.gz: 2f19e650f0908142aedbc09b9d8d372b24da94979f23464d79de17f8c44cd2c3ed86c666dfc3dc46066bf800ecd572c16cc2b3c321692601d29c76c89a38170e
-  data.tar.gz: a9da7e1802d9877e1171eb9cfa5b602ad642de251131ee6cdbe0bf77cbf810a24817af8846e246a2a57cf47191dc00542d60b2591ded65ef87860e3ec3104d63
+  metadata.gz: aea33a82665fb2e565ba9efcbd7de3ed7872e7e28ce95a407013dae8aa966cf916f4963e9e324f631e7cbd1d9ebe915b0832a8627466409c7a1c1e9ffcfd09e4
+  data.tar.gz: cf551997163a891f116e55cbb19225f1bda0006ab32b360cdd78409f4b200ebf5658f414adecc9066ddbcca768f8d2d92a0a425d69898aa9815f6485871fbda5

data/lib/rasti/ai/open_ai/assistant.rb

@@ -5,8 +5,9 @@ module Rasti
 
         attr_reader :state
 
-        def initialize(client:nil, state:nil, model:nil, tools:[], logger:nil)
+        def initialize(client:nil, json_schema:nil, state:nil, model:nil, tools:[], logger:nil)
           @client = client || Client.new
+          @json_schema = json_schema
           @state = state || AssistantState.new
           @model = model
           @tools = {}
@@ -29,7 +30,8 @@ module Rasti
           loop do
             response = client.chat_completions messages: messages,
                                                model: model,
-                                               tools: serialized_tools
+                                               tools: serialized_tools,
+                                               response_format: response_format
 
             choice = response['choices'][0]['message']
 
@@ -64,7 +66,7 @@ module Rasti
 
         private
 
-        attr_reader :client, :model, :tools, :serialized_tools, :logger
+        attr_reader :client, :json_schema, :model, :tools, :serialized_tools, :logger
 
         def messages
           state.messages
@@ -90,7 +92,16 @@ module Rasti
           "Error: #{ex.message}"
         end
 
+        def response_format
+          return nil if json_schema.nil?
+
+          {
+            type: 'json_schema',
+            json_schema: json_schema
+          }
+        end
+
       end
     end
   end
-end
+end

data/lib/rasti/ai/open_ai/client.rb

@@ -10,7 +10,7 @@ module Rasti
           @logger = logger || Rasti::AI.logger
         end
 
-        def chat_completions(messages:, model:nil, tools:[])
+        def chat_completions(messages:, model:nil, tools:[], response_format: nil)
           body = {
             model: model || Rasti::AI.openai_default_model,
             messages: messages,
@@ -18,6 +18,8 @@ module Rasti
             tool_choice: tools.empty? ? 'none' : 'auto'
           }
 
+          body[:response_format] = response_format unless response_format.nil?
+
           post '/chat/completions', body
         end
 
@@ -54,4 +56,4 @@ module Rasti
     end
 
   end
-end
+end

data/lib/rasti/ai/version.rb

@@ -1,5 +1,5 @@
 module Rasti
   module AI
-    VERSION = '1.0.0'
+    VERSION = '1.1.0'
   end
 end

data/spec/open_ai/assistant_spec.rb

@@ -8,14 +8,17 @@ describe Rasti::AI::OpenAI::Assistant do
 
   let(:answer) { 'Lionel Messi scored 672 goals in 778 official matches for FC Barcelona.' }
 
-  def stub_open_ai_chat_completions(model:nil, question:, answer:)
+  def stub_open_ai_chat_completions(question:, answer:, model:nil, json_schema:nil)
     model ||= Rasti::AI.openai_default_model
 
+    body = read_json_resource('open_ai/basic_request.json', model: model, prompt: question)
+    body['response_format'] = {type: 'json_schema', json_schema: json_schema} if json_schema
+
     stub_request(:post, api_url)
-      .with(body: read_resource('open_ai/basic_request.json', model: model, prompt: question))
+      .with(body: JSON.dump(body))
       .to_return(body: read_resource('open_ai/basic_response.json', content: answer))
   end
-
+ 
 
   it 'Default' do
     stub_open_ai_chat_completions question: question, answer: answer
@@ -39,7 +42,8 @@ describe Rasti::AI::OpenAI::Assistant do
               role: Rasti::AI::OpenAI::Roles::USER,
               content: question
             }
-          ]
+          ],
+          response_format: nil
         }
       ]
 
@@ -107,6 +111,19 @@ describe Rasti::AI::OpenAI::Assistant do
       assert_equal answer, response
     end
 
+    it 'JSON Schema' do
+      json_schema = {answer: 'Response answer'}
+      json_answer = "{\\\"answer\\\": \\\"#{answer}\\\"}"
+
+      stub_open_ai_chat_completions question: question, answer: json_answer, json_schema: json_schema
+
+      assistant = Rasti::AI::OpenAI::Assistant.new json_schema: json_schema
+
+      response = assistant.call question
+
+      assert_equal answer, JSON.parse(response)['answer']
+    end
+
   end
 
   describe 'Tools' do
@@ -278,4 +295,4 @@ describe Rasti::AI::OpenAI::Assistant do
 
   end
 
-end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rasti-ai
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.1.0
 platform: ruby
 authors:
 - Gabriel Naiman
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-04-28 00:00:00.000000000 Z
+date: 2025-06-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: multi_require
@@ -210,7 +210,7 @@ homepage: https://github.com/gabynaiman/rasti-ai
 licenses:
 - MIT
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -225,8 +225,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.3.26
-signing_key: 
+rubygems_version: 3.0.9
+signing_key:
 specification_version: 4
 summary: AI for apps
 test_files: