Upstream/waveterm
1
0
Fork 0
mirror of https://github.com/wavetermdev/waveterm.git synced 2024-12-22 16:48:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

Adding engineered query to chat command (#488)

Browse Source 
* added engineered query to chat command

* let ai know responses are rendered in markdown
This commit is contained in:
Cole Lashley 2024-03-21 18:00:55 -07:00 committed by GitHub
parent be37916ca6
commit 75a82de5bf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
wavesrv/pkg/cmdrunner/cmdrunner.go
View File

@ -2625,7 +2625,7 @@ func getCmdInfoEngineeredPrompt(userQuery string, curLineStr string, shellType s
// Enclose the command in triple backticks to format it as a code block. // Enclose the command in triple backticks to format it as a code block.
promptCurrentCommand = " The user is currently working with the command: ```\n" + curLineStr + "\n```\n\n" promptCurrentCommand = " The user is currently working with the command: ```\n" + curLineStr + "\n```\n\n"
} }
promptFormattingInstruction := "Please ensure any command line suggestions or code snippets that are meant to be run by the user are enclosed in triple backquotes for easy copy and paste into the terminal." promptFormattingInstruction := "Please ensure any command line suggestions or code snippets or scripts that are meant to be run by the user are enclosed in triple backquotes for easy copy and paste into the terminal. Also note that any response you give will be rendered in markdown."
promptQuestion := " The user's question is:\n\n" + userQuery + "" promptQuestion := " The user's question is:\n\n" + userQuery + ""
return promptBase + promptCurrentCommand + promptFormattingInstruction + promptQuestion return promptBase + promptCurrentCommand + promptFormattingInstruction + promptQuestion
@ -2875,7 +2875,9 @@ func OpenAICommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus
update := scbus.MakeUpdatePacket() update := scbus.MakeUpdatePacket()
return update, nil return update, nil
} }
prompt := []packet.OpenAIPromptMessageType{{Role: sstore.OpenAIRoleUser, Content: promptStr}} osType := GetOsTypeFromRuntime()
engineeredQuery := getCmdInfoEngineeredPrompt(promptStr, "", ids.Remote.ShellType, osType)
prompt := []packet.OpenAIPromptMessageType{{Role: sstore.OpenAIRoleUser, Content: engineeredQuery}}
if resolveBool(pk.Kwargs["cmdinfoclear"], false) { if resolveBool(pk.Kwargs["cmdinfoclear"], false) {
update := sstore.UpdateWithClearOpenAICmdInfo(cmd.ScreenId) update := sstore.UpdateWithClearOpenAICmdInfo(cmd.ScreenId)
if err != nil { if err != nil {