ask-llm-providers 0.1.1 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e1c7c2b703b28c45fa414ddf18bb9aa7ddf896c00d97418c01b8f55966e61b52
-  data.tar.gz: d0c7a219f49a0f981f2991fa72e249c813450a9d84f2269478ac733bfdad608a
+  metadata.gz: 22c3960f084ce514405b73a25093a058458f9f71ec5d3bf076a62ad312c74b94
+  data.tar.gz: eaf4407367c1adc93c55c35762772d599d774c8a856967c74ae9da73870243c6
 SHA512:
-  metadata.gz: dd4dbd35bc0efe7d19a5844246c76e117601d3da4d90831eedc2e53a6345c048666b28af0d5c831dfe62557eb1434c95c51668fa010bfb676032a3159ba57195
-  data.tar.gz: 47be20c465f50f4cd586df677d5f984d14c034df110606c965085bcf2a17a05f19d45e4716eda4f41884abffb3afca622a39b6ed81961a27c4bea88754b258cf
+  metadata.gz: 9c5fc807124321bde12f4b239c0c5d32ff278edf9a446949346237cd24bd562582c9b6f0acecd27dc64be27adb2d8250eee34a4b4b64ae779c413da0e15f34d1
+  data.tar.gz: 2691464aefbae83c68ddf63c3a278e5f459d7fc222adb320d96d2c57a4cf2702ed50da1a056f8a83b22b64d7dbf804261371b7ceff7b8569799096d6b590b65f

data/lib/ask/llm/version.rb

@@ -2,6 +2,6 @@
 
 module Ask
   module LLM
-    VERSION = "0.1.1"
+    VERSION = "0.1.3"
   end
 end

data/lib/ask/provider/mimo.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+module Ask
+  module Providers
+    # Mimo API — an OpenAI-compatible provider
+    class Mimo < OpenAI
+      def api_base
+        @config.base_url || ENV["MIMO_API_BASE"] || "https://token-plan-sgp.xiaomimimo.com/v1"
+      end
+
+      def headers
+        key = @config.api_key || ENV["MIMO_API_KEY"]
+        h = { "Content-Type" => "application/json" }
+        h["Authorization"] = "Bearer #{key}" if key
+        h
+      end
+
+      class << self
+        def slug; "mimo"; end
+        def configuration_options; %i[api_key base_url]; end
+        def configuration_requirements; %i[api_key]; end
+        def configured?(config)
+          key = config.respond_to?(:api_key) ? config.api_key : nil
+          key ||= ENV["MIMO_API_KEY"]
+          key.to_s.length > 0
+        end
+      end
+    end
+  end
+end

data/lib/ask/provider/openai.rb

@@ -7,6 +7,7 @@ module Ask
     # +base_url+ override.
     class OpenAI < Ask::Provider
       def initialize(config = {})
+        @provider_keys = extract_provider_keys(config)
         config = normalize_config(config)
         super(config)
         @http = build_http
@@ -57,22 +58,31 @@ module Ask
         end
         def configuration_options; %i[api_key base_url organization_id project_id]; end
         def configuration_requirements; %i[api_key]; end
-        def configured?(config)
-          (config.respond_to?(:api_key) && !config.api_key.to_s.empty?) ||
-            (config.respond_to?(:openai_api_key) && !config.openai_api_key.to_s.empty?)
-        end
+        def assume_models_exist?; false; end
       end
 
       private
 
+      # Extract and store any provider-specific config keys (e.g., opencode_api_key).
+      # These are not part of the standard OpenAI config but are used by subclasses.
+      def extract_provider_keys(config)
+        return {} unless config.is_a?(Hash)
+        known = %i[api_key base_url organization_id project_id openai_api_key]
+        config.reject { |k, _| known.include?(k.to_sym) }
+      end
+
+      # Restore provider-specific keys after normalize_config strips standard ones.
       def normalize_config(config)
         return config if !config.is_a?(Hash)
-        Ask::LLM::Config.new(
+
+        merged = {
           api_key: config[:api_key] || config["api_key"] || config[:openai_api_key],
           base_url: config[:base_url] || config["base_url"],
           organization_id: config[:organization_id] || config["organization_id"],
           project_id: config[:project_id] || config["project_id"]
-        )
+        }.merge(@provider_keys)
+
+        Ask::LLM::Config.new(merged)
       end
 
       def build_http
@@ -140,12 +150,22 @@ module Ask
           parsed = JSON.parse(data) rescue next
           choice = parsed.dig("choices", 0) or next
           delta = choice["delta"] || {}
-          chunk = Ask::Chunk.new(content: delta["content"], tool_calls: parse_stream_tool_calls(delta["tool_calls"]), finish_reason: choice["finish_reason"], usage: parsed["usage"])
+          thinking = extract_thinking(parsed, delta)
+          chunk = Ask::Chunk.new(content: delta["content"], tool_calls: parse_stream_tool_calls(delta["tool_calls"]), finish_reason: choice["finish_reason"], usage: parsed["usage"], thinking: thinking)
           stream.add(chunk)
           yield chunk if block_given?
         end
       end
 
+      # Extract thinking/reasoning content from provider response.
+      # Some providers (Anthropic, DeepSeek) send thinking in a separate field.
+      def extract_thinking(parsed, delta)
+        delta["reasoning_content"] || delta["thinking"] ||
+          parsed.dig("choices", 0, "delta", "reasoning_content") ||
+          parsed.dig("choices", 0, "delta", "thinking") ||
+          parsed.dig("choices", 0, "reasoning_content")
+      end
+
       def parse_stream_tool_calls(calls)
         return nil unless calls&.any?
         calls.map { |tc| { id: tc["id"], name: tc.dig("function", "name"), arguments: tc.dig("function", "arguments"), index: tc["index"] } }
@@ -153,3 +173,23 @@ module Ask
     end
   end
 end
+
+# When the OpenAI provider is subclassed (e.g. OpenCode), normalize_config
+# should also check for env vars matching the subclass slug.
+def normalize_config(config)
+  return config if !config.is_a?(Hash)
+
+  slug = self.class.slug
+  env_key = ENV["#{slug.upcase}_API_KEY"]
+  env_base = ENV["#{slug.upcase}_API_BASE"]
+
+  merged = {
+    api_key: config[:api_key] || config["api_key"] || config[:"#{slug}_api_key"] || config[:"#{slug}_api_key"] || config[:openai_api_key] || env_key,
+    base_url: config[:base_url] || config["base_url"] || env_base,
+    organization_id: config[:organization_id] || config["organization_id"],
+    project_id: config[:project_id] || config["project_id"]
+  }.merge(config.reject { |k, _| %i[api_key base_url organization_id project_id openai_api_key].include?(k.to_sym) })
+
+  # Also preserve original config for subclass-specific key access
+  Ask::LLM::Config.new(merged)
+end

data/lib/ask/provider/opencode.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+module Ask
+  module Providers
+    # OpenCode API — an OpenAI-compatible provider at opencode.ai
+    class OpenCode < OpenAI
+      def api_base
+        @config.base_url || ENV["OPENCODE_API_BASE"] || "https://opencode.ai/zen/v1"
+      end
+
+      def headers
+        key = @config.api_key || ENV["OPENCODE_API_KEY"]
+        h = { "Content-Type" => "application/json" }
+        h["Authorization"] = "Bearer #{key}" if key
+        h
+      end
+
+      class << self
+        def slug; "opencode"; end
+        def configuration_options; %i[api_key base_url]; end
+        def configuration_requirements; %i[api_key]; end
+        def configured?(config)
+          key = config.respond_to?(:api_key) ? config.api_key : nil
+          key ||= ENV["OPENCODE_API_KEY"]
+          key.to_s.length > 0
+        end
+      end
+    end
+  end
+end

data/lib/ask/provider/opencode_go.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+module Ask
+  module Providers
+    # OpenCode Go API — an OpenAI-compatible provider at opencode.ai/zen/go
+    class OpenCodeGo < OpenAI
+      def api_base
+        @config.base_url || ENV["OPENCODE_GO_API_BASE"] || "https://opencode.ai/zen/go/v1"
+      end
+
+      def headers
+        key = @config.api_key || ENV["OPENCODE_API_KEY"] || ENV["OPENCODE_GO_API_KEY"]
+        h = { "Content-Type" => "application/json" }
+        h["Authorization"] = "Bearer #{key}" if key
+        h
+      end
+
+      class << self
+        def slug; "opencode_go"; end
+        def configuration_options; %i[api_key base_url]; end
+        def configuration_requirements; %i[api_key]; end
+        def configured?(config)
+          key = config.respond_to?(:api_key) ? config.api_key : nil
+          key ||= ENV["OPENCODE_API_KEY"] || ENV["OPENCODE_GO_API_KEY"]
+          key.to_s.length > 0
+        end
+      end
+    end
+  end
+end

data/lib/ask-llm-providers.rb

@@ -20,6 +20,9 @@ require_relative "ask/provider/bedrock"
 require_relative "ask/provider/ollama"
 require_relative "ask/provider/mistral"
 require_relative "ask/provider/cloudflare"
+require_relative "ask/provider/opencode"
+require_relative "ask/provider/opencode_go"
+require_relative "ask/provider/mimo"
 
 # Register providers with the Ask::Provider registry
 Ask::Provider.register(:openai, Ask::Providers::OpenAI)
@@ -47,3 +50,6 @@ Ask::Provider.register(:cloudflare, Ask::Providers::Cloudflare)
     ))
   end
 end
+
+# configured via environment variables:
+

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ask-llm-providers
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.3
 platform: ruby
 authors:
 - Kaka Ruto
@@ -183,9 +183,12 @@ files:
 - lib/ask/provider/bedrock.rb
 - lib/ask/provider/cloudflare.rb
 - lib/ask/provider/google.rb
+- lib/ask/provider/mimo.rb
 - lib/ask/provider/mistral.rb
 - lib/ask/provider/ollama.rb
 - lib/ask/provider/openai.rb
+- lib/ask/provider/opencode.rb
+- lib/ask/provider/opencode_go.rb
 homepage: https://github.com/ask-rb/ask-llm-providers
 licenses:
 - MIT