feat(gemini): update "auto" chat on --output
This commit is contained in:
parent
e6737ba635
commit
48ba121435
31
doOutput.go
31
doOutput.go
|
@ -3,6 +3,7 @@ package main
|
||||||
import (
|
import (
|
||||||
"os"
|
"os"
|
||||||
|
|
||||||
|
"go.wit.com/lib/protobuf/chatpb"
|
||||||
"go.wit.com/log"
|
"go.wit.com/log"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -19,4 +20,34 @@ func doOutput(s string) {
|
||||||
}
|
}
|
||||||
|
|
||||||
log.Info("OUTPUT LOGGED TO", filename)
|
log.Info("OUTPUT LOGGED TO", filename)
|
||||||
|
|
||||||
|
// Load the existing chats.
|
||||||
|
all := chatpb.NewChats()
|
||||||
|
if err := all.ConfigLoad(); err != nil {
|
||||||
|
log.Warn("Error loading config, can't add to auto chat:", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Find the "auto" chat.
|
||||||
|
var autoChat *chatpb.Chat
|
||||||
|
for _, chat := range all.GetChats() {
|
||||||
|
if chat.GetChatName() == "auto" {
|
||||||
|
autoChat = chat
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// If the "auto" chat is found, add the new entry.
|
||||||
|
if autoChat != nil {
|
||||||
|
newEntry := &chatpb.ChatEntry{
|
||||||
|
From: chatpb.Who_GEMINI, // Or USER, depending on context
|
||||||
|
Content: s,
|
||||||
|
}
|
||||||
|
autoChat.Entries = append(autoChat.Entries, newEntry)
|
||||||
|
if err := all.ConfigSave(); err != nil {
|
||||||
|
log.Warn("Error saving config after adding to auto chat:", err)
|
||||||
|
} else {
|
||||||
|
log.Info("Added new entry to 'auto' chat.")
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue