omnibot-ruby 0.2.0 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6a9e962ec1fc30233e103cc753f9db7db243db279f3bb51f6fa8a7bf5abdd940
-  data.tar.gz: 73f4531396f0d8a73a7e6302195a6ff42a9d70ab9b3436a0c9c877a5f234af85
+  metadata.gz: 3e09a524c80d29220a03ae50ab6a48a60646fa9ba5ab9b19f574bf75454a423c
+  data.tar.gz: 1b143d6d5928d2e987b450b5d4c314e3b01b4d2dad6f310fd9129a7d23ab02d3
 SHA512:
-  metadata.gz: d9373a4af7299c9a2ef450b93242c387cefd23ec8efd21b9eeec30dc51e1f7c003848b861e1ae3e52d6698800725e419ad1b09db0256f7bb57a5445ee7e25bdf
-  data.tar.gz: 9b24f2b865bee9764f474e7613e91d14d09c9afc9144175caf48d2a996294588ca501c47bfb5e48ed554a42189f5cfff7fcba76497e963c6b8fd6f6dde69c776
+  metadata.gz: 64417bee74cbe85d183c62f1c7f8cae0f9e4e4ee2c1c266f46560985af130fbb0057cf120c1cdc36785fbe008c996365a22bdc74b089687df6506948088b5df9
+  data.tar.gz: db6b5bfb1f62029572f1aad07f32057c9b7a96760eebeacc3ac76159479c71fba6339161db775b8e95bfc9ace70b850cf419cf1c76f06cbab9c70f5bc8275958

data/README.md

@@ -58,6 +58,7 @@ Semantics worth knowing:
 - `instructions` support `{{var}}` interpolation from `context`; a missing variable raises `KeyError`.
 - `history` is a plain array of `{ role:, content: }` hashes (or anything that responds to `#role`/`#content`) — the gem never persists conversations itself.
 - A custom `Omnibot.chat_factory` lambda must accept extra kwargs (e.g. `->(model:, **) { ... }`) — Agent always calls it with `agent_class:` in addition to `model:`.
+- Per-agent factories (v0.2.1): declare `chat_factory ->(model:, **) { RubyLLM.chat(model: model).with_temperature(0.9) }` inside an agent class to customize chat construction for that agent only (inherited by subclasses; overridable). Precedence: `Omnibot::Testing.fake!` > class-level `chat_factory` > global `Omnibot.chat_factory` — so specs stay offline even for agents with custom factories.
 - Block-tool params are always JSON type `"string"` — models send `"123"`, not `123`. Declare param types via class-form tools (`param :n, type: "integer"`) when types matter.
 - Class-form tools must declare `param` explicitly — the wrapper that adds error capture shadows `#execute`'s signature, so ruby_llm's automatic param inference doesn't see your keyword args:
 

data/lib/omnibot/agent.rb

@@ -28,6 +28,18 @@ module Omnibot
       def fast_paths = @fast_paths ||= []
       def fast_path(&block) = fast_paths << block
 
+      # Per-agent chat factory (e.g. to set temperature/params for THIS agent
+      # only). Precedence: Omnibot::Testing.fake! override > class factory >
+      # Omnibot.chat_factory. Accepts a callable or a block.
+      def chat_factory(callable = nil, &block)
+        factory = callable || block
+        factory ? @chat_factory = factory : @chat_factory
+      end
+
+      def resolved_chat_factory
+        Omnibot.chat_factory_override || chat_factory || Omnibot.chat_factory
+      end
+
       def inherited(subclass)
         super
         subclass.instance_variable_set(:@model, @model)
@@ -35,6 +47,7 @@ module Omnibot
         subclass.instance_variable_set(:@max_turns, @max_turns)
         subclass.instance_variable_set(:@tools, tools.dup)
         subclass.instance_variable_set(:@fast_paths, fast_paths.dup)
+        subclass.instance_variable_set(:@chat_factory, @chat_factory)
       end
 
       def run(message, history: [], context: {}, stream: nil)
@@ -69,7 +82,7 @@ module Omnibot
     def reply(text) = throw(FAST_REPLY, text)
 
     def extract(input, schema:)
-      chat = Omnibot.chat_factory.call(model: self.class.model, agent_class: self.class)
+      chat = self.class.resolved_chat_factory.call(model: self.class.model, agent_class: self.class)
       chat.with_instructions(interpolated_instructions) if self.class.instructions
       chat.with_schema(schema)
 
@@ -137,7 +150,7 @@ module Omnibot
     end
 
     def build_chat
-      chat = Omnibot.chat_factory.call(model: self.class.model, agent_class: self.class)
+      chat = self.class.resolved_chat_factory.call(model: self.class.model, agent_class: self.class)
       chat.with_instructions(interpolated_instructions) if self.class.instructions
       attach_history(chat)
       tools = tools_for(context).map { |t| t.is_a?(Class) ? t.new(context) : t }

data/lib/omnibot/config.rb

@@ -17,5 +17,9 @@ module Omnibot
       @chat_factory ||= ->(model:, **) { RubyLLM.chat(model: model) }
     end
     attr_writer :chat_factory
+
+    # Set by Omnibot::Testing.fake! — beats per-agent factories so specs
+    # stay offline even for agents that declare their own chat_factory.
+    attr_accessor :chat_factory_override
   end
 end

data/lib/omnibot/testing.rb

@@ -7,14 +7,19 @@ module Omnibot
     class << self
       def fake!
         @original_factory ||= Omnibot.chat_factory
-        Omnibot.chat_factory = ->(model:, agent_class: nil, **) {
+        fake = ->(model:, agent_class: nil, **) {
           FakeChat.new(agent_class: agent_class)
         }
+        Omnibot.chat_factory = fake
+        # Override beats per-agent chat_factory declarations, so specs stay
+        # offline even for agents that configure their own factory.
+        Omnibot.chat_factory_override = fake
       end
 
       def reset!
         Omnibot.chat_factory = @original_factory if @original_factory
         @original_factory = nil
+        Omnibot.chat_factory_override = nil
         scripts.clear
       end
 

data/lib/omnibot/version.rb

@@ -1,3 +1,3 @@
 module Omnibot
-  VERSION = "0.2.0"
+  VERSION = "0.2.1"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: omnibot-ruby
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.2.1
 platform: ruby
 authors:
 - Johannes Dwicahyo