debug-agent 0.7.0 → 0.7.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fe4d325302d304db5821ef8bc5283ce3978e26d7721604efcbf00a646229346c
-  data.tar.gz: '058ae474cc6e35f83f59ee86da1ca807af84a51762c66a9452827805e28664c9'
+  metadata.gz: 3e6a7d1a5a0a33669a5e1b1c830761c5bf5a7a0b339e58328d29c7af1621dda4
+  data.tar.gz: 5beda4ad577f6598bcfd967e6124e819ebcbd853e00e5291a14baa08a7750ba4
 SHA512:
-  metadata.gz: f72cf134d401b2f055e5103eea6b6d35e06bbc75f7962a6f33d37f906f91e6899a5f562a5a032209448f469c5830b24d582b43b34730fbf949a1c47d4a8b25cb
-  data.tar.gz: dbb70e23e5bce9c717c8ecbdf708b2724811beeac9fa3672f496ac25dfedda108f0c051f71c7af5a1421cf181d120534aff1a94f6c511df3af04a56745a19098
+  metadata.gz: a4e5b1d103102c3446309f51b5d27dc689ff0a83bd32ab83eb4d800befd3fb72b4efbe0ae57362acc225974b36ec8b3f95ee24e7f1323dec9287f119a13c116c
+  data.tar.gz: b4105c3b8e1f29d7df9dce15c10852f6c7cf6eb2b9797dfd71fa0f8b5ea56989d9a34386e0b9166a5987c199ec86da5b1a5be8bdf3c084d1f30ed01f074b2f76

data/lib/debug_agent/inspectors/leak_detector.rb

@@ -45,6 +45,11 @@ module DebugAgent
       heap_pages: gc_stats[:heap_length]
     }
 
+    # Enforce max 50 heap snapshots to prevent unbounded memory growth
+    if @heap_snapshots.size >= 50
+      oldest_key = @heap_snapshots.keys.min_by { |k| k[/\d+/].to_i }
+      @heap_snapshots.delete(oldest_key)
+    end
     @heap_snapshots[snapshot_id] = snapshot
 
     {

data/lib/debug_agent/inspectors/snapshot.rb

@@ -49,6 +49,12 @@ module DebugAgent
 
     @metric_snapshot_counter += 1
     snapshot_id = "snap-#{@metric_snapshot_counter}"
+
+    # Enforce max 100 snapshots to prevent unbounded growth
+    if @metric_snapshots.size >= 100
+      oldest_key = @metric_snapshots.keys.min_by { |k| k[/\d+/].to_i }
+      @metric_snapshots.delete(oldest_key)
+    end
     @metric_snapshots[snapshot_id] = metrics
 
     {

data/lib/debug_agent/llm_client.rb

@@ -84,7 +84,7 @@ module DebugAgent
     def stream_request(path, body, handler)
       uri = URI(@cfg.base_url + path)
 
-      Net::HTTP.start(uri.host, uri.port, use_ssl: uri.scheme == 'https', read_timeout: @cfg.timeout_seconds) do |http|
+      Net::HTTP.start(uri.host, uri.port, use_ssl: uri.scheme == 'https', read_timeout: @cfg.timeout_seconds, open_timeout: 10) do |http|
         request = Net::HTTP::Post.new(uri.path)
         request['Authorization'] = "Bearer #{@cfg.api_key}"
         request['Content-Type'] = 'application/json'
@@ -181,6 +181,7 @@ module DebugAgent
       http = Net::HTTP.new(uri.host, uri.port)
       http.use_ssl = uri.scheme == 'https'
       http.read_timeout = @cfg.timeout_seconds
+      http.open_timeout = 10
 
       request = Net::HTTP::Post.new(uri.path)
       request['Authorization'] = "Bearer #{@cfg.api_key}"

data/lib/debug_agent/middleware.rb

@@ -2,10 +2,13 @@ require 'json'
 require_relative 'chat_page'
 
 module DebugAgent
-  # SSE callback that bridges engine to SSE response lines
+  # SSE callback that collects events for both buffered and streaming modes.
+  # In streaming mode, events are yielded in real-time via an Enumerator.
   class SseCallback < ChatCallback
     def initialize
       @events = []
+      @queue = Queue.new
+      @done = false
     end
 
     def events
@@ -14,27 +17,47 @@ module DebugAgent
 
     def on_content(chunk)
       @events << ['content', JSON.generate(chunk)]
+      @queue << ['content', JSON.generate(chunk)]
     end
 
     def on_tool_start(tool_name, args)
       @events << ['tool_start', tool_name]
+      @queue << ['tool_start', tool_name]
     end
 
     def on_tool_result(tool_name, result)
       @events << ['tool_result', "#{tool_name}: #{result}"]
+      @queue << ['tool_result', "#{tool_name}: #{result}"]
     end
 
     def on_complete
       @events << ['done', '']
+      @queue << ['done', '']
+      @done = true
+      @queue.close
     end
 
     def on_error(message)
       @events << ['error', message]
+      @queue << ['error', message]
+      @done = true
+      @queue.close
     end
 
     def on_context_compressed(original, compressed, removed_rounds)
       info = JSON.generate({ originalTokens: original, compressedTokens: compressed, removedRounds: removed_rounds })
       @events << ['context_compressed', info]
+      @queue << ['context_compressed', info]
+    end
+
+    # Real-time streaming Enumerator: yields SSE-formatted lines as events arrive.
+    def streaming_enum
+      Enumerator.new do |yielder|
+        while (item = @queue.pop)
+          event_type, data = item
+          yielder << "event: #{event_type}\ndata: #{data}\n\n"
+        end
+      end
     end
   end
 
@@ -67,35 +90,55 @@ module DebugAgent
 
   # Shared routing logic used by both RackMiddleware class and Middleware lambda
   module MiddlewareCore
+    CORS_HEADERS = {
+      'Access-Control-Allow-Origin' => '*',
+      'Access-Control-Allow-Methods' => 'GET, POST, OPTIONS',
+      'Access-Control-Allow-Headers' => 'Content-Type, Authorization',
+    }.freeze
+
     def self.call(env, app, engine, config)
       path = env['PATH_INFO']
       method = env['REQUEST_METHOD']
       base = config.base_path
 
+      # CORS preflight for all agent routes
+      if path.start_with?(base) && method == 'OPTIONS'
+        return [204, CORS_HEADERS.merge('Content-Length' => '0'), []]
+      end
+
       # Chat UI
       if path == base || path == "#{base}/"
         if method == 'GET'
           html = ChatPage.render(base)
-          return [200, { 'Content-Type' => 'text/html; charset=utf-8' }, [html]]
+          return [200, { 'Content-Type' => 'text/html; charset=utf-8' }.merge(CORS_HEADERS), [html]]
         end
       end
 
-      # SSE streaming chat
+      # SSE streaming chat — real-time streaming via Rack response body
       if path == "#{base}/api/chat" && method == 'POST'
         body = JSON.parse(env['rack.input'].read)
         message = body['message'] || ''
         session_id = body['sessionId'] || "session-#{Time.now.to_i}"
 
         cb = SseCallback.new
-        engine.chat(message, session_id, cb)
 
-        stream = cb.events.map { |event_type, data| "event: #{event_type}\ndata: #{data}\n\n" }.join
+        # Run engine in a background thread for real-time streaming
+        Thread.new do
+          begin
+            engine.chat(message, session_id, cb)
+          rescue => e
+            cb.on_error("Internal error: #{e.message}")
+          end
+        end
+
+        # Return a streaming response body (Rack 2+ / Rack 3 compatible)
+        streaming_body = cb.streaming_enum
 
         return [200, {
           'Content-Type' => 'text/event-stream',
           'Cache-Control' => 'no-cache',
-          'Connection' => 'keep-alive'
-        }, [stream]]
+          'Connection' => 'keep-alive',
+        }.merge(CORS_HEADERS), streaming_body]
       end
 
       # Clear conversation
@@ -103,17 +146,17 @@ module DebugAgent
         body = JSON.parse(env['rack.input'].read)
         session_id = body['sessionId'] || ''
         engine.clear_session(session_id) if session_id && !session_id.empty?
-        return [200, { 'Content-Type' => 'application/json' }, [JSON.generate({ status: 'cleared' })]]
+        return [200, { 'Content-Type' => 'application/json' }.merge(CORS_HEADERS), [JSON.generate({ status: 'cleared' })]]
       end
 
       # Health check
       if path == "#{base}/api/health" && method == 'GET'
-        return [200, { 'Content-Type' => 'application/json' }, [JSON.generate({ status: 'ok', agent: 'ruby-debug-agent' })]]
+        return [200, { 'Content-Type' => 'application/json' }.merge(CORS_HEADERS), [JSON.generate({ status: 'ok', agent: 'ruby-debug-agent' })]]
       end
 
       # List tools
       if path == "#{base}/api/tools" && method == 'GET'
-        return [200, { 'Content-Type' => 'application/json' }, [JSON.generate({ tools: engine.tools.all_schemas })]]
+        return [200, { 'Content-Type' => 'application/json' }.merge(CORS_HEADERS), [JSON.generate({ tools: engine.tools.all_schemas })]]
       end
 
       # Pass through to the wrapped app

data/lib/debug_agent/system_prompt_builder.rb

@@ -1,5 +1,6 @@
 module DebugAgent
   CATEGORY_MAP = {
+    # Memory & GC
     'gc'            => 'Memory & GC',
     'object_space'  => 'Memory & GC',
     'memory'        => 'Memory & GC',
@@ -7,22 +8,101 @@ module DebugAgent
     'allocations'   => 'Memory & GC',
     'force_gc'      => 'Memory & GC',
     'trigger_gc'    => 'Memory & GC',
+    'leak'          => 'Memory & GC',
+    'heap'          => 'Memory & GC',
+    'snapshot'      => 'Memory & Snapshots',
+    'compare'       => 'Memory & Snapshots',
+    # Process & Runtime
     'process'       => 'Process Info',
     'cpu'           => 'Process Info',
     'uptime'        => 'Process Info',
-    'thread'        => 'Threads',
+    'runtime'       => 'Runtime Info',
     'system'        => 'System Info',
     'disk'          => 'System Info',
-    'environment'   => 'System Info',
+    'environment'   => 'Environment & Config',
+    # Threads & Locks
+    'thread'        => 'Threads & Locks',
+    'lock'          => 'Threads & Locks',
+    'deadlock'      => 'Threads & Locks',
+    'contention'    => 'Threads & Locks',
+    'gvl'           => 'Threads & Locks',
+    'mutex'         => 'Threads & Locks',
+    'fiber'         => 'Threads & Locks',
+    # Framework
     'routes'        => 'Framework & Routes',
     'middleware'    => 'Framework & Routes',
-    'runtime'       => 'Runtime Info',
+    'rails'         => 'Framework & Routes',
+    'sinatra'       => 'Framework & Routes',
+    # HTTP
     'recent'        => 'HTTP Requests',
     'slow'          => 'HTTP Requests',
-    'error'         => 'HTTP Requests',
+    'error'         => 'Error Tracking',
     'request'       => 'HTTP Requests',
-    'gem'           => 'Dependencies',
+    'http'          => 'HTTP Requests',
+    'outbound'      => 'HTTP Requests',
+    'faraday'       => 'HTTP Requests',
+    # Database
+    'active_record' => 'Database',
+    'migration'     => 'Database Migration',
+    'pending'       => 'Database Migration',
+    'sql'           => 'Database',
+    'database'      => 'Database',
+    # Configuration
+    'config'        => 'Configuration',
+    'env'           => 'Configuration',
+    # Cache
+    'cache'         => 'Cache',
+    # Health & Security
+    'health'        => 'Health Checks',
+    'auth'          => 'Security',
+    'cors'          => 'Security',
+    # WebSocket
+    'ws'            => 'WebSocket',
+    'websocket'     => 'WebSocket',
+    # Profiling
+    'start'         => 'Profiling',
+    'stop'          => 'Profiling',
+    'top'           => 'Profiling',
+    'profile'       => 'Profiling',
+    # Feature Flags
+    'feature'       => 'Feature Flags',
+    'flag'          => 'Feature Flags',
+    'evaluate'      => 'Feature Flags',
+    # Endpoint Testing
+    'test'          => 'Endpoint Testing',
+    'batch'         => 'Endpoint Testing',
+    'endpoint'      => 'Endpoint Testing',
+    'coverage'      => 'Endpoint Testing',
+    # Connection Pool
+    'pool'          => 'Connection Pool',
+    'connection'    => 'Connection Pool',
+    # File Descriptors
+    'fd'            => 'File Descriptors',
+    'handle'        => 'File Descriptors',
+    # Metrics
+    'metric'        => 'Metrics',
+    'counter'       => 'Metrics',
+    # Build & Deployment
+    'build'         => 'Build & Deployment',
+    'deployment'    => 'Build & Deployment',
+    'version'       => 'Build & Deployment',
+    # Service Registry
+    'registered'    => 'Service Registry',
+    'service'       => 'Service Registry',
+    'dependencies'  => 'Service Registry',
+    # Job Queue
+    'sidekiq'       => 'Job Queue',
+    'queue'         => 'Job Queue',
+    'job'           => 'Job Queue',
+    # Redis
+    'redis'         => 'Redis',
+    # Logging
+    'log'           => 'Logging',
+    # Modules & Dependencies
     'module'        => 'Module Info',
+    'gem'           => 'Dependencies',
+    'loaded'        => 'Module Info',
+    'installed'     => 'Dependencies',
   }.freeze
 
   class SystemPromptBuilder

data/lib/debug_agent/version.rb

@@ -1,3 +1,3 @@
 module DebugAgent
-  VERSION = '0.7.0'.freeze
+  VERSION = '0.7.1'.freeze
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: debug-agent
 version: !ruby/object:Gem::Version
-  version: 0.7.0
+  version: 0.7.1
 platform: ruby
 authors:
 - ggcode