eliaskohout/crowsnest
1
0
Fork 0
Code Issues 8 Pull Requests Actions Packages Projects Releases Wiki Activity

add structure for ai short summaries

Browse Source
This commit is contained in:
Elias Kohout
2025-01-07 15:26:26 +01:00
parent 321ccfe44d
commit 706ebe25a0
5 changed files with 43 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
internal/model/article.go
View File

@@ -13,22 +13,28 @@ type Article struct {
FetchDate time.Time
Title string
Content string
AiSummary string
}
// TODO docstring
type ArticleViewModel struct {
Title string
PublishDate string
SourceUrl string
ShortSource string
Summary string
Title string
PublishDate string
SourceUrl string
ShortSource string
Summary string
AiSummarized bool
}
// TODO docstring
func (a *Article) ViewModel() *ArticleViewModel {
summary := a.Content
if len(a.Content) > 300 {
summary = summary[:300]
summary := a.AiSummary
if summary == "" {
if len(a.Content) > 200 {
summary = a.Content[:200]
} else {
summary = a.Content
}
}
short_url := ""
@@ -38,10 +44,11 @@ func (a *Article) ViewModel() *ArticleViewModel {
}
return &ArticleViewModel{
Title: a.Title,
PublishDate: a.PublishDate.Local().Format("02.01.2006"),
SourceUrl: a.SourceUrl,
ShortSource: short_url,
Summary: summary,
Title: a.Title,
PublishDate: a.PublishDate.Local().Format("02.01.2006"),
SourceUrl: a.SourceUrl,
ShortSource: short_url,
Summary: summary,
AiSummarized: a.AiSummary != "",
}
}

16
internal/model/database/articles.go
View File

@@ -13,7 +13,7 @@ type ArticleModel struct {
// the connection to the database fails.
func (m *ArticleModel) All(limit int) ([]model.Article, error) {
stmt := `
SELECT id, title, sourceUrl, content, publishDate, fetchDate
SELECT id, title, sourceUrl, content, publishDate, fetchDate, aisummary
FROM articles
ORDER BY publishDate DESC
LIMIT $1
@@ -26,7 +26,7 @@ func (m *ArticleModel) All(limit int) ([]model.Article, error) {
articles := []model.Article{}
for rows.Next() {
a := model.Article{}
err := rows.Scan(&a.Identifier, &a.Title, &a.SourceUrl, &a.Content, &a.PublishDate, &a.FetchDate)
err := rows.Scan(&a.Identifier, &a.Title, &a.SourceUrl, &a.Content, &a.PublishDate, &a.FetchDate, &a.AiSummary)
if err != nil {
return nil, err
}
@@ -80,19 +80,19 @@ func (m *ArticleModel) Search(query string) ([]model.Article, error) {
// query fails.
func (m *ArticleModel) Insert(a *model.Article) error {
// insert article
stmt := `INSERT INTO articles (title, sourceUrl, content, publishDate, fetchDate)
VALUES ($1, $2, $3, $4, $5)
stmt := `INSERT INTO articles (title, sourceUrl, content, publishDate, fetchDate, aisummary)
VALUES ($1, $2, $3, $4, $5, $6)
`
_, err := m.DB.Exec(stmt, a.Title, a.SourceUrl, a.Content, a.PublishDate, a.FetchDate)
_, err := m.DB.Exec(stmt, a.Title, a.SourceUrl, a.Content, a.PublishDate, a.FetchDate, a.AiSummary)
return err
}
// TODO docstring
func (m *ArticleModel) Update(a *model.Article) error {
stmt := `UPDATE articles
SET title = $1, sourceUrl = $2, content = $4, publishDate = $5, fetchDate = $6
WHERE id = $7
SET title = $1, sourceUrl = $2, content = $4, publishDate = $5, fetchDate = $6, aisummary = $7
WHERE id = $8
`
_, err := m.DB.Exec(stmt, a.Title, a.SourceUrl, a.Content, a.PublishDate, a.FetchDate, a.Identifier)
_, err := m.DB.Exec(stmt, a.Title, a.SourceUrl, a.Content, a.PublishDate, a.FetchDate, a.AiSummary, a.Identifier)
return err
}