RubyGems - prompt_objects - Versions diffs - 0.2.0 → 0.3.0 - Mend

prompt_objects 0.2.0 → 0.3.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (51) hide show

checksums.yaml +4 -4
data/CHANGELOG.md +68 -0
data/Gemfile.lock +1 -1
data/README.md +2 -2
data/exe/prompt_objects +387 -1
data/frontend/src/App.tsx +11 -3
data/frontend/src/components/ContextMenu.tsx +67 -0
data/frontend/src/components/MessageBus.tsx +4 -3
data/frontend/src/components/ModelSelector.tsx +5 -1
data/frontend/src/components/ThreadsSidebar.tsx +46 -2
data/frontend/src/components/UsagePanel.tsx +105 -0
data/frontend/src/hooks/useWebSocket.ts +53 -0
data/frontend/src/store/index.ts +10 -0
data/frontend/src/types/index.ts +4 -1
data/lib/prompt_objects/cli.rb +1 -0
data/lib/prompt_objects/connectors/mcp.rb +1 -0
data/lib/prompt_objects/environment.rb +24 -1
data/lib/prompt_objects/llm/anthropic_adapter.rb +15 -1
data/lib/prompt_objects/llm/factory.rb +93 -6
data/lib/prompt_objects/llm/gemini_adapter.rb +13 -1
data/lib/prompt_objects/llm/openai_adapter.rb +21 -4
data/lib/prompt_objects/llm/pricing.rb +49 -0
data/lib/prompt_objects/llm/response.rb +3 -2
data/lib/prompt_objects/mcp/server.rb +1 -0
data/lib/prompt_objects/message_bus.rb +27 -8
data/lib/prompt_objects/prompt_object.rb +5 -3
data/lib/prompt_objects/server/api/routes.rb +186 -29
data/lib/prompt_objects/server/public/assets/index-Bkme6COu.css +1 -0
data/lib/prompt_objects/server/public/assets/index-CQ7lVDF_.js +77 -0
data/lib/prompt_objects/server/public/index.html +2 -2
data/lib/prompt_objects/server/websocket_handler.rb +93 -9
data/lib/prompt_objects/server.rb +54 -0
data/lib/prompt_objects/session/store.rb +399 -4
data/lib/prompt_objects.rb +1 -0
data/prompt_objects.gemspec +1 -1
data/templates/arc-agi-1/manifest.yml +22 -0
data/templates/arc-agi-1/objects/data_manager.md +42 -0
data/templates/arc-agi-1/objects/observer.md +100 -0
data/templates/arc-agi-1/objects/solver.md +118 -0
data/templates/arc-agi-1/objects/verifier.md +79 -0
data/templates/arc-agi-1/primitives/check_arc_data.rb +53 -0
data/templates/arc-agi-1/primitives/find_objects.rb +72 -0
data/templates/arc-agi-1/primitives/grid_diff.rb +70 -0
data/templates/arc-agi-1/primitives/grid_info.rb +42 -0
data/templates/arc-agi-1/primitives/grid_transform.rb +50 -0
data/templates/arc-agi-1/primitives/load_arc_task.rb +68 -0
data/templates/arc-agi-1/primitives/render_grid.rb +78 -0
data/templates/arc-agi-1/primitives/test_solution.rb +131 -0
metadata +20 -3
data/lib/prompt_objects/server/public/assets/index-CeNJvqLG.js +0 -77
data/lib/prompt_objects/server/public/assets/index-Vx4-uMOU.css +0 -1

data/lib/prompt_objects/server/public/index.html CHANGED Viewed

@@ -5,8 +5,8 @@
     <link rel="icon" type="image/svg+xml" href="/vite.svg" />
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
     <title>PromptObjects</title>
-    <script type="module" crossorigin src="/assets/index-CeNJvqLG.js"></script>
-    <link rel="stylesheet" crossorigin href="/assets/index-Vx4-uMOU.css">
+    <script type="module" crossorigin src="/assets/index-CQ7lVDF_.js"></script>
+    <link rel="stylesheet" crossorigin href="/assets/index-Bkme6COu.css">
   </head>
   <body class="bg-po-bg text-gray-100">
     <div id="root"></div>

data/lib/prompt_objects/server/websocket_handler.rb CHANGED Viewed

@@ -121,7 +121,8 @@ module PromptObjects
           payload: {
             from: entry[:from],
             to: entry[:to],
-            content: entry[:message],
+            summary: entry[:summary],
+            content: serialize_bus_content(entry[:message]),
             timestamp: entry[:timestamp].iso8601
           }
         )
@@ -175,7 +176,8 @@ module PromptObjects
             payload: {
               from: entry[:from],
               to: entry[:to],
-              content: entry[:message],
+              summary: entry[:summary],
+              content: serialize_bus_content(entry[:message]),
               timestamp: entry[:timestamp].iso8601
             }
           )
@@ -221,6 +223,10 @@ module PromptObjects
           handle_switch_llm(message["payload"])
         when "update_prompt"
           handle_update_prompt(message["payload"])
+        when "get_session_usage"
+          handle_get_session_usage(message["payload"])
+        when "export_thread"
+          handle_export_thread(message["payload"])
         when "ping"
           send_message(type: "pong", payload: {})
         else
@@ -563,6 +569,63 @@ module PromptObjects
         end
       end
+      def handle_get_session_usage(payload)
+        session_id = payload["session_id"]
+        include_tree = payload["include_tree"] || false
+        return send_error("Session ID required") unless session_id
+        return send_error("No session store available") unless @runtime.session_store
+        usage = if include_tree
+                  @runtime.session_store.thread_tree_usage(session_id)
+                else
+                  @runtime.session_store.session_usage(session_id)
+                end
+        # Convert by_model symbol keys to strings for JSON
+        by_model = {}
+        usage[:by_model].each { |model, data| by_model[model.to_s] = data }
+        send_message(
+          type: "session_usage",
+          payload: {
+            session_id: session_id,
+            include_tree: include_tree,
+            input_tokens: usage[:input_tokens],
+            output_tokens: usage[:output_tokens],
+            total_tokens: usage[:total_tokens],
+            estimated_cost_usd: usage[:estimated_cost_usd].round(6),
+            calls: usage[:calls],
+            by_model: by_model
+          }
+        )
+      end
+      def handle_export_thread(payload)
+        session_id = payload["session_id"]
+        format = payload["format"] || "markdown"
+        return send_error("Session ID required") unless session_id
+        return send_error("No session store available") unless @runtime.session_store
+        content = case format
+                  when "markdown"
+                    @runtime.session_store.export_thread_tree_markdown(session_id)
+                  when "json"
+                    data = @runtime.session_store.export_thread_tree_json(session_id)
+                    data ? JSON.pretty_generate(data) : nil
+                  end
+        return send_error("Session not found") unless content
+        send_message(
+          type: "thread_export",
+          payload: {
+            session_id: session_id,
+            format: format,
+            content: content
+          }
+        )
+      end
       # === Helpers ===
       def send_error(message)
@@ -649,21 +712,29 @@ module PromptObjects
       def message_to_hash(msg)
         case msg[:role]
         when :user
-          { role: "user", content: msg[:content], from: msg[:from] }
+          # In-memory messages use :from, SQLite-loaded messages use :from_po
+          from = msg[:from] || msg[:from_po]
+          { role: "user", content: msg[:content], from: from }
         when :assistant
           hash = { role: "assistant", content: msg[:content] }
           if msg[:tool_calls]
             hash[:tool_calls] = msg[:tool_calls].map do |tc|
-              # Handle both ToolCall objects and Hashes
-              tc_id = tc.respond_to?(:id) ? tc.id : (tc[:id] || tc["id"])
-              tc_name = tc.respond_to?(:name) ? tc.name : (tc[:name] || tc["name"])
-              tc_args = tc.respond_to?(:arguments) ? tc.arguments : (tc[:arguments] || tc["arguments"] || {})
-              { id: tc_id, name: tc_name, arguments: tc_args }
+              # Handle both ToolCall objects and Hashes (from DB with symbol or string keys)
+              if tc.is_a?(LLM::ToolCall)
+                { id: tc.id, name: tc.name, arguments: tc.arguments }
+              else
+                tc_id = tc[:id] || tc["id"]
+                tc_name = tc[:name] || tc["name"]
+                tc_args = tc[:arguments] || tc["arguments"] || {}
+                { id: tc_id, name: tc_name, arguments: tc_args }
+              end
             end
           end
           hash
         when :tool
-          { role: "tool", results: msg[:results] }
+          # In-memory messages use :results, SQLite-loaded messages use :tool_results
+          results = msg[:results] || msg[:tool_results]
+          { role: "tool", results: results }
         else
           { role: msg[:role].to_s, content: msg[:content] }
         end
@@ -678,6 +749,19 @@ module PromptObjects
           options: request.options || []
         }
       end
+      # Serialize bus message content for JSON transmission.
+      # Handles String, Hash, and other types gracefully.
+      def serialize_bus_content(message)
+        case message
+        when Hash
+          message
+        when String
+          message
+        else
+          message.to_s
+        end
+      end
     end
   end
 end

data/lib/prompt_objects/server.rb CHANGED Viewed

@@ -69,6 +69,9 @@ module PromptObjects
         file_watcher.start
       end
+      # Write .server file for CLI discovery
+      server_file = write_server_file(env_path, host: host, port: port) if env_path
       Async do |task|
         endpoint = Async::HTTP::Endpoint.parse(url)
@@ -82,7 +85,58 @@ module PromptObjects
         end
       ensure
         file_watcher&.stop
+        remove_server_file(server_file) if server_file
+      end
+    end
+    # Write a .server file so CLI clients can discover the running server.
+    # @param env_path [String] Path to the environment directory
+    # @param host [String] Server host
+    # @param port [Integer] Server port
+    # @return [String] Path to the .server file
+    def self.write_server_file(env_path, host:, port:)
+      return nil unless env_path
+      server_file = File.join(env_path, ".server")
+      data = {
+        pid: Process.pid,
+        host: host,
+        port: port,
+        started_at: Time.now.iso8601
+      }
+      File.write(server_file, JSON.generate(data))
+      server_file
+    end
+    # Remove the .server file on shutdown.
+    # @param path [String] Path to the .server file
+    def self.remove_server_file(path)
+      File.delete(path) if path && File.exist?(path)
+    rescue StandardError
+      # Ignore cleanup errors
+    end
+    # Read a .server file to discover a running server.
+    # Returns nil if no server is running or the file is stale.
+    # @param env_path [String] Path to the environment directory
+    # @return [Hash, nil] Server info with :host, :port, :pid
+    def self.read_server_file(env_path)
+      server_file = File.join(env_path, ".server")
+      return nil unless File.exist?(server_file)
+      data = JSON.parse(File.read(server_file), symbolize_names: true)
+      # Check if the process is still running
+      begin
+        Process.kill(0, data[:pid])
+        data
+      rescue Errno::ESRCH
+        # Process is dead, clean up stale file
+        File.delete(server_file)
+        nil
       end
+    rescue StandardError
+      nil
     end
     # Handle file change events and broadcast to connected clients.