grail/gf-lt
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix: listen to the finish signal

Browse Source
This commit is contained in:
Grail Finder
2025-09-11 13:53:17 +03:00
parent 0068cd17ff
commit ff24ad4af7
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
bot.go
View File

@@ -190,7 +190,6 @@ func sendMsgToLLM(body io.Reader) {
for { for {
var ( var (
answerText string answerText string
stop bool
chunk *models.TextChunk chunk *models.TextChunk
) )
counter++ counter++
@@ -239,9 +238,11 @@ func sendMsgToLLM(body io.Reader) {
streamDone <- true streamDone <- true
break break
} }
if stop { if chunk.Finished {
if chunk.Chunk != "" { if chunk.Chunk != "" {
logger.Warn("text inside of finish llmchunk", "chunk", chunk, "counter", counter) logger.Warn("text inside of finish llmchunk", "chunk", chunk, "counter", counter)
answerText = strings.ReplaceAll(chunk.Chunk, "\n\n", "\n")
chunkChan <- answerText
} }
streamDone <- true streamDone <- true
break break