1 files changed, 94 insertions, 0 deletions
diff --git a/config/config.go b/config/config.go
new file mode 100644
index 0000000..e612aa7
--- /dev/null
+++ b/config/config.go
@@ -0,0 +1,94 @@
+package config
+
+import (
+	"fmt"
+
+	"github.com/BurntSushi/toml"
+)
+
+type Config struct {
+	EnableCluedo    bool   `toml:"EnableCluedo"` // Cluedo game mode toggle
+	CluedoRole2     string `toml:"CluedoRole2"`  // Secondary AI role name
+	ChatAPI         string `toml:"ChatAPI"`
+	CompletionAPI   string `toml:"CompletionAPI"`
+	CurrentAPI      string
+	CurrentProvider string
+	APIMap          map[string]string
+	//
+	ShowSys       bool   `toml:"ShowSys"`
+	LogFile       string `toml:"LogFile"`
+	UserRole      string `toml:"UserRole"`
+	ToolRole      string `toml:"ToolRole"`
+	ToolUse       bool   `toml:"ToolUse"`
+	ThinkUse      bool   `toml:"ThinkUse"`
+	AssistantRole string `toml:"AssistantRole"`
+	SysDir        string `toml:"SysDir"`
+	ChunkLimit    uint32 `toml:"ChunkLimit"`
+	// embeddings
+	RAGEnabled bool   `toml:"RAGEnabled"`
+	EmbedURL   string `toml:"EmbedURL"`
+	HFToken    string `toml:"HFToken"`
+	RAGDir     string `toml:"RAGDir"`
+	// rag settings
+	RAGWorkers   uint32 `toml:"RAGWorkers"`
+	RAGBatchSize int    `toml:"RAGBatchSize"`
+	RAGWordLimit uint32 `toml:"RAGWordLimit"`
+	// deepseek
+	DeepSeekChatAPI       string `toml:"DeepSeekChatAPI"`
+	DeepSeekCompletionAPI string `toml:"DeepSeekCompletionAPI"`
+	DeepSeekToken         string `toml:"DeepSeekToken"`
+	DeepSeekModel         string `toml:"DeepSeekModel"`
+	ApiLinks              []string
+	// TTS
+	TTS_URL     string  `toml:"TTS_URL"`
+	TTS_ENABLED bool    `toml:"TTS_ENABLED"`
+	TTS_SPEED   float32 `toml:"TTS_SPEED"`
+	// STT
+	STT_URL     string `toml:"STT_URL"`
+	STT_ENABLED bool   `toml:"STT_ENABLED"`
+}
+
+func LoadConfigOrDefault(fn string) *Config {
+	if fn == "" {
+		fn = "config.toml"
+	}
+	config := &Config{}
+	_, err := toml.DecodeFile(fn, &config)
+	if err != nil {
+		fmt.Println("failed to read config from file, loading default", "error", err)
+		config.ChatAPI = "http://localhost:8080/v1/chat/completions"
+		config.CompletionAPI = "http://localhost:8080/completion"
+		config.DeepSeekCompletionAPI = "https://api.deepseek.com/beta/completions"
+		config.DeepSeekChatAPI = "https://api.deepseek.com/chat/completions"
+		config.RAGEnabled = false
+		config.EmbedURL = "http://localhost:8080/v1/embiddings"
+		config.ShowSys = true
+		config.LogFile = "log.txt"
+		config.UserRole = "user"
+		config.ToolRole = "tool"
+		config.AssistantRole = "assistant"
+		config.SysDir = "sysprompts"
+		config.ChunkLimit = 8192
+		//
+		config.RAGBatchSize = 100
+		config.RAGWordLimit = 80
+		config.RAGWorkers = 5
+		// tts
+		config.TTS_ENABLED = false
+		config.TTS_URL = "http://localhost:8880/v1/audio/speech"
+	}
+	config.CurrentAPI = config.ChatAPI
+	config.APIMap = map[string]string{
+		config.ChatAPI:               config.CompletionAPI,
+		config.CompletionAPI:         config.DeepSeekChatAPI,
+		config.DeepSeekChatAPI:       config.DeepSeekCompletionAPI,
+		config.DeepSeekCompletionAPI: config.ChatAPI,
+	}
+	for _, el := range []string{config.ChatAPI, config.CompletionAPI, config.DeepSeekChatAPI, config.DeepSeekCompletionAPI} {
+		if el != "" {
+			config.ApiLinks = append(config.ApiLinks, el)
+		}
+	}
+	// if any value is empty fill with default
+	return config
+}