refactor(config): remove image input and entity memory mode

nomihq · Oct 2, 2024 · 918ac76 · 918ac76
1 parent 5ecd8b3
commit 918ac76
Show file tree

Hide file tree

Showing 5 changed files with 4 additions and 21 deletions.
diff --git a/internal/config/config.go b/internal/config/config.go
@@ -77,7 +77,6 @@ func defaultConfig() Config {
 	return Config{
 		Input: InputConfig{
 			Text:  EnabledConfig{Enabled: true},
-			Image: EnabledConfig{Enabled: true},
 			Voice: EnabledConfig{Enabled: true},
 		},
 		Output: OutputConfig{

diff --git a/internal/config/paths.go b/internal/config/paths.go
@@ -8,7 +8,7 @@ import (
 
 const (
 	// configDir is the directory where the configuration file is stored.
-	configDir = ".ai"
+	configDir = ".golem"
 )
 
 // GetHomeDir returns the user's home directory.

diff --git a/internal/config/set.go b/internal/config/set.go
@@ -34,8 +34,6 @@ func setInputConfigValue(
 	switch keys[0] {
 	case "text":
 		enabled = &input.Text.Enabled
-	case "image":
-		enabled = &input.Image.Enabled
 	case "voice":
 		enabled = &input.Voice.Enabled
 	default:
@@ -99,8 +97,7 @@ func setMemoryConfigValue(
 		return fmt.Errorf("invalid key for memory: %v", keys)
 	}
 	switch MemoryMode(value) {
-	case MemoryModeKnowledgeGraph,
-		MemoryModeEntity,
+	case
 		MemoryModeConversation,
 		MemoryModeNone:
 		memory.Mode = MemoryMode(value)

diff --git a/internal/config/setup.go b/internal/config/setup.go
@@ -15,10 +15,6 @@ func Setup() error {
 		"Enable text input",
 		cfg.Input.Text.Enabled,
 	)
-	cfg.Input.Image.Enabled = promptForBool(
-		"Enable image input",
-		cfg.Input.Image.Enabled,
-	)
 	cfg.Input.Voice.Enabled = promptForBool(
 		"Enable voice input",
 		cfg.Input.Voice.Enabled,
@@ -47,8 +43,6 @@ func Setup() error {
 	}
 
 	memoryModes := []MemoryMode{
-		MemoryModeKnowledgeGraph,
-		MemoryModeEntity,
 		MemoryModeConversation,
 		MemoryModeNone,
 	}
@@ -125,10 +119,6 @@ func promptForSelect(label string, items []string, defaultIndex int) int {
 
 func getDefaultMemoryModeIndex(mode MemoryMode) int {
 	switch mode {
-	case MemoryModeKnowledgeGraph:
-		return 0
-	case MemoryModeEntity:
-		return 1
 	case MemoryModeConversation:
 		return 2
 	case MemoryModeNone:

diff --git a/internal/config/types.go b/internal/config/types.go
@@ -8,7 +8,6 @@ type Config struct {
 
 type InputConfig struct {
 	Text  EnabledConfig `yaml:"text"  json:"text"`
-	Image EnabledConfig `yaml:"image" json:"image"`
 	Voice EnabledConfig `yaml:"voice" json:"voice"`
 }
 
@@ -33,10 +32,8 @@ type OutputDetailConfig struct {
 type MemoryMode string
 
 const (
-	MemoryModeKnowledgeGraph MemoryMode = "knowledge_graph"
-	MemoryModeEntity         MemoryMode = "entity"
-	MemoryModeConversation   MemoryMode = "conversation"
-	MemoryModeNone           MemoryMode = "none"
+	MemoryModeConversation MemoryMode = "conversation"
+	MemoryModeNone         MemoryMode = "none"
 )
 
 func (m MemoryMode) String() string {