blob: 20f0d4a3be3876b56b322a21b2a15230fda55566 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
// Copyright 2017-2025 WIT.COM Inc. All rights reserved.
// Use of this source code is governed by the GPL 3.0
package main
// An app to submit patches for the 30 GO GUI repos
import (
"embed"
"github.com/google/uuid"
"go.wit.com/lib/protobuf/argvpb"
"go.wit.com/lib/protobuf/chatpb"
"go.wit.com/log"
)
// sent via -ldflags
var VERSION string
var BUILDTIME string
// this optionally can store the GUI plugins
//
//go:embed resources/*
var resources embed.FS
// used for shell auto completion
var ARGNAME string = "regex"
// using this for now. triggers config save
var configSave bool
func main() {
me = new(mainType)
// me.myGui = prep.Gui() // prepares the GUI package for go-args
me.argv = argvpb.Autocomplete(&argv) // adds shell auto complete to go-args
var err error
// load the default chat protobuf
me.chats = chatpb.NewChats()
if err := me.chats.ConfigLoad(); err != nil {
badExit(err)
}
// verify all the chats have Uuid's
if verifyUuids(me.chats) {
me.chats.ConfigSave()
}
// Get the last chat
numChats := len(me.chats.GetChats())
if numChats > 0 {
me.lastChat = me.chats.GetChats()[numChats-1]
log.Printf("The current Gemini API session is UUID: %s\n", me.lastChat.GetUuid())
}
err = initGeminiAPI()
if err != nil {
badExit(err)
}
if argv.JsonFile != "" {
doJSON()
okExit("")
}
if argv.Interact != nil {
log.Info("testing AI client with simpleHello()")
err = simpleHello()
if err != nil {
badExit(err)
}
doInteract()
okExit("")
}
if argv.Stats != "" {
doStats()
okExit("")
}
if argv.NewChat != nil {
doNewChat()
okExit("")
}
if argv.Playback != nil {
if argv.Uuid != "" {
showChat(argv.Uuid)
} else {
doPlayback()
}
okExit("")
}
if argv.Clean != nil {
doClean()
okExit("")
}
me.myGui.Start() // loads the GUI toolkit
doGui() // start making our forge GUI
debug() // sits here forever
}
func verifyUuids(chats *chatpb.Chats) bool {
var changed bool
for _, chat := range chats.GetChats() {
if chat.GetUuid() == "" {
chat.Uuid = uuid.New().String()
changed = true
}
for _, entry := range chat.GetEntries() {
if entry.GetUuid() == "" {
entry.Uuid = uuid.New().String()
changed = true
}
}
}
return changed
}
|
