Refactored, added comprehensive testing.

api/payload.go

@@ -0,0 +1,36 @@
+package api
+
+import "git.wisehodl.dev/jay/aicli/config"
+
+// buildPayload constructs the JSON payload for the API request based on protocol.
+func buildPayload(cfg config.ConfigData, model string, query string) map[string]interface{} {
+	if cfg.Protocol == config.ProtocolOllama {
+		payload := map[string]interface{}{
+			"model":  model,
+			"prompt": query,
+			"stream": false,
+		}
+		if cfg.SystemPrompt != "" {
+			payload["system"] = cfg.SystemPrompt
+		}
+		return payload
+	}
+
+	// OpenAI protocol
+	messages := []map[string]string{}
+	if cfg.SystemPrompt != "" {
+		messages = append(messages, map[string]string{
+			"role":    "system",
+			"content": cfg.SystemPrompt,
+		})
+	}
+	messages = append(messages, map[string]string{
+		"role":    "user",
+		"content": query,
+	})
+
+	return map[string]interface{}{
+		"model":    model,
+		"messages": messages,
+	}
+}