52 lines
1.1 KiB
Go
52 lines
1.1 KiB
Go
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
|
|
}
|