Skip to content

Commit 28ef94e

Browse files
authored
fix wsh ai arg handling after "-" (for stdin) (wavetermdev#2129)
1 parent d405a79 commit 28ef94e

2 files changed

Lines changed: 9 additions & 0 deletions

File tree

.vscode/settings.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
"editor.tabSize": 4,
66
"editor.insertSpaces": false,
77
"prettier.useEditorConfig": true,
8+
"diffEditor.renderSideBySide": false,
89
"[javascript]": {
910
"editor.defaultFormatter": "esbenp.prettier-vscode"
1011
},

cmd/wsh/cmd/wshcmd-ai.go

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -135,6 +135,14 @@ func aiRun(cmd *cobra.Command, args []string) (rtnErr error) {
135135
return fmt.Errorf("reading from stdin: %w", err)
136136
}
137137
message.Write(data)
138+
139+
// Also include any remaining arguments (excluding the "-" itself)
140+
if len(args) > 1 {
141+
if message.Len() > 0 {
142+
message.WriteString(" ")
143+
}
144+
message.WriteString(strings.Join(args[1:], " "))
145+
}
138146
} else {
139147
message.WriteString(strings.Join(args, " "))
140148
}

0 commit comments

Comments
 (0)