@optilogic/chat 1.3.2 → 1.3.4

package/dist/index.cjs

@@ -845,6 +845,188 @@ function deduplicateEntries(entries) {
   return result;
 }
 
+// src/components/agent-response/reducer.ts
+function reduceAgentMessage(prev, payload) {
+  const now = payload.timestamp ?? Date.now();
+  let newStatus = prev.status;
+  const isFirstMessage = prev.status === "idle" && payload.type !== "status";
+  if (isFirstMessage) {
+    newStatus = "processing";
+  }
+  const firstMessageTime = prev.firstMessageTime ?? (isFirstMessage ? now : null);
+  switch (payload.type) {
+    case "status":
+      if (payload.message === "Harness connected" || payload.status === "Harness connected") {
+        return { ...initialAgentResponseState };
+      }
+      return { ...prev, status: newStatus };
+    case "thinking": {
+      if (payload.thinkingStep) {
+        const newStep = {
+          id: payload.thinkingStep.id || `step-${now}`,
+          label: payload.thinkingStep.label,
+          content: payload.thinkingStep.content,
+          depth: payload.thinkingStep.depth ?? payload.depth ?? 0,
+          isCollapsed: payload.thinkingStep.isCollapsed,
+          timestamp: now,
+          agentName: payload.agentName,
+          parentAgent: payload.parentAgent
+        };
+        const thinkingStartTime2 = prev.thinkingStartTime ?? now;
+        const next2 = {
+          ...prev,
+          status: newStatus,
+          thinkingSteps: [...prev.thinkingSteps || [], newStep],
+          thinkingStartTime: thinkingStartTime2,
+          firstMessageTime
+        };
+        return { ...next2, timelineEntries: buildTimelineEntries(next2) };
+      }
+      const newThinking = payload.message || payload.content || "";
+      const separator = prev.thinking && newThinking ? "\n\n" : "";
+      const thinkingStartTime = prev.thinkingStartTime ?? (newThinking ? now : null);
+      const prevSteps = prev.thinkingSteps || [];
+      const plainStep = {
+        id: `step-${prevSteps.length}`,
+        label: newThinking,
+        content: newThinking,
+        depth: payload.depth ?? 0,
+        timestamp: now,
+        agentName: payload.agentName,
+        parentAgent: payload.parentAgent
+      };
+      const next = {
+        ...prev,
+        status: newStatus,
+        thinking: prev.thinking + separator + newThinking,
+        thinkingSteps: [...prevSteps, plainStep],
+        thinkingStartTime,
+        firstMessageTime
+      };
+      return { ...next, timelineEntries: buildTimelineEntries(next) };
+    }
+    case "tool_call": {
+      const toolName = payload.message || payload.tool?.name;
+      if (toolName) {
+        const newToolCall = {
+          id: payload.tool?.id || `tool-${now}`,
+          name: toolName,
+          arguments: payload.tool?.arguments,
+          timestamp: now,
+          agentName: payload.agentName,
+          parentAgent: payload.parentAgent,
+          depth: payload.depth
+        };
+        const next = {
+          ...prev,
+          status: newStatus,
+          toolCalls: [...prev.toolCalls, newToolCall],
+          firstMessageTime
+        };
+        return { ...next, timelineEntries: buildTimelineEntries(next) };
+      }
+      return { ...prev, status: newStatus, firstMessageTime };
+    }
+    case "knowledge": {
+      const knowledgeContent = payload.message || payload.knowledge?.content;
+      if (knowledgeContent) {
+        const newKnowledge = {
+          id: payload.knowledge?.id || `knowledge-${now}`,
+          source: payload.knowledge?.source || "unknown",
+          content: knowledgeContent,
+          timestamp: now,
+          agentName: payload.agentName,
+          parentAgent: payload.parentAgent,
+          depth: payload.depth
+        };
+        const next = {
+          ...prev,
+          status: newStatus,
+          knowledge: [...prev.knowledge, newKnowledge],
+          firstMessageTime
+        };
+        return { ...next, timelineEntries: buildTimelineEntries(next) };
+      }
+      return { ...prev, status: newStatus, firstMessageTime };
+    }
+    case "memory": {
+      const memoryContent = payload.message || payload.memory?.content;
+      if (memoryContent) {
+        const newMemory = {
+          id: payload.memory?.id || `memory-${now}`,
+          type: payload.memory?.type || "unknown",
+          content: memoryContent,
+          timestamp: now,
+          agentName: payload.agentName,
+          parentAgent: payload.parentAgent,
+          depth: payload.depth
+        };
+        const next = {
+          ...prev,
+          status: newStatus,
+          memory: [...prev.memory, newMemory],
+          firstMessageTime
+        };
+        return { ...next, timelineEntries: buildTimelineEntries(next) };
+      }
+      return { ...prev, status: newStatus, firstMessageTime };
+    }
+    case "response":
+      return {
+        ...prev,
+        status: "complete",
+        response: payload.message || payload.content || "",
+        responseCompleteTime: now,
+        firstMessageTime: prev.firstMessageTime ?? now
+      };
+    case "status_update": {
+      const statusMessage = payload.message || payload.statusUpdate?.message;
+      if (statusMessage) {
+        const newStatusItem = {
+          id: payload.statusUpdate?.id || `status-${now}`,
+          message: statusMessage,
+          agent: payload.statusUpdate?.agent,
+          timestamp: now,
+          agentName: payload.agentName,
+          parentAgent: payload.parentAgent,
+          depth: payload.depth
+        };
+        const next = {
+          ...prev,
+          status: newStatus,
+          statusUpdates: [...prev.statusUpdates, newStatusItem],
+          firstMessageTime
+        };
+        return { ...next, timelineEntries: buildTimelineEntries(next) };
+      }
+      return { ...prev, status: newStatus, firstMessageTime };
+    }
+    case "potential_response": {
+      const respContent = payload.message || payload.content || "";
+      if (respContent) {
+        const newResp = {
+          id: `resp-${now}`,
+          content: respContent,
+          timestamp: now,
+          agentName: payload.agentName,
+          parentAgent: payload.parentAgent,
+          depth: payload.depth
+        };
+        const next = {
+          ...prev,
+          status: newStatus,
+          potentialResponses: [...prev.potentialResponses || [], newResp],
+          firstMessageTime
+        };
+        return { ...next, timelineEntries: buildTimelineEntries(next) };
+      }
+      return { ...prev, status: newStatus, firstMessageTime };
+    }
+    default:
+      return { ...prev, status: newStatus, firstMessageTime };
+  }
+}
+
 // src/components/agent-response/hooks/useAgentResponseAccumulator.ts
 function useAgentResponseAccumulator(options) {
   const [state, setState] = React11.useState(initialAgentResponseState);
@@ -859,185 +1041,7 @@ function useAgentResponseAccumulator(options) {
       } else {
         payload = message;
       }
-      setState((prev) => {
-        let newStatus = prev.status;
-        const isFirstMessage = prev.status === "idle" && payload.type !== "status";
-        if (isFirstMessage) {
-          newStatus = "processing";
-        }
-        const firstMessageTime = prev.firstMessageTime ?? (isFirstMessage ? Date.now() : null);
-        switch (payload.type) {
-          case "status":
-            if (payload.message === "Harness connected" || payload.status === "Harness connected") {
-              return { ...initialAgentResponseState };
-            }
-            return { ...prev, status: newStatus };
-          case "thinking": {
-            if (payload.thinkingStep) {
-              const newStep = {
-                id: payload.thinkingStep.id || `step-${Date.now()}`,
-                label: payload.thinkingStep.label,
-                content: payload.thinkingStep.content,
-                depth: payload.thinkingStep.depth ?? payload.depth ?? 0,
-                isCollapsed: payload.thinkingStep.isCollapsed,
-                timestamp: Date.now(),
-                agentName: payload.agentName,
-                parentAgent: payload.parentAgent
-              };
-              const thinkingStartTime2 = prev.thinkingStartTime ?? Date.now();
-              const next2 = {
-                ...prev,
-                status: newStatus,
-                thinkingSteps: [...prev.thinkingSteps || [], newStep],
-                thinkingStartTime: thinkingStartTime2,
-                firstMessageTime
-              };
-              return { ...next2, timelineEntries: buildTimelineEntries(next2) };
-            }
-            const newThinking = payload.message || payload.content || "";
-            const separator = prev.thinking && newThinking ? "\n\n" : "";
-            const thinkingStartTime = prev.thinkingStartTime ?? (newThinking ? Date.now() : null);
-            const prevSteps = prev.thinkingSteps || [];
-            const plainStep = {
-              id: `step-${prevSteps.length}`,
-              label: newThinking,
-              content: newThinking,
-              depth: payload.depth ?? 0,
-              timestamp: Date.now(),
-              agentName: payload.agentName,
-              parentAgent: payload.parentAgent
-            };
-            const next = {
-              ...prev,
-              status: newStatus,
-              thinking: prev.thinking + separator + newThinking,
-              thinkingSteps: [...prevSteps, plainStep],
-              thinkingStartTime,
-              firstMessageTime
-            };
-            return { ...next, timelineEntries: buildTimelineEntries(next) };
-          }
-          case "tool_call": {
-            const toolName = payload.message || payload.tool?.name;
-            if (toolName) {
-              const newToolCall = {
-                id: payload.tool?.id || `tool-${Date.now()}`,
-                name: toolName,
-                arguments: payload.tool?.arguments,
-                timestamp: Date.now(),
-                agentName: payload.agentName,
-                parentAgent: payload.parentAgent,
-                depth: payload.depth
-              };
-              const next = {
-                ...prev,
-                status: newStatus,
-                toolCalls: [...prev.toolCalls, newToolCall],
-                firstMessageTime
-              };
-              return { ...next, timelineEntries: buildTimelineEntries(next) };
-            }
-            return { ...prev, status: newStatus, firstMessageTime };
-          }
-          case "knowledge": {
-            const knowledgeContent = payload.message || payload.knowledge?.content;
-            if (knowledgeContent) {
-              const newKnowledge = {
-                id: payload.knowledge?.id || `knowledge-${Date.now()}`,
-                source: payload.knowledge?.source || "unknown",
-                content: knowledgeContent,
-                timestamp: Date.now(),
-                agentName: payload.agentName,
-                parentAgent: payload.parentAgent,
-                depth: payload.depth
-              };
-              const next = {
-                ...prev,
-                status: newStatus,
-                knowledge: [...prev.knowledge, newKnowledge],
-                firstMessageTime
-              };
-              return { ...next, timelineEntries: buildTimelineEntries(next) };
-            }
-            return { ...prev, status: newStatus, firstMessageTime };
-          }
-          case "memory": {
-            const memoryContent = payload.message || payload.memory?.content;
-            if (memoryContent) {
-              const newMemory = {
-                id: payload.memory?.id || `memory-${Date.now()}`,
-                type: payload.memory?.type || "unknown",
-                content: memoryContent,
-                timestamp: Date.now(),
-                agentName: payload.agentName,
-                parentAgent: payload.parentAgent,
-                depth: payload.depth
-              };
-              const next = {
-                ...prev,
-                status: newStatus,
-                memory: [...prev.memory, newMemory],
-                firstMessageTime
-              };
-              return { ...next, timelineEntries: buildTimelineEntries(next) };
-            }
-            return { ...prev, status: newStatus, firstMessageTime };
-          }
-          case "response":
-            return {
-              ...prev,
-              status: "complete",
-              response: payload.message || payload.content || "",
-              responseCompleteTime: Date.now(),
-              firstMessageTime: prev.firstMessageTime ?? Date.now()
-            };
-          case "status_update": {
-            const statusMessage = payload.message || payload.statusUpdate?.message;
-            if (statusMessage) {
-              const newStatusItem = {
-                id: payload.statusUpdate?.id || `status-${Date.now()}`,
-                message: statusMessage,
-                agent: payload.statusUpdate?.agent,
-                timestamp: Date.now(),
-                agentName: payload.agentName,
-                parentAgent: payload.parentAgent,
-                depth: payload.depth
-              };
-              const next = {
-                ...prev,
-                status: newStatus,
-                statusUpdates: [...prev.statusUpdates, newStatusItem],
-                firstMessageTime
-              };
-              return { ...next, timelineEntries: buildTimelineEntries(next) };
-            }
-            return { ...prev, status: newStatus, firstMessageTime };
-          }
-          case "potential_response": {
-            const respContent = payload.message || payload.content || "";
-            if (respContent) {
-              const newResp = {
-                id: `resp-${Date.now()}`,
-                content: respContent,
-                timestamp: Date.now(),
-                agentName: payload.agentName,
-                parentAgent: payload.parentAgent,
-                depth: payload.depth
-              };
-              const next = {
-                ...prev,
-                status: newStatus,
-                potentialResponses: [...prev.potentialResponses || [], newResp],
-                firstMessageTime
-              };
-              return { ...next, timelineEntries: buildTimelineEntries(next) };
-            }
-            return { ...prev, status: newStatus, firstMessageTime };
-          }
-          default:
-            return { ...prev, status: newStatus, firstMessageTime };
-        }
-      });
+      setState((prev) => reduceAgentMessage(prev, payload));
     },
     [topic]
   );
@@ -1345,6 +1349,7 @@ var AgentResponse = React11__namespace.forwardRef(
     renderMarkdown,
     renderThinkingMarkdown,
     timelineMaxHeight,
+    classNames,
     className,
     ...props
   }, ref) => {
@@ -1389,7 +1394,7 @@ var AgentResponse = React11__namespace.forwardRef(
         onMouseLeave: () => setIsHovered(false),
         ...props,
         children: [
-          /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "border border-border rounded-lg overflow-hidden", children: [
+          /* @__PURE__ */ jsxRuntime.jsxs("div", { className: core.cn("border border-border rounded-lg overflow-hidden", classNames?.container), children: [
             showMetadataRow && /* @__PURE__ */ jsxRuntime.jsxs(jsxRuntime.Fragment, { children: [
               /* @__PURE__ */ jsxRuntime.jsx(
                 MetadataRow,
@@ -1435,7 +1440,8 @@ var AgentResponse = React11__namespace.forwardRef(
               {
                 className: core.cn(
                   "bg-muted/50 p-4",
-                  showMetadataRow && "border-t border-border"
+                  showMetadataRow && "border-t border-border",
+                  classNames?.response
                 ),
                 children: renderMarkdown ? renderMarkdown(state.response) : /* @__PURE__ */ jsxRuntime.jsx("span", { className: "whitespace-pre-wrap", children: state.response })
               }
@@ -1856,6 +1862,7 @@ exports.formatTotalTime = formatTotalTime;
 exports.groupIntoAgentRuns = groupIntoAgentRuns;
 exports.initialAgentResponseState = initialAgentResponseState;
 exports.parseResponseSegments = parseResponseSegments;
+exports.reduceAgentMessage = reduceAgentMessage;
 exports.useAgentResponseAccumulator = useAgentResponseAccumulator;
 exports.useThinkingTimer = useThinkingTimer;
 //# sourceMappingURL=index.cjs.map