langgraph_rb 0.1.6 → 0.1.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 071bb22811337400b9569f90e07b50c0f7c6784b1025ce30b0502c5d321965b5
-  data.tar.gz: dad734ef3e60f76d8bf47de4c1f5f50a1b361dad8cfd7e36ed57e3cd4f998dca
+  metadata.gz: 6882db2f57746e6a18576eefd6bd26749009361880d1cd858ee91d3d06385abe
+  data.tar.gz: fa195eb1ed6763a3e0b53803792e92e7dd1dd327c9dd9372d7acf7ab3ea49d8f
 SHA512:
-  metadata.gz: 20c5c1c2ab980770d8ef0ca69159c1566c889c3ecd80d9185f6767b7954d92f5a15847ac50b1024dbd976670dc90d76b1bec9a95fa5d55b6123d6cdb26633bd2
-  data.tar.gz: a6edae4423708c7f807a050a9d4238d36ed188d0a0697e04eb2f21cebb3987ddf9d7f627cd6657d0b128470a58ca2a6895d02614da30341c73dbd5af1584337e
+  metadata.gz: aa652301646c12a2d2ade30faf5d5f463a89190813b9dbccb8c1b8df675bc6e0e8911c3358166224617239f763578b95fe78f27850e11690fc3e7726b85edc4c
+  data.tar.gz: 1a7fa46cf43d02b0c356c58f5e5aaed84023c5c2cece84eaf49ff084627ac046f1dd170f4e9cf2477ed1981d1ca8b7321ad416c9e4610c854793bdcdbf6b990d

data/langgraph_rb.gemspec

@@ -33,7 +33,6 @@ Gem::Specification.new do |spec|
 
   # Runtime dependencies
   spec.add_dependency "json", "~> 2.0"
-  spec.add_dependency "openai", "~> 0.24.0"
 
   # Development dependencies
   spec.add_development_dependency "bundler", "~> 2.0"

data/lib/langgraph_rb/chat_openai.rb

@@ -1,12 +1,20 @@
-require 'openai'
+begin
+  require 'openai'
+rescue LoadError
+  raise "LangGraphRB::ChatOpenAI requires gem 'openai' (~> 0.24). Add it to your Gemfile."
+end
 require_relative 'llm_base'
 
 module LangGraphRB
   # ChatOpenAI wrapper compatible with LLMBase, supporting tool binding
   class ChatOpenAI < LLMBase
-    def initialize(model:, temperature: 0.0, api_key: ENV['OPENAI_API_KEY'])
+    def initialize(model:, temperature: 0.0, api_key: ENV['OPENAI_API_KEY'], client: nil)
       super(model: model, temperature: temperature)
-      @client = OpenAI::Client.new(api_key: api_key)
+      @client = client || OpenAI::Client.new(api_key: api_key)
+
+      unless @client.respond_to?(:chat) && @client.chat.respond_to?(:completions)
+        raise "LangGraphRB::ChatOpenAI expects 'openai' gem ~> 0.24 (client.chat.completions.create)"
+      end
     end
 
     # Returns a new instance with tools bound (non-destructive)
@@ -77,8 +85,14 @@ module LangGraphRB
       else
         text_content
       end
+    rescue => e
+      notify_llm_error({
+        error: e.message
+      })
+      raise e
     end
 
+
     private
 
     def normalize_messages(messages)

data/lib/langgraph_rb/chat_ruby_openai.rb

@@ -0,0 +1,171 @@
+begin
+  require 'openai'
+rescue LoadError
+  raise "LangGraphRB::ChatRubyOpenAI requires gem 'ruby-openai' (~> 8.1). Add it to your Gemfile."
+end
+
+require_relative 'llm_base'
+
+module LangGraphRB
+  class ChatRubyOpenAI < LLMBase
+    def initialize(model:, temperature: 0.0, api_key: ENV['OPENAI_API_KEY'], client: nil)
+      super(model: model, temperature: temperature)
+      @client = client || OpenAI::Client.new(access_token: api_key)
+    end
+
+    def bind_tools(tools)
+      dup_instance = self.class.new(model: @model, temperature: @temperature)
+      dup_instance.instance_variable_set(:@client, @client)
+      dup_instance.instance_variable_set(:@bound_tools, Array(tools))
+      dup_instance
+    end
+
+    def call(messages, tools: nil)
+      raise ArgumentError, 'messages must be an Array' unless messages.is_a?(Array)
+
+      tool_definitions = (tools || @bound_tools)
+      tool_schemas = Array(tool_definitions).flat_map do |tool|
+        if tool.respond_to?(:to_openai_tool_schema)
+          Array(tool.to_openai_tool_schema)
+        else
+          [tool]
+        end
+      end
+
+      request_payload = {
+        model: @model,
+        temperature: @temperature,
+        messages: normalize_messages(messages)
+      }
+
+      if tool_schemas && !tool_schemas.empty?
+        request_payload[:tools] = tool_schemas
+        request_payload[:tool_choice] = 'auto'
+      end
+
+      notify_llm_request({
+        name: 'OpenAI::ChatCompletion',
+        model: @model,
+        model_parameters: { temperature: @temperature },
+        input: request_payload[:messages]
+      })
+
+      # ruby-openai 8.1.x: client.chat(parameters: {...}) returns a Hash
+      response = @client.chat(parameters: request_payload)
+
+      message = extract_message_from_response(response)
+      tool_calls = message[:tool_calls]
+      text_content = message[:content]
+
+      usage = extract_usage_from_response(response)
+      notify_llm_response({
+        output: tool_calls ? { tool_calls: tool_calls } : text_content,
+        prompt_tokens: usage[:prompt_tokens],
+        completion_tokens: usage[:completion_tokens],
+        total_tokens: usage[:total_tokens]
+      })
+
+      if tool_calls && !tool_calls.empty?
+        normalized_calls = tool_calls.map do |tc|
+          {
+            id: tc[:id],
+            name: tc[:function][:name],
+            arguments: parse_tool_arguments(tc[:function][:arguments])
+          }
+        end
+        { tool_calls: normalized_calls }
+      else
+        text_content
+      end
+    rescue => e
+      notify_llm_error({ error: e.message })
+      raise e
+    end
+
+    private
+
+    def normalize_messages(messages)
+      messages.map do |m|
+        role = (m[:role] || m['role'])
+        content = m[:content] || m['content']
+
+        normalized = { role: role }
+
+        if content.is_a?(Array)
+          normalized[:content] = content
+        elsif content.nil?
+          normalized[:content] = nil
+        else
+          normalized[:content] = content.to_s
+        end
+
+        tool_calls = m[:tool_calls] || m['tool_calls']
+        if tool_calls && role.to_s == 'assistant'
+          normalized[:tool_calls] = Array(tool_calls).map do |tc|
+            if tc[:function] || tc['function']
+              fn = tc[:function] || tc['function']
+              raw_args = fn[:arguments] || fn['arguments']
+              args_str = raw_args.is_a?(String) ? raw_args : JSON.dump(raw_args || {})
+              {
+                id: (tc[:id] || tc['id']),
+                type: 'function',
+                function: {
+                  name: (fn[:name] || fn['name']).to_s,
+                  arguments: args_str
+                }
+              }
+            else
+              raw_args = tc[:arguments] || tc['arguments']
+              args_str = raw_args.is_a?(String) ? raw_args : JSON.dump(raw_args || {})
+              {
+                id: (tc[:id] || tc['id']),
+                type: 'function',
+                function: {
+                  name: (tc[:name] || tc['name']).to_s,
+                  arguments: args_str
+                }
+              }
+            end
+          end
+        end
+
+        if role.to_s == 'tool'
+          tool_call_id = m[:tool_call_id] || m['tool_call_id']
+          name = m[:name] || m['name']
+          normalized[:tool_call_id] = tool_call_id if tool_call_id
+          normalized[:name] = name if name
+        end
+
+        normalized
+      end
+    end
+
+    def parse_tool_arguments(raw)
+      return {} if raw.nil?
+      case raw
+      when String
+        JSON.parse(raw) rescue {}
+      when Hash
+        raw
+      else
+        {}
+      end
+    end
+
+    def extract_message_from_response(response)
+      (response['choices'] || []).dig(0, 'message') || {}
+    end
+
+    def extract_usage_from_response(response)
+      usage = response['usage']
+      return { prompt_tokens: nil, completion_tokens: nil, total_tokens: nil } unless usage
+      {
+        prompt_tokens: usage['prompt_tokens'],
+        completion_tokens: usage['completion_tokens'],
+        total_tokens: usage['total_tokens']
+      }
+    end
+  end
+end
+
+

data/lib/langgraph_rb/llm_base.rb

@@ -54,6 +54,16 @@ module LangGraphRB
         end
       end
     end
+
+    def notify_llm_error(payload)
+      @observers.each do |observer|
+        begin
+          observer.on_llm_error(payload, @node_name)
+        rescue => _e
+          # Ignore observer errors
+        end
+      end
+    end
   end
 end
 

data/lib/langgraph_rb/observers/base.rb

@@ -70,6 +70,10 @@ module LangGraphRB
         # Override in subclasses
       end
 
+      def on_llm_error(event)
+        # Override in subclasses
+      end
+
       protected
 
       # Helper method to create standardized event structure

data/lib/langgraph_rb/observers/langfuse.rb

@@ -165,6 +165,24 @@ module LangGraphRB
       rescue => _e
       end
 
+      def on_llm_error(data, node_name)
+        record = with_records_lock do
+          stack = @records_by_node[node_name]
+          stack.empty? ? nil : stack[-1]
+        end
+        return unless record && record[:generation]
+
+        generation = record[:generation]        
+        generation.output = data[:error]
+        generation.end_time = Time.now.utc
+        Langfuse.update_generation(generation)
+
+        with_records_lock do
+          record[:generation] = nil
+        end
+      rescue => _e
+      end
+
       private
 
       def ensure_trace!(event)

data/lib/langgraph_rb/version.rb

@@ -1,3 +1,3 @@
 module LangGraphRB
-  VERSION = "0.1.6"
+  VERSION = "0.1.8"
 end 

data/lib/langgraph_rb.rb

@@ -11,7 +11,6 @@ require_relative 'langgraph_rb/observers/logger'
 require_relative 'langgraph_rb/observers/structured'
 require_relative 'langgraph_rb/observers/langfuse'
 require_relative 'langgraph_rb/llm_base'
-require_relative 'langgraph_rb/chat_openai'
 require_relative 'langgraph_rb/tool_definition'
 
 module LangGraphRB

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: langgraph_rb
 version: !ruby/object:Gem::Version
-  version: 0.1.6
+  version: 0.1.8
 platform: ruby
 authors:
 - Julian Toro
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2025-09-20 00:00:00.000000000 Z
+date: 2025-10-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: json
@@ -24,20 +24,6 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '2.0'
-- !ruby/object:Gem::Dependency
-  name: openai
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: 0.24.0
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: 0.24.0
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -136,6 +122,7 @@ files:
 - langgraph_rb.gemspec
 - lib/langgraph_rb.rb
 - lib/langgraph_rb/chat_openai.rb
+- lib/langgraph_rb/chat_ruby_openai.rb
 - lib/langgraph_rb/command.rb
 - lib/langgraph_rb/edge.rb
 - lib/langgraph_rb/graph.rb