aboutsummaryrefslogtreecommitdiffstats
path: root/main.go
diff options
context:
space:
mode:
authorterminaldweller <devi@terminaldweller.com>2024-06-03 15:09:18 +0000
committerterminaldweller <devi@terminaldweller.com>2024-06-03 15:09:18 +0000
commit165f49620cec0272952f3bc03001a5dbabd6b2ef (patch)
tree8796b60780a4c03b92b1d4107ca97ffe891b1729 /main.go
parenttables are working now (diff)
downloadmilla-165f49620cec0272952f3bc03001a5dbabd6b2ef.tar.gz
milla-165f49620cec0272952f3bc03001a5dbabd6b2ef.zip
some cleanup
Diffstat (limited to 'main.go')
-rw-r--r--main.go97
1 files changed, 0 insertions, 97 deletions
diff --git a/main.go b/main.go
index b4ebe04..16f249c 100644
--- a/main.go
+++ b/main.go
@@ -42,77 +42,6 @@ var (
errUnsupportedType = errors.New("unsupported type")
)
-type LogModel struct {
- // Id int64 `db:"id"`
- // Channel string `db:"channel"`
- Log string `db:"log"`
- // Nick string `db:"nick"`
- // DateAdded pgtype.Date `db:"dateadded"`
-}
-
-type CustomCommand struct {
- SQL string `toml:"sql"`
- Limit int `toml:"limit"`
- Prompt string `toml:"prompt"`
-}
-
-type TomlConfig struct {
- IrcServer string `toml:"ircServer"`
- IrcNick string `toml:"ircNick"`
- IrcSaslUser string `toml:"ircSaslUser"`
- IrcSaslPass string `toml:"ircSaslPass"`
- OllamaEndpoint string `toml:"ollamaEndpoint"`
- Model string `toml:"model"`
- ChromaStyle string `toml:"chromaStyle"`
- ChromaFormatter string `toml:"chromaFormatter"`
- Provider string `toml:"provider"`
- Apikey string `toml:"apikey"`
- OllamaSystem string `toml:"ollamaSystem"`
- ClientCertPath string `toml:"clientCertPath"`
- ServerPass string `toml:"serverPass"`
- Bind string `toml:"bind"`
- Name string `toml:"name"`
- DatabaseAddress string `toml:"databaseAddress"`
- DatabasePassword string `toml:"databasePassword"`
- DatabaseUser string `toml:"databaseUser"`
- DatabaseName string `toml:"databaseName"`
- LLMProxy string `toml:"llmProxy"`
- IRCProxy string `toml:"ircProxy"`
- IRCDName string `toml:"ircdName"`
- WebIRCPassword string `toml:"webIRCPassword"`
- WebIRCGateway string `toml:"webIRCGateway"`
- WebIRCHostname string `toml:"webIRCHostname"`
- WebIRCAddress string `toml:"webIRCAddress"`
- PluginPath string `toml:"pluginPath"`
- CustomCommands map[string]CustomCommand `toml:"customCommands"`
- Temp float64 `toml:"temp"`
- RequestTimeout int `toml:"requestTimeout"`
- MillaReconnectDelay int `toml:"millaReconnectDelay"`
- IrcPort int `toml:"ircPort"`
- KeepAlive int `toml:"keepAlive"`
- MemoryLimit int `toml:"memoryLimit"`
- PingDelay int `toml:"pingDelay"`
- PingTimeout int `toml:"pingTimeout"`
- TopP float32 `toml:"topP"`
- TopK int32 `toml:"topK"`
- EnableSasl bool `toml:"enableSasl"`
- SkipTLSVerify bool `toml:"skipTLSVerify"`
- UseTLS bool `toml:"useTLS"`
- DisableSTSFallback bool `toml:"disableSTSFallback"`
- AllowFlood bool `toml:"allowFlood"`
- Debug bool `toml:"debug"`
- Out bool `toml:"out"`
- AdminOnly bool `toml:"adminOnly"`
- pool *pgxpool.Pool
- Admins []string `toml:"admins"`
- IrcChannels []string `toml:"ircChannels"`
- ScrapeChannels []string `toml:"scrapeChannels"`
-}
-
-type AppConfig struct {
- Ircd map[string]TomlConfig `toml:"ircd"`
-}
-
func addSaneDefaults(config *TomlConfig) {
if config.IrcNick == "" {
config.IrcNick = "milla"
@@ -175,32 +104,6 @@ func addSaneDefaults(config *TomlConfig) {
}
}
-type OllamaRequestOptions struct {
- Temperature float64 `json:"temperature"`
-}
-
-type OllamaChatResponse struct {
- Role string `json:"role"`
- Content string `json:"content"`
-}
-
-type OllamaChatMessagesResponse struct {
- Messages OllamaChatResponse `json:"message"`
-}
-
-type OllamaChatRequest struct {
- Model string `json:"model"`
- Stream bool `json:"stream"`
- KeepAlive time.Duration `json:"keep_alive"`
- Options OllamaRequestOptions `json:"options"`
- Messages []MemoryElement `json:"messages"`
-}
-
-type MemoryElement struct {
- Role string `json:"role"`
- Content string `json:"content"`
-}
-
func getTableFromChanName(channel, ircdName string) string {
tableName := ircdName + "_" + channel
tableName = strings.ReplaceAll(tableName, "#", "")