Merge pull request #48 from Hegghammer/custom_summarize_prompt

Author: piqoni

config.go

@@ -36,6 +36,7 @@ openai_api_key:
 openai_base_url:
 openai_model:
 summary_feeds:
+summary_prompt:
 show_images: false
 analyst_feeds:
   - https://feeds.bbci.co.uk/news/business/rss.xml
@@ -137,7 +138,9 @@ func bootstrapConfig() {
 	if viper.IsSet("openai_model") {
 		openaiModel = viper.Get("openai_model").(string)
 	}
-
+	if viper.IsSet("summary_prompt") {
+		summaryPrompt = viper.Get("summary_prompt").(string)
+	}
 	if viper.IsSet("summary_feeds") {
 		summaryFeeds := viper.Get("summary_feeds")
 
@@ -146,11 +149,12 @@ func bootstrapConfig() {
 			myFeeds = append(myFeeds, RSS{url: url, limit: limit, summarize: true})
 		}
 	}
-
-	for _, feed := range feeds.([]any) {
-		url, limit := getFeedAndLimit(feed.(string))
-		myFeeds = append(myFeeds, RSS{url: url, limit: limit})
-	}
+	if feeds != nil {
+			for _, feed := range feeds.([]any) {
+				url, limit := getFeedAndLimit(feed.(string))
+				myFeeds = append(myFeeds, RSS{url: url, limit: limit})
+			}
+		}
 
 	if viper.IsSet("google_news_keywords") {
 		googleNewsKeywords := url.QueryEscape(viper.Get("google_news_keywords").(string))

feeds_writer.go

@@ -28,6 +28,7 @@ var show_images bool
 var sunrise_sunset bool
 var myFeeds []RSS
 var db *sql.DB
+var summaryPrompt string
 
 type RSS struct {
 	url       string

summarize.go

@@ -7,6 +7,7 @@ import (
 
 	readability "github.com/go-shiori/go-readability"
 	openai "github.com/sashabaranov/go-openai"
+	"github.com/spf13/viper"
 )
 
 func getSummaryFromLink(url string) string {
@@ -30,6 +31,12 @@ func summarize(text string) string {
 	if len(text) < 200 {
 		return ""
 	}
+	
+	prompt := summaryPrompt
+	if prompt == "" {
+		prompt = "Summarize the following text:"
+	}
+	
 	clientConfig := openai.DefaultConfig(openaiApiKey)
 	if openaiBaseURL != "" {
 		clientConfig.BaseURL = openaiBaseURL
@@ -46,7 +53,7 @@ func summarize(text string) string {
 			Messages: []openai.ChatCompletionMessage{
 				{
 					Role:    openai.ChatMessageRoleAssistant,
-					Content: "Summarize the following text:",
+					Content: prompt,
 				},
 				{
 					Role:    openai.ChatMessageRoleUser,