85 lines
2.7 KiB
Go
85 lines
2.7 KiB
Go
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
|
|
}
|
|
|
|
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
|
|
}
|
|
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
|
|
}
|