helicone-rb 0.3.0 → 0.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a118f742137f788207fb3e4b724ec2420884ef5610776d1437385cd2fa963649
-  data.tar.gz: 44b4d16c5295b974afa28d2b9f694bf0c2b0cb95ba810826eaa5a72b141cd1e1
+  metadata.gz: 3d7fc91958e3632fb91f474bfa20a5523bf2997ab33b2b0c5222977ff8a0b2e9
+  data.tar.gz: 967e53e49f3e2ddd480102ae5552ce3e3074c2a8f412c7bc35aa519f74e20bad
 SHA512:
-  metadata.gz: b689d259dbc4f8b583a62d5b95e84053aa58f41d594cf03fb5b681dc10d180e4d3ab5d86844d9875821c3d6b8c3270ff7e5faf5660c89501a8e72e8e7bf9d1b0
-  data.tar.gz: 6aa3ef6dda08bccd058deb550c23a0059693f759476b8394fa0d0c5f24497475decd2beb81a7f5818c6a58e243b8d8484bda22e405fcd965ae602bd789451904
+  metadata.gz: 70cc843edb4833e83aaee85d0af7703610f98ebff7e679e04ce5dde54419eaac22027d9c97ddccd0df55f75ff9ce1891d03686509c925ce8d7ab3fc37a2032ae
+  data.tar.gz: '082978063ff56855409433acc6969bb0a244d5e1a34e2050824ad76534beeaff2f9051005fa83a698a932982712929547fc5801312486a9a12bb03c12c09b459'

data/lib/helicone/agent.rb

@@ -10,13 +10,14 @@ module Helicone
     #
     # @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)
+    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
       @model = model
@@ -136,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.3.0"
+  VERSION = "0.3.1"
 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.1
 platform: ruby
 authors:
 - Genevere