helicone-rb 0.2.1 → 0.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ac034e0153e6ec01bbe008066b48c58c754a103b56461fdc24c4e0d1bc05d4d4
-  data.tar.gz: c95390c56a649b16fde96f7be81a1ef537671a19a129039d23d272ea391cfdaf
+  metadata.gz: 3d7fc91958e3632fb91f474bfa20a5523bf2997ab33b2b0c5222977ff8a0b2e9
+  data.tar.gz: 967e53e49f3e2ddd480102ae5552ce3e3074c2a8f412c7bc35aa519f74e20bad
 SHA512:
-  metadata.gz: a4726d935f0db71a01ece1db9f29d758f2c4f47e04b79cb62ed986f1180abf6415576ccfa94e1de6e6d73671d21448c8067b1c74b6b55052195548c67533f64b
-  data.tar.gz: 1b1b93d6cffe0e19a41dbbefec03ab019bda992c4eb7a0dfc7770c60852f7dfde8d879d634aeeec10ca75837e799919be2ecc14f9f601efbb4b63636922b43ec
+  metadata.gz: 70cc843edb4833e83aaee85d0af7703610f98ebff7e679e04ce5dde54419eaac22027d9c97ddccd0df55f75ff9ce1891d03686509c925ce8d7ab3fc37a2032ae
+  data.tar.gz: '082978063ff56855409433acc6969bb0a244d5e1a34e2050824ad76534beeaff2f9051005fa83a698a932982712929547fc5801312486a9a12bb03c12c09b459'

data/lib/helicone/agent.rb

@@ -4,24 +4,23 @@ module Helicone
   class Agent
     MAX_ITERATIONS = 10
 
-    attr_reader :client, :tools, :messages, :context, :turn_model, :response_model
+    attr_reader :client, :tools, :messages, :context, :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 user [Hash] User config for Helicone tracking (id:, name:) - sets Helicone-User-Id header
     # @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)
-    # @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)
+    # @param model [String] Model to use (defaults to Helicone.configuration.default_model)
+    def initialize(client: nil, session: nil, user: nil, tools: [], context: nil, system_prompt: nil, messages: [], model: nil)
+      @client = client || build_client(session: session, user: user)
       @tools = tools
       @context = context
-      @turn_model = turn_model
-      @response_model = response_model || turn_model
+      @model = model
       @messages = messages.dup
 
       # Add system message at the start if provided and not already present
@@ -40,7 +39,7 @@ module Helicone
 
       iterations = 0
       while iterations < max_iterations
-        response = call_llm(model: @turn_model)
+        response = call_llm
 
         tool_calls = response.tool_calls
         if tool_calls && !tool_calls.empty?
@@ -59,17 +58,7 @@ module Helicone
 
           iterations += 1
         else
-          # 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
-
+          # No tool calls - we're done
           return AgentResult.new(
             content: response.content,
             messages: @messages,
@@ -80,7 +69,7 @@ module Helicone
       end
 
       # Max iterations reached - make one final call without tools to get a response
-      final_response = @client.chat(messages: @messages, model: @response_model)
+      final_response = @client.chat(messages: @messages, model: @model)
 
       AgentResult.new(
         content: final_response.content,
@@ -106,10 +95,10 @@ module Helicone
       Helicone.configuration.logger
     end
 
-    def call_llm(model: nil)
+    def call_llm
       @client.chat(
         messages: @messages,
-        model: model,
+        model: @model,
         tools: tools_for_api,
         tool_choice: "auto"
       )
@@ -148,12 +137,14 @@ module Helicone
       @tools.find { |t| t.function_name == name }
     end
 
-    def build_client(session:)
-      return Client.new unless session
+    def build_client(session:, user:)
+      return Client.new unless session || user
 
       Client.new(
-        session_id: session[:id],
-        session_name: session[:name]
+        session_id: session&.dig(:id),
+        session_name: session&.dig(:name),
+        account_id: user&.dig(:id),
+        account_name: user&.dig(:name)
       )
     end
   end

data/lib/helicone/version.rb

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

metadata

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