Refactored, added comprehensive testing.

api/parse.go

@@ -0,0 +1,51 @@
+package api
+
+import (
+	"encoding/json"
+	"fmt"
+
+	"git.wisehodl.dev/jay/aicli/config"
+)
+
+// parseResponse extracts the response content from the API response body.
+func parseResponse(body []byte, protocol config.APIProtocol) (string, error) {
+	var result map[string]interface{}
+	if err := json.Unmarshal(body, &result); err != nil {
+		return "", fmt.Errorf("parse response: %w", err)
+	}
+
+	if protocol == config.ProtocolOllama {
+		response, ok := result["response"].(string)
+		if !ok {
+			return "", fmt.Errorf("no response field in ollama response")
+		}
+		return response, nil
+	}
+
+	// OpenAI protocol
+	choices, ok := result["choices"].([]interface{})
+	if !ok {
+		return "", fmt.Errorf("no choices in response")
+	}
+
+	if len(choices) == 0 {
+		return "", fmt.Errorf("empty choices array")
+	}
+
+	firstChoice, ok := choices[0].(map[string]interface{})
+	if !ok {
+		return "", fmt.Errorf("invalid choice format")
+	}
+
+	message, ok := firstChoice["message"].(map[string]interface{})
+	if !ok {
+		return "", fmt.Errorf("no message in choice")
+	}
+
+	content, ok := message["content"].(string)
+	if !ok {
+		return "", fmt.Errorf("no content in message")
+	}
+
+	return content, nil
+}