helicone-rb 0.1.2 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 22d855f83be3572fa91710cb1a63483ac70e07717b9a77103ca752d71931ad09
-  data.tar.gz: 26a6eca90b9d1c9ff1cee79ae8a774da5a65fec8088c30bae1956c493242e336
+  metadata.gz: 5102ee7eb685c57698bb42b3a1818b1690b634a1f04fe1a2d5689c7d8b24c423
+  data.tar.gz: 0a37bb68ba76a4b51c8472d96d01cde87fd98df5e35d21b1d6519cb6b3991a9d
 SHA512:
-  metadata.gz: de7718c83e259ded888212a02c63f2cac6f02eb457ded45361efa1d903a2490a0774425dccab4861da8dd40d48888ab3a156ab769dac1c8f64d42e59b427aa41
-  data.tar.gz: 60bc8baed47ae2599f8a7f7f5ecfeca1e791b9b1ddf1123a61e99d7e2f1368254dfeef3a29d0d3941f27c1b1215fe1ac28220ce5ec74c292e610ea9c132d6763
+  metadata.gz: 8a801ebeb724ee8428a7c6ab73900a7dddabf1ad7524efe19391e431fc2bdb4bffe91bc9053c08c31f1e7e6398371cdcdd97437994cd4c9fec1532982e2d2f13
+  data.tar.gz: 2abcda0fcc478de60f6192967f9465f815d68476495b6c5988145bcbc7361b0f4dfcf8ef97a3924e09302c48c058f556bd848111eda8e8fe977ffec01dfee9f9

data/lib/helicone/agent.rb

@@ -4,19 +4,24 @@ module Helicone
   class Agent
     MAX_ITERATIONS = 10
 
-    attr_reader :client, :tools, :messages, :context
+    attr_reader :client, :tools, :messages, :context, :turn_model, :response_model
 
     # Create an agent with tools and optional context
     #
     # @param client [Helicone::Client] Optional client (creates new one if not provided)
+    # @param session [Hash] Session config for Helicone tracking (id:, name:) - creates client if no client provided
     # @param tools [Array<Class>] Array of Tool subclasses
     # @param context [Object] Context object passed to tool#initialize
     # @param system_prompt [String] System prompt
     # @param messages [Array<Helicone::Message>] Initial messages (for continuing conversations)
-    def initialize(client: nil, tools: [], context: nil, system_prompt: nil, messages: [])
-      @client = client || Client.new
+    # @param turn_model [String] Model for tool-calling turns (defaults to Helicone.configuration.default_model)
+    # @param response_model [String] Model for final response (defaults to turn_model)
+    def initialize(client: nil, session: nil, tools: [], context: nil, system_prompt: nil, messages: [], turn_model: nil, response_model: nil)
+      @client = client || build_client(session: session)
       @tools = tools
       @context = context
+      @turn_model = turn_model
+      @response_model = response_model || turn_model
       @messages = messages.dup
 
       # Add system message at the start if provided and not already present
@@ -35,7 +40,7 @@ module Helicone
 
       iterations = 0
       while iterations < max_iterations
-        response = call_llm
+        response = call_llm(model: @turn_model)
 
         tool_calls = response.tool_calls
         if tool_calls && !tool_calls.empty?
@@ -54,7 +59,17 @@ module Helicone
 
           iterations += 1
         else
-          # No tool calls - we're done
+          # No tool calls - if response_model differs, make final call with it
+          if @response_model && @response_model != @turn_model
+            final_response = @client.chat(messages: @messages, model: @response_model)
+            return AgentResult.new(
+              content: final_response.content,
+              messages: @messages,
+              iterations: iterations,
+              response: final_response
+            )
+          end
+
           return AgentResult.new(
             content: response.content,
             messages: @messages,
@@ -65,7 +80,7 @@ module Helicone
       end
 
       # Max iterations reached - make one final call without tools to get a response
-      final_response = @client.chat(messages: @messages)
+      final_response = @client.chat(messages: @messages, model: @response_model)
 
       AgentResult.new(
         content: final_response.content,
@@ -91,9 +106,10 @@ module Helicone
       Helicone.configuration.logger
     end
 
-    def call_llm
+    def call_llm(model: nil)
       @client.chat(
         messages: @messages,
+        model: model,
         tools: tools_for_api,
         tool_choice: "auto"
       )
@@ -130,5 +146,14 @@ module Helicone
     def find_tool_class(name)
       @tools.find { |t| t.function_name == name }
     end
+
+    def build_client(session:)
+      return Client.new unless session
+
+      Client.new(
+        session_id: session[:id],
+        session_name: session[:name]
+      )
+    end
   end
 end

data/lib/helicone/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Helicone
-  VERSION = "0.1.2"
+  VERSION = "0.2.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: helicone-rb
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.2.0
 platform: ruby
 authors:
 - Genevere