clean up summarizer function
This commit is contained in:
@@ -5,7 +5,7 @@ import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"io"
|
||||
"net/http"
|
||||
"os"
|
||||
)
|
||||
@@ -62,25 +62,25 @@ func Summarize(text string) (string, error) {
|
||||
defer resp.Body.Close()
|
||||
|
||||
// Read the response
|
||||
body, err := ioutil.ReadAll(resp.Body)
|
||||
body, err := io.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
// Unmarshal the JSON response
|
||||
// Unmarshal the JSON response
|
||||
var response Response
|
||||
err = json.Unmarshal(body, &response)
|
||||
if err != nil {
|
||||
return "", err
|
||||
return "", err
|
||||
}
|
||||
|
||||
// Extract and print the content
|
||||
var content string
|
||||
var content string
|
||||
if len(response.Choices) > 0 {
|
||||
content = response.Choices[0].Message.Content
|
||||
} else {
|
||||
return "", errors.New("could not find content in response")
|
||||
return "", errors.New("could not find content in response")
|
||||
}
|
||||
|
||||
return content, nil
|
||||
return content, nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user