helicone-rb 0.1.3 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 210ac6dc40e739a7ee0ff0478dfec18c300f786afd0b4f149a57048b1bd46ce9
-  data.tar.gz: a11ef6687f94ebdf9c9adcc56635256388718be341941fe13258ec8930c8fe29
+  metadata.gz: ac034e0153e6ec01bbe008066b48c58c754a103b56461fdc24c4e0d1bc05d4d4
+  data.tar.gz: c95390c56a649b16fde96f7be81a1ef537671a19a129039d23d272ea391cfdaf
 SHA512:
-  metadata.gz: a9b4f50614b0f88b150f5a6457f0fb0c8fef5046451a59a5963225202af5f72b6f72602df6b54c10996cd271ce192c1c5e6902198e744b2dafe06021964ab913
-  data.tar.gz: f887ebe7105b9287cdfaa94df8af657c69af7d28e8bc5760c5dc1aabfd5c9ddad7a3297efe9b713542e58fcfd640780079ffddc044e78fb6b203e38eb3111b31
+  metadata.gz: a4726d935f0db71a01ece1db9f29d758f2c4f47e04b79cb62ed986f1180abf6415576ccfa94e1de6e6d73671d21448c8067b1c74b6b55052195548c67533f64b
+  data.tar.gz: 1b1b93d6cffe0e19a41dbbefec03ab019bda992c4eb7a0dfc7770c60852f7dfde8d879d634aeeec10ca75837e799919be2ecc14f9f601efbb4b63636922b43ec

data/lib/helicone/agent.rb

@@ -4,7 +4,7 @@ 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
     #
@@ -14,10 +14,14 @@ module Helicone
     # @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, session: nil, tools: [], context: nil, system_prompt: nil, messages: [])
+    # @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
@@ -26,17 +30,17 @@ module Helicone
       end
     end
 
-    # Run the agent with a prompt, executing tools until done
+    # Run the agent with an optional prompt, executing tools until done
     #
-    # @param prompt [String] User prompt to start with
+    # @param prompt [String, nil] User prompt to add (nil to use existing messages)
     # @param max_iterations [Integer] Maximum tool execution loops
     # @return [AgentResult]
-    def run(prompt, max_iterations: MAX_ITERATIONS)
-      @messages << Message.user_text(prompt)
+    def run(prompt = nil, max_iterations: MAX_ITERATIONS)
+      @messages << Message.user_text(prompt) if prompt
 
       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?
@@ -55,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,
@@ -66,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,
@@ -92,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"
       )
@@ -106,16 +121,17 @@ module Helicone
     end
 
     def execute_tool(tool_call)
-      tool_class = find_tool_class(tool_call.name)
+      tool = find_tool(tool_call.name)
 
-      unless tool_class
+      unless tool
         logger.warn("[Helicone::Agent] Unknown tool: #{tool_call.name}")
         return { error: "Unknown tool: #{tool_call.name}" }
       end
 
       logger.info("[Helicone::Agent] Executing tool: #{tool_call.name} with #{tool_call.arguments}")
 
-      tool_instance = tool_class.new(@context)
+      # Support both tool classes and tool instances
+      tool_instance = tool.is_a?(Class) ? tool.new(@context) : tool
       result = tool_instance.execute(**tool_call.arguments)
 
       result_preview = result.inspect
@@ -128,7 +144,7 @@ module Helicone
       { error: e.message }
     end
 
-    def find_tool_class(name)
+    def find_tool(name)
       @tools.find { |t| t.function_name == name }
     end
 

data/lib/helicone/client.rb

@@ -13,7 +13,7 @@ module Helicone
     def initialize(session_id: nil, session_name: nil, account_id: nil, account_name: nil)
       @client = OpenAI::Client.new(
         access_token: ENV["HELICONE_API_KEY"],
-        uri_base: Helicone::URI_BASE
+        uri_base: Helicone.configuration.base_url
       )
 
       # Add Helicone session headers if provided

data/lib/helicone/configuration.rb

@@ -3,16 +3,17 @@
 require "logger"
 
 module Helicone
-  URI_BASE = "https://ai-gateway.helicone.ai/v1"
+  DEFAULT_BASE_URL = "https://ai-gateway.helicone.ai/v1"
 
   class Configuration
-    attr_accessor :logger, :default_model
+    attr_accessor :logger, :default_model, :base_url
 
     # Initialize configuration with defaults
     #
     # @return [Configuration]
     def initialize
       @default_model = "gpt-4o"
+      @base_url = DEFAULT_BASE_URL
       @logger = Logger.new($stdout, level: Logger::INFO)
     end
   end

data/lib/helicone/message.rb

@@ -113,9 +113,11 @@ module Helicone
     #
     # @return [Hash] Message formatted for the API
     def to_h
-      # If this is a raw message (assistant with tool_calls), return it as-is
       if @raw_message
-        @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
       else
         hash = { role: role, content: content }
         hash[:tool_call_id] = tool_call_id if tool_call_id

data/lib/helicone/tool.rb

@@ -98,5 +98,19 @@ module Helicone
     def execute(**args)
       raise NotImplementedError, "Subclasses must implement #execute"
     end
+
+    # Delegate to class method for instances
+    #
+    # @return [Hash] Tool definition formatted for OpenAI API
+    def to_openai_tool
+      self.class.to_openai_tool
+    end
+
+    # Delegate to class method for instances
+    #
+    # @return [String] The function name
+    def function_name
+      self.class.function_name
+    end
   end
 end

data/lib/helicone/version.rb

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

metadata

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