57 lines
1.1 KiB
Go
57 lines
1.1 KiB
Go
package main
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"time"
|
|
|
|
"go.wit.com/lib/protobuf/chatpb"
|
|
"go.wit.com/log"
|
|
"google.golang.org/protobuf/types/known/timestamppb"
|
|
)
|
|
|
|
func doImport(filename string) {
|
|
content, err := ioutil.ReadFile(filename)
|
|
if err != nil {
|
|
log.Warn("Error reading import file:", err)
|
|
return
|
|
}
|
|
|
|
s := string(content)
|
|
|
|
// 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_REGEX,
|
|
Ctime: timestamppb.New(time.Now()),
|
|
ToolCalls: []*chatpb.ToolCall{
|
|
{
|
|
Name: "Shell",
|
|
Input: 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.")
|
|
}
|
|
}
|
|
}
|