fix: command.join() is not a function (#270)

* fix: command.join() is not a function

* fix: better typing

apps/web/src/components/gen-ui/action-step.tsx

@@ -12,6 +12,18 @@ import {
   MessageSquare,
   FileText,
 } from "lucide-react";
+import {
+  createApplyPatchToolFields,
+  createShellToolFields,
+} from "@open-swe/shared/open-swe/tools";
+import { z } from "zod";
+
+// Used only for Zod type inference.
+const dummyRepo = { owner: "dummy", repo: "dummy" };
+const shellTool = createShellToolFields(dummyRepo);
+type ShellToolArgs = z.infer<typeof shellTool.schema>;
+const applyPatchTool = createApplyPatchToolFields(dummyRepo);
+type ApplyPatchToolArgs = z.infer<typeof applyPatchTool.schema>;
 
 // Common props for all action types
 type BaseActionProps = {
@@ -21,23 +33,23 @@ type BaseActionProps = {
   summaryText?: string;
 };
 
-// Shell command specific props
+// Shell command specific props. We need to wrap the args in Partial<...>
-type ShellActionProps = BaseActionProps & {
+// because even though they're required, they may be undefined at a point in time
-  actionType: "shell";
+// due to streaming.
-  command: string[];
+type ShellActionProps = BaseActionProps &
-  workdir?: string;
+  Partial<ShellToolArgs> & {
-  output?: string;
+    actionType: "shell";
-  errorCode?: number;
+    output?: string;
-};
+    errorCode?: number;
+  };
 
 // Apply patch specific props
-type PatchActionProps = BaseActionProps & {
+type PatchActionProps = BaseActionProps &
-  actionType: "apply-patch";
+  Partial<ApplyPatchToolArgs> & {
-  file: string;
+    actionType: "apply-patch";
-  diff?: string;
+    errorMessage?: string;
-  errorMessage?: string;
+    fixedDiff?: string;
-  fixedDiff?: string;
+  };
-};
 
 // Union type for all possible action props
 export type ActionStepProps =
@@ -137,14 +149,14 @@ export function ActionStep(props: ActionStepProps) {
             </div>
           )}
           <code className="text-foreground/80 text-xs font-normal">
-            {props.command.join(" ")}
+            {props.command?.join(" ")}
           </code>
         </div>
       );
     } else {
       return (
         <code className="text-foreground/80 flex-1 text-xs font-normal">
-          {props.file}
+          {props.file_path}
         </code>
       );
     }

apps/web/src/components/thread/messages/ai.tsx

@@ -128,7 +128,7 @@ export function mapToolMessageToActionStepProps(
       actionType: "apply-patch",
       status,
       success,
-      file: args.file_path || "",
+      file_path: args.file_path || "",
       diff: args.diff,
       reasoningText,
       errorMessage: !success ? getContentString(message.content) : undefined,
@@ -222,24 +222,18 @@ export function AssistantMessage({
         actionType={aiToolCallName === shellTool.name ? "shell" : "apply-patch"}
         status="generating"
         command={
-          aiToolCallName === shellTool.name
+          aiToolCallName === shellTool.name ? aiToolCallArgs?.command || [] : []
-            ? aiToolCallArgs?.command || []
-            : undefined
         }
         workdir={
-          aiToolCallName === shellTool.name
+          aiToolCallName === shellTool.name ? aiToolCallArgs?.workdir : ""
-            ? aiToolCallArgs?.workdir
-            : undefined
         }
-        file={
+        file_path={
           aiToolCallName === applyPatchTool.name
             ? aiToolCallArgs?.file_path || ""
-            : undefined
+            : ""
         }
         diff={
-          aiToolCallName === applyPatchTool.name
+          aiToolCallName === applyPatchTool.name ? aiToolCallArgs?.diff : ""
-            ? aiToolCallArgs?.diff
-            : undefined
         }
         reasoningText={contentString}
       />