ask-agent 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: cc1ea1b84a7139944acf9787cfdd3e5a9fad739b759fc4501f59f68bdd3c0b77
-  data.tar.gz: 37733f0982e0d991e284d46c9202fe5846fff67a433acae0385a703912c7f898
+  metadata.gz: bb3f240855fe04ef37b3d474cd4f9ddae5db28256673aa1d1804bdfbc1a8836c
+  data.tar.gz: 84684f83697510a261de607ede4a6aafd2a396a2151d169290769148214bc4a0
 SHA512:
-  metadata.gz: 8e4ff593f7856c2fe0da46dfe8091ad38217e29b27aa343e95f6f939ab741b8c249217dbaa397ad609c21b9b4f39e09a781738a556368e69307fa01329fb00d9
-  data.tar.gz: fc45d3ae62bc2e325a22dedab88625f312d3e9b07188d9614c5cb2ddfa9683ed31e2bb8473f10344d73ac18f7cef202fcf53cd68272affbab54c208770017dde
+  metadata.gz: 41ac967bf49f11bea0d4a65d9984ddfc0838007096b153c2294691abd8ce3bdd60aae0890e10f7bfed39ce2ed0b7f71f577e33d10a029f1660c6b779b4117f13
+  data.tar.gz: e097d8c0bbd5bf5f78cccc49fd167d951119d91ee473915803c92dfbbae245167b95050ad321ba2ccfbed71fc3c224effe30252c2badeff8c6fbb07f624c801e

data/lib/ask/agent/chat.rb

@@ -26,8 +26,11 @@ module Ask
     # internal usage (Hash of { id => ToolCallInfo }).
     class Chat
       # @return [String] model ID (e.g. "gpt-4o")
-      attr_reader :model
-
+        attr_reader :model_id
+      # @return [String] model ID (e.g. "gpt-4o")
+      def model
+        @model_id
+      end
       # @return [Array<Ask::Message>] all messages in the conversation
       attr_reader :messages
 
@@ -35,13 +38,17 @@ module Ask
       # @param tools [Array<Ask::Tool>] tool instances available to the chat
       # @param temperature [Float, nil] sampling temperature
       # @param schema [Ask::Schema, Hash, nil] structured output schema
-      def initialize(model:, tools: [], temperature: nil, schema: nil, **)
+      # @param assume_model_exists [Boolean] whether to skip model catalog lookup
+      # @param provider [String, Symbol, nil] provider slug to use (overrides model catalog)
+      def initialize(model:, tools: [], temperature: nil, schema: nil,
+                     assume_model_exists: false, provider: nil, **)
         @model_id = model.respond_to?(:id) ? model.id : model.to_s
-        @model_info = resolve_model(@model_id)
+        @model_info = resolve_model(@model_id) unless assume_model_exists
         @tools = tools
         @temperature = temperature
         @schema = schema
         @messages = []
+        @provider_override = provider
         @provider = nil
       end
 
@@ -123,6 +130,15 @@ module Ask
         self
       end
 
+      # Set the structured output schema and return self.
+      #
+      # @param schema [Ask::Schema, Hash] structured output schema
+      # @return [self]
+      def with_schema(schema)
+        @schema = schema.respond_to?(:to_json_schema) ? schema.to_json_schema : schema
+        self
+      end
+
       # Clear all messages from the conversation.
       def reset_messages!
         @messages.clear
@@ -143,19 +159,19 @@ module Ask
       end
 
       def build_provider
-        slug = @model_info&.provider || "openai"
+        slug = @provider_override&.to_s || @model_info&.provider || "openai"
         klass = Ask::Provider.resolve(slug)
         klass.new(provider_config(slug))
       end
 
       def provider_config(slug, extra_keys: {})
-          env_key = "#{slug.upcase}_API_KEY"
-          key = ENV[env_key] || ENV["OPENCODE_API_KEY"] || ENV["OPENAI_API_KEY"]
-          base = ENV["#{slug.upcase}_API_BASE"] || ENV["OPENCODE_API_BASE"]
-          config = { api_key: key }
-          config[:"#{slug}_api_key"] = key
-          config[:"#{slug}_api_base"] = base if base
-          Ask::LLM::Config.new(config)
+        env_key = "#{slug.upcase}_API_KEY"
+        key = ENV[env_key] || ENV["OPENCODE_API_KEY"] || ENV["OPENAI_API_KEY"]
+        base = ENV["#{slug.upcase}_API_BASE"] || ENV["OPENCODE_API_BASE"]
+        config = { api_key: key }
+        config[:"#{slug}_api_key"] = key
+        config[:"#{slug}_api_base"] = base if base
+        Ask::LLM::Config.new(config)
       end
 
       # Accumulate partial tool calls from streaming chunks.

data/lib/ask/agent/version.rb

@@ -2,6 +2,6 @@
 
 module Ask
   module Agent
-    VERSION = "0.1.2"
+    VERSION = "0.1.3"
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ask-agent
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - Kaka Ruto