refactoring, weather fix

analyst.go

@@ -1,85 +1,67 @@
 package main
 
 import (
-	"context"
-	"fmt"
-	"strings"
-
 	"github.com/mmcdole/gofeed"
-	openai "github.com/sashabaranov/go-openai"
-	"github.com/spf13/viper"
 )
 
-const defaultLimit = 20 // default number of articles per feed for analysis
-var model = openai.GPT4o
+const (
+	analystTag   = "#analyst"
+	defaultLimit = 20
+)
 
-func generateAnalysis(fp *gofeed.Parser, writer Writer) {
-	if !viper.IsSet("analyst_feeds") || !viper.IsSet("analyst_prompt") {
+func RunAnalyst(cfg *Config, store *Storage, llm *LLMClient, writer Writer, fp *gofeed.Parser) {
+	if len(cfg.AnalystFeeds) == 0 || cfg.AnalystPrompt == "" {
 		return
 	}
-
-	analystFeeds := viper.GetStringSlice("analyst_feeds")
-	analystPrompt := viper.GetString("analyst_prompt")
-	analystModel := viper.GetString("analyst_model")
-
-	var articleTitles []string
-	for _, feedURL := range analystFeeds {
-		parsedFeed := parseFeed(fp, feedURL, defaultLimit)
-		if parsedFeed == nil {
-			continue
-		}
-		for _, item := range parsedFeed.Items {
-			seen, seen_today, summary := isSeenArticle(item, "#analyst")
-			if seen {
-				continue
-			}
-			articleTitles = append(articleTitles, item.Title+":  "+item.Description) // add also description for better context
-			if !seen_today {
-				addToSeenTable(item.Link+"#analyst", summary)
-			}
-		}
+	if llm == nil {
+		return
 	}
 
-	if len(articleTitles) == 0 {
+	articles := collectArticlesForAnalysis(cfg, store, fp)
+	if len(articles) == 0 {
 		return
 	}
 
-	prompt := fmt.Sprintf("%s\n\n%s", analystPrompt, strings.Join(articleTitles, "\n"))
-	analysis := getLLMAnalysis(prompt, analystModel)
-
+	analysis := llm.Analyze(articles)
 	if analysis != "" {
-		writer.write("\n## Daily Analysis:\n")
-		writer.write(analysis + "\n")
+		writer.Write("\n## Daily Analysis:\n")
+		writer.Write(analysis + "\n")
 	}
 }
 
-func getLLMAnalysis(prompt string, analystModel string) string {
-	clientConfig := openai.DefaultConfig(openaiApiKey)
-	if openaiBaseURL != "" {
-		clientConfig.BaseURL = openaiBaseURL
-	}
-	if analystModel != "" {
-		model = analystModel
-	}
-	client := openai.NewClientWithConfig(clientConfig)
+func collectArticlesForAnalysis(cfg *Config, store *Storage, fp *gofeed.Parser) []string {
+	var articles []string
 
-	resp, err := client.CreateChatCompletion(
-		context.Background(),
-		openai.ChatCompletionRequest{
-			Model: model,
-			Messages: []openai.ChatCompletionMessage{
-				{
-					Role:    openai.ChatMessageRoleUser,
-					Content: prompt,
-				},
-			},
-		},
-	)
+	for _, feedURL := range cfg.AnalystFeeds {
+		feed, err := fp.ParseURL(feedURL)
+		if err != nil {
+			continue
+		}
 
-	if err != nil {
-		fmt.Printf("ChatCompletion error: %v\n", err)
-		return ""
+		// Limit items
+		limit := defaultLimit
+		if len(feed.Items) > limit {
+			feed.Items = feed.Items[:limit]
+		}
+
+		for _, item := range feed.Items {
+			articleLink := item.Link + analystTag
+			seen, seenToday, summary := store.IsSeen(articleLink)
+
+			if seen {
+				continue
+			}
+
+			articleText := item.Title + ": " + item.Description
+			articles = append(articles, articleText)
+
+			if !seenToday {
+				if err := store.MarkAsSeen(articleLink, summary); err != nil {
+					continue
+				}
+			}
+		}
 	}
 
-	return resp.Choices[0].Message.Content
+	return articles
 }