helicone-rb 0.3.0 → 0.3.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a118f742137f788207fb3e4b724ec2420884ef5610776d1437385cd2fa963649
-  data.tar.gz: 44b4d16c5295b974afa28d2b9f694bf0c2b0cb95ba810826eaa5a72b141cd1e1
+  metadata.gz: ce9db5e8198fb863f8dd612b39b70a6d7b533e88d8a6ba19ec584d411f718541
+  data.tar.gz: 9ade07a85413d98df255a94b56099610c468d7644ace498392dc38b2799123b5
 SHA512:
-  metadata.gz: b689d259dbc4f8b583a62d5b95e84053aa58f41d594cf03fb5b681dc10d180e4d3ab5d86844d9875821c3d6b8c3270ff7e5faf5660c89501a8e72e8e7bf9d1b0
-  data.tar.gz: 6aa3ef6dda08bccd058deb550c23a0059693f759476b8394fa0d0c5f24497475decd2beb81a7f5818c6a58e243b8d8484bda22e405fcd965ae602bd789451904
+  metadata.gz: be9bfec349c828c30865d22c32004769b650eca02252fa2401fe097d2fd87ad09711228b8d9980333ded71a6df990b0734ced95b935818c159c238a75502e61d
+  data.tar.gz: f3d70faaeffb523d07f329d0919c6c648f610f0b643314842d533dd9c73ab9c5b3c97d405bcf208ebfd2f895c7bf8c104bc475192e476061676d86829d1ae452

data/lib/helicone/agent.rb

@@ -4,22 +4,25 @@ module Helicone
   class Agent
     MAX_ITERATIONS = 10
 
-    attr_reader :client, :tools, :messages, :context, :model
+    attr_reader :client, :tools, :messages, :context, :model, :options
 
     # 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 model [String] Model to use (defaults to Helicone.configuration.default_model)
-    def initialize(client: nil, session: nil, tools: [], context: nil, system_prompt: nil, messages: [], model: nil)
-      @client = client || build_client(session: session)
+    # @param options [Hash] Additional options passed through to chat calls (e.g., reasoning_effort)
+    def initialize(client: nil, session: nil, user: nil, tools: [], context: nil, system_prompt: nil, messages: [], model: nil, **options)
+      @client = client || build_client(session: session, user: user)
       @tools = tools
       @context = context
       @model = model
+      @options = options
       @messages = messages.dup
 
       # Add system message at the start if provided and not already present
@@ -99,7 +102,8 @@ module Helicone
         messages: @messages,
         model: @model,
         tools: tools_for_api,
-        tool_choice: "auto"
+        tool_choice: "auto",
+        **@options
       )
     end
 
@@ -136,12 +140,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/message.rb

@@ -114,10 +114,9 @@ module Helicone
     # @return [Hash] Message formatted for the API
     def to_h
       if @raw_message
-        # Select only the fields needed for the API
-        hash = { role: @raw_message[:role], content: @raw_message[:content] }
-        hash[:tool_calls] = @raw_message[:tool_calls] if @raw_message[:tool_calls]
-        hash
+        # Return the full raw message to preserve extra_content (thought signatures)
+        # and any other provider-specific fields (e.g., Gemini 3 requires thought_signature)
+        @raw_message
       else
         hash = { role: role, content: content }
         hash[:tool_call_id] = tool_call_id if tool_call_id

data/lib/helicone/version.rb

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

metadata

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