added a new option, context. fixed a bug with the custom commands where the context was not being treated as suchHEADmain

1 files changed, 49 insertions, 3 deletions

diff --git a/main.go b/main.go
index b4c3301..4999c9a 100644
--- a/main.go
+++ b/main.go
@@ -345,7 +345,7 @@ func handleCustomCommand(
 
 		for _, customContext := range customCommand.Context {
 			gptMemory = append(gptMemory, openai.ChatCompletionMessage{
-				Role:    openai.ChatMessageRoleUser,
+				Role:    openai.ChatMessageRoleAssistant,
 				Content: customContext,
 			})
 		}
@@ -376,7 +376,7 @@ func handleCustomCommand(
 				Parts: []genai.Part{
 					genai.Text(customContext),
 				},
-				Role: "user",
+				Role: "model",
 			})
 		}
 
@@ -396,7 +396,7 @@ func handleCustomCommand(
 
 		for _, customContext := range customCommand.Context {
 			ollamaMemory = append(ollamaMemory, MemoryElement{
-				Role:    "user",
+				Role:    "assistant",
 				Content: customContext,
 			})
 		}
@@ -649,6 +649,13 @@ func DoOllamaRequest(
 
 	if len(*ollamaMemory) > appConfig.MemoryLimit {
 		*ollamaMemory = []MemoryElement{}
+
+		for _, context := range appConfig.Context {
+			*ollamaMemory = append(*ollamaMemory, MemoryElement{
+				Role:    "assistant",
+				Content: context,
+			})
+		}
 	}
 
 	*ollamaMemory = append(*ollamaMemory, memoryElement)
@@ -887,6 +894,15 @@ func GeminiRequestProcessor(
 
 	if len(*geminiMemory) > appConfig.MemoryLimit {
 		*geminiMemory = []*genai.Content{}
+
+		for _, context := range appConfig.Context {
+			*geminiMemory = append(*geminiMemory, &genai.Content{
+				Parts: []genai.Part{
+					genai.Text(context),
+				},
+				Role: "model",
+			})
+		}
 	}
 
 	*geminiMemory = append(*geminiMemory, &genai.Content{
@@ -1036,6 +1052,13 @@ func ChatGPTRequestProcessor(
 
 	if len(*gptMemory) > appConfig.MemoryLimit {
 		*gptMemory = []openai.ChatCompletionMessage{}
+
+		for _, context := range appConfig.Context {
+			*gptMemory = append(*gptMemory, openai.ChatCompletionMessage{
+				Role:    openai.ChatMessageRoleAssistant,
+				Content: context,
+			})
+		}
 	}
 
 	var writer bytes.Buffer
@@ -1312,10 +1335,33 @@ func runIRC(appConfig TomlConfig) {
 
 	switch appConfig.Provider {
 	case "ollama":
+		for _, context := range appConfig.Context {
+			OllamaMemory = append(OllamaMemory, MemoryElement{
+				Role:    "assistant",
+				Content: context,
+			})
+		}
+
 		OllamaHandler(irc, &appConfig, &OllamaMemory)
 	case "gemini":
+		for _, context := range appConfig.Context {
+			GeminiMemory = append(GeminiMemory, &genai.Content{
+				Parts: []genai.Part{
+					genai.Text(context),
+				},
+				Role: "model",
+			})
+		}
+
 		GeminiHandler(irc, &appConfig, &GeminiMemory)
 	case "chatgpt":
+		for _, context := range appConfig.Context {
+			GPTMemory = append(GPTMemory, openai.ChatCompletionMessage{
+				Role:    openai.ChatMessageRoleAssistant,
+				Content: context,
+			})
+		}
+
 		ChatGPTHandler(irc, &appConfig, &GPTMemory)
 	}