gogs
/
ai
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

break recurse generation if current output contains a delimiter using first OR last when combined with prev if not contains

master
bel 2023-06-17 14:01:13 -06:00
parent 68a87dc3db
commit 287b72639d
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
vicuna-tools.d/main.go
View File

@ -383,8 +383,8 @@ func chatBotGenerateAndFillInputF(ctx context.Context, cacheF, inputF, reversePr
commands[1:]..., commands[1:]...,
) )
command.Dir = Config.ChatBot.WD command.Dir = Config.ChatBot.WD
command.Stderr = io.Discard
command.Stderr = log.Writer() command.Stderr = log.Writer()
command.Stderr = io.Discard
stdout, err := command.StdoutPipe() stdout, err := command.StdoutPipe()
if err != nil { if err != nil {
@ -412,11 +412,13 @@ func chatBotGenerateAndFillInputF(ctx context.Context, cacheF, inputF, reversePr
return nil, err return nil, err
} }
justNew := oldAndNew[len(priorContent):] justNew := oldAndNew[len(priorContent):]
idx := bytes.LastIndex(
if idx := bytes.Index(justNew, []byte(reversePrompt)); idx > 0 {
justNew = justNew[:idx]
} else if idx := bytes.LastIndex(
append(priorContent, justNew...), append(priorContent, justNew...),
[]byte(reversePrompt), []byte(reversePrompt),
) ); idx+len(reversePrompt) > len(priorContent) {
if idx+len(reversePrompt) > len(priorContent) {
justNew = justNew[:idx+len(reversePrompt)-len(priorContent)] justNew = justNew[:idx+len(reversePrompt)-len(priorContent)]
} }
if err := _appendFile(inputF, string(justNew)); err != nil { if err := _appendFile(inputF, string(justNew)); err != nil {