langgraph_rb 0.1.6 → 0.1.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 071bb22811337400b9569f90e07b50c0f7c6784b1025ce30b0502c5d321965b5
-  data.tar.gz: dad734ef3e60f76d8bf47de4c1f5f50a1b361dad8cfd7e36ed57e3cd4f998dca
+  metadata.gz: bd46b52a6c57bec611953702facad328eda63e0c5f66f85c403230d05548709b
+  data.tar.gz: 19e32e26b6c38044536b8498c7900bcb19386adf1853881cc3a9c3ca3861dae4
 SHA512:
-  metadata.gz: 20c5c1c2ab980770d8ef0ca69159c1566c889c3ecd80d9185f6767b7954d92f5a15847ac50b1024dbd976670dc90d76b1bec9a95fa5d55b6123d6cdb26633bd2
-  data.tar.gz: a6edae4423708c7f807a050a9d4238d36ed188d0a0697e04eb2f21cebb3987ddf9d7f627cd6657d0b128470a58ca2a6895d02614da30341c73dbd5af1584337e
+  metadata.gz: ed37faa9f9cc70f30a63ac31f96daafd2fa8f6362f78607ac3532ee396fdd04037265a428fc2e5492b4971100be836845e29228e59b5876abc8b50239c9b83a4
+  data.tar.gz: 33a4b49fa73f482c4a66299b0a58964be9a81f505e44d35c36b033334b73a24341fc3e78349ff0652f26ca0ae6884b5623fa07d8263570367a790b1324e278ec

data/examples/chat_openai_tools_example.rb

@@ -38,7 +38,7 @@ end
 def run_chat_openai_tools
   tools = [MovieInfoTool.new(api_key: ENV['TMDB_API_KEY'] || 'demo')]
 
-  chat = LangGraphRB::ChatOpenAI.new(model: ENV.fetch('OPENAI_MODEL', 'gpt-4o-mini'), temperature: 0)
+  chat = LangGraphRB::ChatOpenAI.new(model: ENV.fetch('OPENAI_MODEL', 'gpt-4o-min3i'), temperature: 0)
   chat = chat.bind_tools(tools)
 
   observers = [LangGraphRB::Observers::LangfuseObserver.new(name: 'chat-openai-tools-example')]
@@ -50,7 +50,11 @@ def run_chat_openai_tools
       { messages: existing + [user_msg] }
     end
 
-    llm_node :chat, llm_client: chat, system_prompt: "You are a movie assistant. Use tools when helpful."
+    # Optional: callback to observe each assistant message as it's produced
+    add_message_callback = ->(message) { puts "New message: #{message}" }
+    sys_prompt = "You are a movie assistant. Use tools when helpful."
+
+    llm_node :chat, llm_client: chat, system_prompt: sys_prompt, add_message_callback: add_message_callback
 
     tool_node :tool, tools: tools
 

data/lib/langgraph_rb/chat_openai.rb

@@ -77,8 +77,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/graph.rb

@@ -35,11 +35,11 @@ module LangGraphRB
       end
     end
 
-    def llm_node(name, llm_client:, system_prompt: nil, &block)
+    def llm_node(name, llm_client:, system_prompt: nil, add_message_callback: nil, &block)
       name = name.to_sym
       raise GraphError, "Node '#{name}' already exists" if @nodes.key?(name)
       
-      @nodes[name] = LLMNode.new(name, llm_client: llm_client, system_prompt: system_prompt, &block)
+      @nodes[name] = LLMNode.new(name, llm_client: llm_client, system_prompt: system_prompt, add_message_callback: add_message_callback, &block)
     end
 
     def tool_node(name, tools:, &block)

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/node.rb

@@ -35,11 +35,12 @@ module LangGraphRB
 
   # Specialized node for LLM calls
   class LLMNode < Node
-    attr_reader :llm_client, :system_prompt
+    attr_reader :llm_client, :system_prompt, :add_message_callback
 
-    def initialize(name, llm_client:, system_prompt: nil, &block)
+    def initialize(name, llm_client:, system_prompt: nil, add_message_callback: nil, &block)
       @llm_client = llm_client
       @system_prompt = system_prompt
+      @add_message_callback = add_message_callback
 
       # Use default LLM behavior if no custom block provided
       super(name, &(block || method(:default_llm_call)))
@@ -49,7 +50,8 @@ module LangGraphRB
       # Auto-inject LLM config into the context for both default and custom blocks
       merged_context = (context || {}).merge(
         llm_client: @llm_client,
-        system_prompt: @system_prompt
+        system_prompt: @system_prompt,
+        add_message_callback: @add_message_callback
       )
 
       begin
@@ -87,12 +89,14 @@ module LangGraphRB
           content: nil,
           tool_calls: response[:tool_calls]
         }
+        @add_message_callback&.call(assistant_msg)
         {
           messages: (state[:messages] || []) + [assistant_msg],
           tool_call: response[:tool_calls].first
         }
       else
         assistant_msg = { role: 'assistant', content: response.to_s }
+        @add_message_callback&.call(assistant_msg)
         {
           messages: (state[:messages] || []) + [assistant_msg],
           last_response: response.to_s

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.7"
 end 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: langgraph_rb
 version: !ruby/object:Gem::Version
-  version: 0.1.6
+  version: 0.1.7
 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