experiments/bees-knees
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Generate both title and summary on improvement

Browse source
This commit is contained in:
Marty Oehme 2025-06-05 18:33:07 +02:00
parent 24dc6fc6c8
commit 19c5dd2e60
Signed by: Marty
GPG key ID: 4E535BC19C61886E
1 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

10
prophet/app.py
View file

@ -12,7 +12,7 @@ from fastapi.middleware.cors import CORSMiddleware
from groq import Groq from groq import Groq
BEE_FEED = "https://babylonbee.com/feed" BEE_FEED = "https://babylonbee.com/feed"
BEE_FEED_TEST = "test/resources/feed.atom" # NOTE: Switch out when done testing BEE_FEED_TEST = "test/resources/feed_short.atom" # NOTE: Switch out when done testing
PICKLE_DIR = "/tmp/pollenprophet" PICKLE_DIR = "/tmp/pollenprophet"
@ -81,9 +81,12 @@ def load_existing_improvements() -> list[Improvement]:
def improve_originals(originals: list[Original]) -> list[Improvement]: def improve_originals(originals: list[Original]) -> list[Improvement]:
improvements: list[Improvement] = [] improvements: list[Improvement] = []
for orig in originals: for orig in originals:
better_title = improve_with_groq(orig.title) new_title = improve_with_groq(orig.title)
new_summary = rewrite_summary_with_groq(orig, new_title)
improvements.append(Improvement(original=orig, title=better_title, summary="")) improvements.append(
Improvement(original=orig, title=new_title, summary=new_summary)
)
return improvements return improvements
@ -160,6 +163,7 @@ def start() -> None:
from uvicorn import run from uvicorn import run
orig = grab_latest_originals() orig = grab_latest_originals()
improve_originals(orig)
improvements = [ improvements = [
Improvement(original=o, title="hithere", summary="alongsummary") for o in orig Improvement(original=o, title="hithere", summary="alongsummary") for o in orig
] ]