rasti-ai 1.0.1 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9410e72e90cbc69453db4573c4ea4f611719feafc816c389f0490fe003f5f7ee
-  data.tar.gz: 3e5776057dd26b4adf2a138d94511a79a88fa55418ca0b2c834aeb62ada9d70f
+  metadata.gz: 3bd63fcf9d5f5af5cd58627893322ee4f09c3a5d45c489865ff3cebe9c11aaf3
+  data.tar.gz: ed2992eabeb5a0c0fe429d8c9dbaa96036658cf3c27540b4069e80e5ee6b83df
 SHA512:
-  metadata.gz: 889f7e24c2438cba3e7f434e6d1578c734a0bbc6b2ef867da7ee8a74a69d6b6451c4c20da3bc34944f4b2113d679991fb3aa40f761e285266ad6d3cb4a0881ae
-  data.tar.gz: ddafe4e189697dfcc5980da19bb9339387867f1ceacab79fa4f74aba50ebf84d92754863c066e2db86586d7dd637075ce7fd36ced4ea948ca59c67b09f88aba7
+  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/version.rb

@@ -1,5 +1,5 @@
 module Rasti
   module AI
-    VERSION = '1.0.1'
+    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.1
+  version: 1.1.0
 platform: ruby
 authors:
 - Gabriel Naiman
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-05-26 00:00:00.000000000 Z
+date: 2025-06-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: multi_require