Skip to content

Commit

Permalink
removed stale prints and comments, readded non stream completion for now
Browse files Browse the repository at this point in the history
  • Loading branch information
colel1410@gmail.com committed Dec 15, 2023
1 parent fabb26c commit 8abde7e
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions wavesrv/pkg/cmdrunner/cmdrunner.go
Original file line number Diff line number Diff line change
Expand Up @@ -1449,7 +1449,6 @@ func doOpenAICompletion(cmd *sstore.CmdType, opts *sstore.OpenAIOptsType, prompt
}()
var respPks []*packet.OpenAIPacketType
var err error
log.Printf("TODO: fix this condition - short circuited to access cloud code path\n")
// run open ai completion locally
respPks, err = openai.RunCompletion(ctx, opts, prompt)
if err != nil {
Expand Down Expand Up @@ -1502,7 +1501,6 @@ func doOpenAIStreamCompletion(cmd *sstore.CmdType, opts *sstore.OpenAIOptsType,
}()
var ch chan *packet.OpenAIPacketType
var err error
log.Printf("TODO: fix this condition - short circuited to access cloud code path\n")
if opts.APIToken == "" {
clientData, err := sstore.EnsureClientData(ctx)
if err != nil {
Expand All @@ -1513,10 +1511,8 @@ func doOpenAIStreamCompletion(cmd *sstore.CmdType, opts *sstore.OpenAIOptsType,
writeErrorToPty(cmd, fmt.Sprintf("Error: must have telemetry enabled to use wave cloud"), outputPos)
return
}
// run open ai completion in the cloud
ch, err = openai.RunCloudCompletionStream(ctx, opts, prompt)
} else {
// run open ai completion locally
ch, err = openai.RunCompletionStream(ctx, opts, prompt)
}
if err != nil {
Expand Down Expand Up @@ -1592,7 +1588,11 @@ func OpenAICommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (sstor
return nil, fmt.Errorf("cannot add new line: %v", err)
}
prompt := []sstore.OpenAIPromptMessageType{{Role: sstore.OpenAIRoleUser, Content: promptStr}}
go doOpenAIStreamCompletion(cmd, opts, prompt)
if resolveBool(pk.Kwargs["stream"], true) {
go doOpenAIStreamCompletion(cmd, opts, prompt)
} else {
go doOpenAICompletion(cmd, opts, prompt)
}
updateHistoryContext(ctx, line, cmd)
updateMap := make(map[string]interface{})
updateMap[sstore.ScreenField_SelectedLine] = line.LineNum
Expand Down

0 comments on commit 8abde7e

Please sign in to comment.