rasti-ai 3.0.0 → 3.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f40d9534d2254ebcf7efe528f42b7187841ee5eb96ff4b406fcfd6a27a7bdf79
-  data.tar.gz: 87a8eb4fbcae1b57ae77c4a9676424b80bb8fbfd8d29e61bb89425c0ed5d5d32
+  metadata.gz: 81ed3eb7239ec9d47de179db57b6d55abb21c512ee6c830de189c77692bb1bf0
+  data.tar.gz: fb25a1ab3f55861676625021c91bfed4aea5c40058c3b6282041dadd39c9fdb3
 SHA512:
-  metadata.gz: cfc8303f52540e6c6058df5d7a7fe353e5bffd88a6b9122f789066877d0b8a87e7fb5daef1c78db740e54e159c2877dd6ea6540a080eed1a4feb2a1f466326ba
-  data.tar.gz: 745f6ea9e9e2a95497514e1330feccebd3a7660709477f43fce3a6dfb03193d8f18f8012a44a976ed574dd4267d4b6979a59c970655e1a6580ef50291ccdd28d
+  metadata.gz: f3a1f07c3580de138323486de7255633eb67e1e14a3ea7ec847fededaf6f2ba771cf425d5ac27f80ed7180582e8b2070562ff42c84a1dbedbc541da9e1bb6a59
+  data.tar.gz: afa8bf99945c06a833b62ae3d53e42860b49a923715473b8a1cb8b4793dc653b9d48b93a991676d7cc79c79aa714bc2a8d6c94156c9fdb36f60bea125f660936

data/AGENTS.md

@@ -76,6 +76,14 @@ lib/
         roles.rb
         client.rb
         assistant.rb
+      open_router/
+        roles.rb
+        client.rb
+        assistant.rb
+      huawei_maas/
+        roles.rb
+        client.rb
+        assistant.rb
       mcp/
         server.rb              # Rack middleware exposing tools via JSON-RPC 2.0
         tools_registry.rb      # per-request tool registry used by the middleware
@@ -100,6 +108,12 @@ spec/
   anthropic/
     client_spec.rb
     assistant_spec.rb
+  open_router/
+    client_spec.rb
+    assistant_spec.rb
+  huawei_maas/
+    client_spec.rb
+    assistant_spec.rb
   mcp/
     client_spec.rb
     server_spec.rb
@@ -140,6 +154,19 @@ spec/
 - Gemini: renames to `parameters`
 - Anthropic: renames to `input_schema`
 
+### OpenAI-compatible providers
+
+OpenRouter and Huawei MaaS speak the OpenAI chat completions protocol verbatim — same endpoint shape (`POST /chat/completions`), same request body, same response structure, same tool calling format, same Bearer token auth. Their clients and assistants inherit from the OpenAI provider and override only `base_url`, `default_api_key`, `default_model`, and `provider_name` (client) / `build_default_client` (assistant). No template methods are reimplemented.
+
+| | OpenRouter | Huawei MaaS |
+|---|---|---|
+| Base URL | `https://openrouter.ai/api/v1` | `https://api-ap-southeast-1.modelarts-maas.com/v2` |
+| Auth | `Authorization: Bearer {key}` | `Authorization: Bearer {key}` |
+| Config | `openrouter_api_key` / `openrouter_default_model` | `huawei_maas_api_key` / `huawei_maas_default_model` |
+| `provider` in Usage | `'open_router'` | `'huawei_maas'` |
+
+Thinking is inherited too: `reasoning_effort` is passed through as-is (same as OpenAI).
+
 
 ## Thinking levels
 
@@ -153,6 +180,8 @@ The base `Assistant` accepts `thinking: 'low' | 'medium' | 'high'` (validated on
 
 For Gemini, `thinking_config` goes inside `generation_config` — the client doesn't need a new param. For OpenAI and Anthropic, it's a separate top-level param in the client method (`reasoning_effort:` and `thinking:` respectively).
 
+OpenRouter and Huawei MaaS inherit OpenAI's behavior: `thinking` is passed through as `reasoning_effort` with the same string values (`'low'`/`'medium'`/`'high'`). No `thinking_config` override is needed. Note that Huawei's models typically accept only `'high'` and `'max'` — `'high'` is the safe choice, and `'max'` is not expressible through the gem's universal levels.
+
 The loop does not change. Anthropic thinking blocks (`type: 'thinking'`) in responses are ignored by `parse_content` (looks for `type == 'text'`) and preserved automatically by `build_assistant_tool_calls_message` (passes full `response['content']` array).
 
 
@@ -189,6 +218,17 @@ config.<provider>_api_key       = 'test_<provider>_api_key'
 config.<provider>_default_model = '<provider>-test'
 ```
 
+### OpenAI-compatible providers
+
+If the new provider speaks the OpenAI chat completions protocol (same endpoint, request body, response shape, tool calling, Bearer auth), inherit from `Rasti::AI::OpenAI::Client` and `Rasti::AI::OpenAI::Assistant` instead of the abstract bases. The assistant only overrides `build_default_client`; the client overrides `base_url`, `default_api_key`, `default_model`, and `provider_name`. No template methods are reimplemented.
+
+The OpenAI client exposes two private hooks that make this possible without duplicating `chat_completions` or `parse_usage`:
+
+- **`default_model`** — returns the provider's default model from `Rasti::AI` config
+- **`provider_name`** — the string written to `Usage#provider`
+
+See `open_router/` and `huawei_maas/` for reference implementations.
+
 ### ⚠️ multi_require load order
 
 `require_relative_pattern 'ai/**/*'` loads files alphabetically. If the new provider name sorts before `assistant` or `client` (e.g. `anthropic` < `assistant`), the subclass is loaded before the base class and raises `NameError`.
@@ -203,9 +243,14 @@ require_relative 'ai/tool'
 require_relative 'ai/tool_serializer'
 require_relative 'ai/client'
 require_relative 'ai/assistant'
+require_relative 'ai/open_ai/roles'
+require_relative 'ai/open_ai/client'
+require_relative 'ai/open_ai/assistant'
 require_relative_pattern 'ai/**/*'   # duplicates are skipped by Ruby's require
 ```
 
+The OpenAI provider is also explicitly required because OpenAI-compatible providers (OpenRouter, Huawei MaaS) inherit from it, and some of their directory names sort before `open_ai` alphabetically (e.g. `huawei_maas` < `open_ai`).
+
 If you add a provider whose name sorts before `client` alphabetically, the same mechanism protects it.
 
 
@@ -519,6 +564,8 @@ It's a simple class (not a `Tool` subclass) that returns a plain string — cove
   rake assistant:openai      # OPENAI_API_KEY
   rake assistant:gemini      # GEMINI_API_KEY
   rake assistant:anthropic   # ANTHROPIC_API_KEY
+  rake assistant:openrouter  # OPENROUTER_API_KEY
+  rake assistant:huawei_maas # HUAWEI_MAAS_API_KEY
   ```
   Each task validates the key, writes logs to `log/<provider>.log`, connects to the [Pipeworx](https://pipeworx.io) public weather MCP server, and starts a `You:` / `Assistant:` prompt loop (`exit` or `Ctrl+C` to quit). The model can be overridden with the matching env variable (e.g. `OPENAI_DEFAULT_MODEL=gpt-4o`).
 

data/README.md

@@ -45,6 +45,14 @@ Rasti::AI.configure do |config|
   config.anthropic_api_key = 'sk-ant-12345' # Default ENV['ANTHROPIC_API_KEY']
   config.anthropic_default_model = 'claude-opus-4-5' # Default ENV['ANTHROPIC_DEFAULT_MODEL']
 
+  # OpenRouter
+  config.openrouter_api_key = 'sk-or-12345' # Default ENV['OPENROUTER_API_KEY']
+  config.openrouter_default_model = 'openai/gpt-4o-mini' # Default ENV['OPENROUTER_DEFAULT_MODEL']
+
+  # Huawei MaaS
+  config.huawei_maas_api_key = 'sk-maas-12345' # Default ENV['HUAWEI_MAAS_API_KEY']
+  config.huawei_maas_default_model = 'DeepSeek-V3' # Default ENV['HUAWEI_MAAS_DEFAULT_MODEL']
+
   # Usage tracking
   config.usage_tracker = ->(usage) { puts "#{usage.provider}: #{usage.input_tokens} in / #{usage.output_tokens} out" }
 end
@@ -55,8 +63,12 @@ end
 - **OpenAI** - `Rasti::AI::OpenAI::Assistant`
 - **Gemini** - `Rasti::AI::Gemini::Assistant`
 - **Anthropic** - `Rasti::AI::Anthropic::Assistant`
+- **OpenRouter** - `Rasti::AI::OpenRouter::Assistant`
+- **Huawei MaaS** - `Rasti::AI::HuaweiMaaS::Assistant`
+
+All providers share the same interface. The examples below use OpenAI, but apply equally to any provider by replacing `OpenAI` with the provider name.
 
-All providers share the same interface. The examples below use OpenAI, but apply equally to Gemini or Anthropic by replacing `OpenAI` with the provider name.
+OpenRouter and Huawei MaaS are [OpenAI-compatible](https://platform.openai.com/docs/api-reference/chat) APIs — they use the same chat completions format, tool calling, and Bearer token auth. Their clients and assistants inherit from the OpenAI provider, so everything in the examples below works identically.
 
 ### Assistant
 
@@ -428,9 +440,11 @@ assistant.call 'What is 5 plus 3?'
 The gem includes interactive chat tasks wired to the [Pipeworx](https://pipeworx.io) public weather MCP server (no auth required):
 
 ```bash
-OPENAI_API_KEY=sk-...    rake assistant:openai
-GEMINI_API_KEY=AIza...   rake assistant:gemini
-ANTHROPIC_API_KEY=sk-... rake assistant:anthropic
+OPENAI_API_KEY=sk-...       rake assistant:openai
+GEMINI_API_KEY=AIza...      rake assistant:gemini
+ANTHROPIC_API_KEY=sk-...    rake assistant:anthropic
+OPENROUTER_API_KEY=sk-or... rake assistant:openrouter
+HUAWEI_MAAS_API_KEY=sk-...  rake assistant:huawei_maas
 ```
 
 Type your message and press Enter. Type `exit` or `Ctrl+C` to quit.

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

@@ -0,0 +1,15 @@
+module Rasti
+  module AI
+    module HuaweiMaaS
+      class Assistant < Rasti::AI::OpenAI::Assistant
+
+        private
+
+        def build_default_client
+          Client.new
+        end
+
+      end
+    end
+  end
+end

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

@@ -0,0 +1,27 @@
+module Rasti
+  module AI
+    module HuaweiMaaS
+      class Client < Rasti::AI::OpenAI::Client
+
+        private
+
+        def default_api_key
+          Rasti::AI.huawei_maas_api_key
+        end
+
+        def default_model
+          Rasti::AI.huawei_maas_default_model
+        end
+
+        def provider_name
+          'huawei_maas'
+        end
+
+        def base_url
+          'https://api-ap-southeast-1.modelarts-maas.com/v2'
+        end
+
+      end
+    end
+  end
+end

data/lib/rasti/ai/huawei_maas/roles.rb

@@ -0,0 +1,14 @@
+module Rasti
+  module AI
+    module HuaweiMaaS
+      module Roles
+
+        ASSISTANT = 'assistant'.freeze
+        SYSTEM = 'system'.freeze
+        TOOL = 'tool'.freeze
+        USER = 'user'.freeze
+
+      end
+    end
+  end
+end

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

@@ -5,7 +5,7 @@ module Rasti
 
         def chat_completions(messages:, model:nil, tools:[], response_format:nil, reasoning_effort:nil)
           body = {
-            model: model || Rasti::AI.openai_default_model,
+            model: model || default_model,
             messages: messages,
             tools: tools,
             tool_choice: tools.empty? ? 'none' : 'auto'
@@ -23,7 +23,7 @@ module Rasti
           usage = response['usage']
           return unless usage
           Usage.new(
-            provider: 'open_ai',
+            provider: provider_name,
             model: response['model'],
             input_tokens: usage['prompt_tokens'],
             output_tokens: usage['completion_tokens'],
@@ -37,6 +37,14 @@ module Rasti
           Rasti::AI.openai_api_key
         end
 
+        def default_model
+          Rasti::AI.openai_default_model
+        end
+
+        def provider_name
+          'open_ai'
+        end
+
         def base_url
           'https://api.openai.com/v1'
         end

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

@@ -0,0 +1,15 @@
+module Rasti
+  module AI
+    module OpenRouter
+      class Assistant < Rasti::AI::OpenAI::Assistant
+
+        private
+
+        def build_default_client
+          Client.new
+        end
+
+      end
+    end
+  end
+end

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

@@ -0,0 +1,27 @@
+module Rasti
+  module AI
+    module OpenRouter
+      class Client < Rasti::AI::OpenAI::Client
+
+        private
+
+        def default_api_key
+          Rasti::AI.openrouter_api_key
+        end
+
+        def default_model
+          Rasti::AI.openrouter_default_model
+        end
+
+        def provider_name
+          'open_router'
+        end
+
+        def base_url
+          'https://openrouter.ai/api/v1'
+        end
+
+      end
+    end
+  end
+end

data/lib/rasti/ai/open_router/roles.rb

@@ -0,0 +1,14 @@
+module Rasti
+  module AI
+    module OpenRouter
+      module Roles
+
+        ASSISTANT = 'assistant'.freeze
+        SYSTEM = 'system'.freeze
+        TOOL = 'tool'.freeze
+        USER = 'user'.freeze
+
+      end
+    end
+  end
+end

data/lib/rasti/ai/tool_serializer.rb

@@ -66,7 +66,7 @@ module Rasti
           when :integer         then {type: 'integer'}
           when :float           then {type: 'number'}
           when :boolean         then {type: 'boolean'}
-          when :time            then {type: 'string', format: 'date'}
+          when :time            then json_schema_for_time_format(type_hash[:format])
           when :enum            then {type: 'string', enum: type_hash[:values]}
           when :array           then {type: 'array', items: json_schema_for_type(type_hash[:items])}
           when :model           then json_schema_from_model_schema(type_hash[:schema])
@@ -79,6 +79,21 @@ module Rasti
           description.split("\n").map(&:strip).join(' ').strip
         end
 
+        def json_schema_for_time_format(format)
+          return {type: 'string'} unless format.is_a?(::String)
+
+          has_date = format.include?('%F') || format.include?('%Y')
+          has_time = format.include?('%T') || format.include?('%H')
+
+          if has_date && has_time
+            {type: 'string', format: 'date-time'}
+          elsif has_date
+            {type: 'string', format: 'date'}
+          else
+            {type: 'string'}
+          end
+        end
+
       end
     end
   end

data/lib/rasti/ai/version.rb

@@ -1,5 +1,5 @@
 module Rasti
   module AI
-    VERSION = '3.0.0'
+    VERSION = '3.1.0'
   end
 end

data/lib/rasti/ai.rb

@@ -21,6 +21,7 @@ module Rasti
     require_relative 'ai/tool_serializer'
     require_relative 'ai/client'
     require_relative 'ai/assistant'
+    require_relative_pattern 'ai/open_ai/*'
     require_relative_pattern 'ai/**/*'
 
     attr_config :logger, Logger.new(STDOUT)
@@ -38,7 +39,13 @@ module Rasti
     attr_config :anthropic_api_key, ENV['ANTHROPIC_API_KEY']
     attr_config :anthropic_default_model, ENV['ANTHROPIC_DEFAULT_MODEL']
 
+    attr_config :openrouter_api_key, ENV['OPENROUTER_API_KEY']
+    attr_config :openrouter_default_model, ENV['OPENROUTER_DEFAULT_MODEL']
+
+    attr_config :huawei_maas_api_key, ENV['HUAWEI_MAAS_API_KEY']
+    attr_config :huawei_maas_default_model, ENV['HUAWEI_MAAS_DEFAULT_MODEL']
+
     attr_config :usage_tracker, nil
 
   end
-end
+end

data/spec/huawei_maas/assistant_spec.rb

@@ -0,0 +1,80 @@
+require 'minitest_helper'
+
+describe Rasti::AI::HuaweiMaaS::Assistant do
+
+  let(:api_url) { 'https://api-ap-southeast-1.modelarts-maas.com/v2/chat/completions' }
+
+  let(:question) { 'How many goals has Messi scored for Barca?' }
+
+  let(:answer) { 'Lionel Messi scored 672 goals in 778 official matches for FC Barcelona.' }
+
+  it 'Default' do
+    stub_request(:post, api_url)
+      .with(body: read_resource('open_ai/basic_request.json', model: Rasti::AI.huawei_maas_default_model, prompt: question))
+      .to_return(body: read_resource('open_ai/basic_response.json', content: answer))
+
+    assistant = Rasti::AI::HuaweiMaaS::Assistant.new
+
+    response = assistant.call question
+
+    assert_equal answer, response
+  end
+
+  describe 'Tools' do
+
+    let(:client) { Minitest::Mock.new }
+
+    let(:tool_response) do
+      read_json_resource(
+        'open_ai/tool_response.json',
+        name: 'goals_by_player',
+        arguments: {player: 'Lionel Messi', team: 'Barcelona'}
+      )
+    end
+
+    let(:tool_result) { '672' }
+
+    let(:answer_with_tool) { 'Lionel Messi scored 672 goals for FC Barcelona.' }
+
+    def basic_response(content)
+      read_json_resource('open_ai/basic_response.json', content: content)
+    end
+
+    def stub_client_request(role:, content:, response:, tools:[])
+      serialized_tools = tools.map do |tool|
+        {type: 'function', function: Rasti::AI::ToolSerializer.serialize(tool.class)}
+      end
+
+      client.expect :chat_completions, response do |params|
+        last_message = params[:messages].last
+        last_message[:role] == role &&
+          last_message[:content] == content &&
+          params[:tools] == serialized_tools
+      end
+    end
+
+    it 'Call tool' do
+      tool = GoalsByPlayer.new
+
+      stub_client_request role: Rasti::AI::HuaweiMaaS::Roles::USER,
+                          content: question,
+                          tools: [tool],
+                          response: tool_response
+
+      stub_client_request role: Rasti::AI::HuaweiMaaS::Roles::TOOL,
+                          content: tool_result,
+                          tools: [tool],
+                          response: basic_response(answer_with_tool)
+
+      assistant = Rasti::AI::HuaweiMaaS::Assistant.new client: client, tools: [tool]
+
+      response = assistant.call question
+
+      assert_equal answer_with_tool, response
+
+      client.verify
+    end
+
+  end
+
+end

data/spec/huawei_maas/client_spec.rb

@@ -0,0 +1,60 @@
+require 'minitest_helper'
+
+describe Rasti::AI::HuaweiMaaS::Client do
+
+  let(:api_url) { 'https://api-ap-southeast-1.modelarts-maas.com/v2/chat/completions' }
+
+  let(:question) { 'who is Messi?' }
+
+  let(:answer) { 'Lionel Messi is the best player ever' }
+
+  def user_message(content)
+    {role: Rasti::AI::HuaweiMaaS::Roles::USER, content: content}
+  end
+
+  it 'Basic message with correct base URL and auth header' do
+    stub_request(:post, api_url)
+      .with(
+        headers: {'Authorization' => "Bearer #{Rasti::AI.huawei_maas_api_key}"},
+        body: read_resource('open_ai/basic_request.json', model: Rasti::AI.huawei_maas_default_model, prompt: question)
+      )
+      .to_return(body: read_resource('open_ai/basic_response.json', content: answer))
+
+    client = Rasti::AI::HuaweiMaaS::Client.new
+
+    response = client.chat_completions messages: [user_message(question)]
+
+    assert_equal answer, response.dig('choices', 0, 'message', 'content')
+  end
+
+  it 'Usage tracking with provider name' do
+    stub_request(:post, api_url)
+      .to_return(body: read_resource('open_ai/basic_response.json', content: answer))
+
+    tracked = []
+    tracker = ->(usage) { tracked << usage }
+
+    client = Rasti::AI::HuaweiMaaS::Client.new usage_tracker: tracker
+
+    client.chat_completions messages: [user_message(question)]
+
+    assert_equal 1, tracked.count
+    assert_instance_of Rasti::AI::Usage, tracked[0]
+    assert_equal 'huawei_maas', tracked[0].provider
+  end
+
+  it 'Custom model' do
+    model = SecureRandom.uuid
+
+    stub_request(:post, api_url)
+      .with(body: read_resource('open_ai/basic_request.json', model: model, prompt: question))
+      .to_return(body: read_resource('open_ai/basic_response.json', content: answer))
+
+    client = Rasti::AI::HuaweiMaaS::Client.new
+
+    response = client.chat_completions messages: [user_message(question)], model: model
+
+    assert_equal answer, response.dig('choices', 0, 'message', 'content')
+  end
+
+end

data/spec/minitest_helper.rb

@@ -25,6 +25,12 @@ Rasti::AI.configure do |config|
 
   config.anthropic_api_key = 'test_anthropic_api_key'
   config.anthropic_default_model = 'claude-test'
+
+  config.openrouter_api_key = 'test_openrouter_api_key'
+  config.openrouter_default_model = 'openrouter-test'
+
+  config.huawei_maas_api_key = 'test_huawei_maas_api_key'
+  config.huawei_maas_default_model = 'huawei-test'
 end
 
 

data/spec/open_router/assistant_spec.rb

@@ -0,0 +1,80 @@
+require 'minitest_helper'
+
+describe Rasti::AI::OpenRouter::Assistant do
+
+  let(:api_url) { 'https://openrouter.ai/api/v1/chat/completions' }
+
+  let(:question) { 'How many goals has Messi scored for Barca?' }
+
+  let(:answer) { 'Lionel Messi scored 672 goals in 778 official matches for FC Barcelona.' }
+
+  it 'Default' do
+    stub_request(:post, api_url)
+      .with(body: read_resource('open_ai/basic_request.json', model: Rasti::AI.openrouter_default_model, prompt: question))
+      .to_return(body: read_resource('open_ai/basic_response.json', content: answer))
+
+    assistant = Rasti::AI::OpenRouter::Assistant.new
+
+    response = assistant.call question
+
+    assert_equal answer, response
+  end
+
+  describe 'Tools' do
+
+    let(:client) { Minitest::Mock.new }
+
+    let(:tool_response) do
+      read_json_resource(
+        'open_ai/tool_response.json',
+        name: 'goals_by_player',
+        arguments: {player: 'Lionel Messi', team: 'Barcelona'}
+      )
+    end
+
+    let(:tool_result) { '672' }
+
+    let(:answer_with_tool) { 'Lionel Messi scored 672 goals for FC Barcelona.' }
+
+    def basic_response(content)
+      read_json_resource('open_ai/basic_response.json', content: content)
+    end
+
+    def stub_client_request(role:, content:, response:, tools:[])
+      serialized_tools = tools.map do |tool|
+        {type: 'function', function: Rasti::AI::ToolSerializer.serialize(tool.class)}
+      end
+
+      client.expect :chat_completions, response do |params|
+        last_message = params[:messages].last
+        last_message[:role] == role &&
+          last_message[:content] == content &&
+          params[:tools] == serialized_tools
+      end
+    end
+
+    it 'Call tool' do
+      tool = GoalsByPlayer.new
+
+      stub_client_request role: Rasti::AI::OpenRouter::Roles::USER,
+                          content: question,
+                          tools: [tool],
+                          response: tool_response
+
+      stub_client_request role: Rasti::AI::OpenRouter::Roles::TOOL,
+                          content: tool_result,
+                          tools: [tool],
+                          response: basic_response(answer_with_tool)
+
+      assistant = Rasti::AI::OpenRouter::Assistant.new client: client, tools: [tool]
+
+      response = assistant.call question
+
+      assert_equal answer_with_tool, response
+
+      client.verify
+    end
+
+  end
+
+end

data/spec/open_router/client_spec.rb

@@ -0,0 +1,60 @@
+require 'minitest_helper'
+
+describe Rasti::AI::OpenRouter::Client do
+
+  let(:api_url) { 'https://openrouter.ai/api/v1/chat/completions' }
+
+  let(:question) { 'who is Messi?' }
+
+  let(:answer) { 'Lionel Messi is the best player ever' }
+
+  def user_message(content)
+    {role: Rasti::AI::OpenRouter::Roles::USER, content: content}
+  end
+
+  it 'Basic message with correct base URL and auth header' do
+    stub_request(:post, api_url)
+      .with(
+        headers: {'Authorization' => "Bearer #{Rasti::AI.openrouter_api_key}"},
+        body: read_resource('open_ai/basic_request.json', model: Rasti::AI.openrouter_default_model, prompt: question)
+      )
+      .to_return(body: read_resource('open_ai/basic_response.json', content: answer))
+
+    client = Rasti::AI::OpenRouter::Client.new
+
+    response = client.chat_completions messages: [user_message(question)]
+
+    assert_equal answer, response.dig('choices', 0, 'message', 'content')
+  end
+
+  it 'Usage tracking with provider name' do
+    stub_request(:post, api_url)
+      .to_return(body: read_resource('open_ai/basic_response.json', content: answer))
+
+    tracked = []
+    tracker = ->(usage) { tracked << usage }
+
+    client = Rasti::AI::OpenRouter::Client.new usage_tracker: tracker
+
+    client.chat_completions messages: [user_message(question)]
+
+    assert_equal 1, tracked.count
+    assert_instance_of Rasti::AI::Usage, tracked[0]
+    assert_equal 'open_router', tracked[0].provider
+  end
+
+  it 'Custom model' do
+    model = SecureRandom.uuid
+
+    stub_request(:post, api_url)
+      .with(body: read_resource('open_ai/basic_request.json', model: model, prompt: question))
+      .to_return(body: read_resource('open_ai/basic_response.json', content: answer))
+
+    client = Rasti::AI::OpenRouter::Client.new
+
+    response = client.chat_completions messages: [user_message(question)], model: model
+
+    assert_equal answer, response.dig('choices', 0, 'message', 'content')
+  end
+
+end

data/spec/tool_serializer_spec.rb

@@ -115,7 +115,7 @@ describe Rasti::AI::ToolSerializer do
         tool_class.verify
       end
 
-      it 'Time' do
+      it 'Time with date and time' do
         form_class = Rasti::Form[timestamp: Rasti::Types::Time['%Y-%m-%dT%H:%M:%S%z']]
 
         tool_class = build_tool_class form_class
@@ -123,7 +123,37 @@ describe Rasti::AI::ToolSerializer do
         serialization = serializer.serialize tool_class
 
         expeted_serialization = build_serializaton param_name: 'timestamp', param_type: 'string'
-        expeted_serialization[:inputSchema][:properties][:timestamp][:format] = 'date'
+        expeted_serialization[:inputSchema][:properties][:timestamp][:format] = 'date-time'
+
+        assert_equal expeted_serialization, serialization
+
+        tool_class.verify
+      end
+
+      it 'Time with milliseconds and timezone' do
+        form_class = Rasti::Form[started_at: Rasti::Types::Time['%FT%T.%L%z']]
+
+        tool_class = build_tool_class form_class
+
+        serialization = serializer.serialize tool_class
+
+        expeted_serialization = build_serializaton param_name: 'started_at', param_type: 'string'
+        expeted_serialization[:inputSchema][:properties][:started_at][:format] = 'date-time'
+
+        assert_equal expeted_serialization, serialization
+
+        tool_class.verify
+      end
+
+      it 'Time with date only' do
+        form_class = Rasti::Form[day: Rasti::Types::Time['%Y-%m-%d']]
+
+        tool_class = build_tool_class form_class
+
+        serialization = serializer.serialize tool_class
+
+        expeted_serialization = build_serializaton param_name: 'day', param_type: 'string'
+        expeted_serialization[:inputSchema][:properties][:day][:format] = 'date'
 
         assert_equal expeted_serialization, serialization
 

data/tasks/assistant.rake

@@ -19,9 +19,11 @@ require 'logger'
 WEATHER_MCP_URL = 'https://gateway.pipeworx.io/weather/mcp'.freeze
 
 PROVIDERS = {
-  'openai'    => {key: 'OPENAI_API_KEY',    klass: -> { Rasti::AI::OpenAI::Assistant }},
-  'gemini'    => {key: 'GEMINI_API_KEY',    klass: -> { Rasti::AI::Gemini::Assistant }},
-  'anthropic' => {key: 'ANTHROPIC_API_KEY', klass: -> { Rasti::AI::Anthropic::Assistant }}
+  'openai'     => {key: 'OPENAI_API_KEY',       klass: -> { Rasti::AI::OpenAI::Assistant }},
+  'gemini'     => {key: 'GEMINI_API_KEY',       klass: -> { Rasti::AI::Gemini::Assistant }},
+  'anthropic'  => {key: 'ANTHROPIC_API_KEY',    klass: -> { Rasti::AI::Anthropic::Assistant }},
+  'openrouter' => {key: 'OPENROUTER_API_KEY',   klass: -> { Rasti::AI::OpenRouter::Assistant }},
+  'huawei_maas'=> {key: 'HUAWEI_MAAS_API_KEY',  klass: -> { Rasti::AI::HuaweiMaaS::Assistant }}
 }.freeze
 
 def build_weather_mcp

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rasti-ai
 version: !ruby/object:Gem::Version
-  version: 3.0.0
+  version: 3.1.0
 platform: ruby
 authors:
 - Gabriel Naiman
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2026-06-23 00:00:00.000000000 Z
+date: 2026-07-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: multi_require
@@ -260,6 +260,9 @@ files:
 - lib/rasti/ai/gemini/assistant.rb
 - lib/rasti/ai/gemini/client.rb
 - lib/rasti/ai/gemini/roles.rb
+- lib/rasti/ai/huawei_maas/assistant.rb
+- lib/rasti/ai/huawei_maas/client.rb
+- lib/rasti/ai/huawei_maas/roles.rb
 - lib/rasti/ai/mcp/client.rb
 - lib/rasti/ai/mcp/constants.rb
 - lib/rasti/ai/mcp/server.rb
@@ -267,6 +270,9 @@ files:
 - lib/rasti/ai/open_ai/assistant.rb
 - lib/rasti/ai/open_ai/client.rb
 - lib/rasti/ai/open_ai/roles.rb
+- lib/rasti/ai/open_router/assistant.rb
+- lib/rasti/ai/open_router/client.rb
+- lib/rasti/ai/open_router/roles.rb
 - lib/rasti/ai/tool.rb
 - lib/rasti/ai/tool_serializer.rb
 - lib/rasti/ai/usage.rb
@@ -278,12 +284,16 @@ files:
 - spec/coverage_helper.rb
 - spec/gemini/assistant_spec.rb
 - spec/gemini/client_spec.rb
+- spec/huawei_maas/assistant_spec.rb
+- spec/huawei_maas/client_spec.rb
 - spec/mcp/client_spec.rb
 - spec/mcp/server_spec.rb
 - spec/mcp/tools_registry_spec.rb
 - spec/minitest_helper.rb
 - spec/open_ai/assistant_spec.rb
 - spec/open_ai/client_spec.rb
+- spec/open_router/assistant_spec.rb
+- spec/open_router/client_spec.rb
 - spec/resources/anthropic/basic_request.json
 - spec/resources/anthropic/basic_response.json
 - spec/resources/anthropic/tool_request.json
@@ -329,12 +339,16 @@ test_files:
 - spec/coverage_helper.rb
 - spec/gemini/assistant_spec.rb
 - spec/gemini/client_spec.rb
+- spec/huawei_maas/assistant_spec.rb
+- spec/huawei_maas/client_spec.rb
 - spec/mcp/client_spec.rb
 - spec/mcp/server_spec.rb
 - spec/mcp/tools_registry_spec.rb
 - spec/minitest_helper.rb
 - spec/open_ai/assistant_spec.rb
 - spec/open_ai/client_spec.rb
+- spec/open_router/assistant_spec.rb
+- spec/open_router/client_spec.rb
 - spec/resources/anthropic/basic_request.json
 - spec/resources/anthropic/basic_response.json
 - spec/resources/anthropic/tool_request.json