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

Rename llm methods

Browse source
This commit is contained in:
Marty Oehme 2025-06-06 11:13:25 +02:00
parent c537b1e750
commit 003eebfebf
Signed by: Marty
GPG key ID: 4E535BC19C61886E
2 changed files with 7 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

9
prophet/app.py
View file

@ -84,8 +84,8 @@ def keep_only_new_originals(
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:
new_title = llm.rewrite_title_with_groq(orig.title) new_title = llm.rewrite_title(orig.title)
new_summary = llm.rewrite_summary_with_groq(orig, new_title) new_summary = llm.rewrite_summary(orig, new_title)
improvements.append( improvements.append(
Improvement(original=orig, title=new_title, summary=new_summary) Improvement(original=orig, title=new_title, summary=new_summary)
@ -111,7 +111,7 @@ app.add_middleware(
@app.get("/improve-title") @app.get("/improve-title")
def improve_headline(content: str): def improve_headline(content: str):
return llm.rewrite_title_with_groq(content) return llm.rewrite_title(content)
@app.get("/improve-summary") @app.get("/improve-summary")
@ -119,7 +119,7 @@ def improve_summary(original_title: str, new_title: str, original_summary: str):
o = Original( o = Original(
title=original_title, summary=original_summary, link="", date=datetime.now() title=original_title, summary=original_summary, link="", date=datetime.now()
) )
return llm.rewrite_summary_with_groq(o, new_title) return llm.rewrite_summary(o, new_title)
@app.on_event("startup") @app.on_event("startup")
@ -137,6 +137,7 @@ async def fetch_update():
return json.dumps(improved) return json.dumps(improved)
## HTML (& hyperdata) responses
@app.get("/improvements", response_class=HTMLResponse) @app.get("/improvements", response_class=HTMLResponse)
def list_improvements(): def list_improvements():
improved = load_existing_improvements() improved = load_existing_improvements()

4
prophet/llm.py
View file

@ -14,7 +14,7 @@ class LLMClient:
self.config_ai = config_ai if config_ai else AiConfig.from_env() self.config_ai = config_ai if config_ai else AiConfig.from_env()
self.client = client if client else Groq(api_key=self.config_ai.API_KEY) self.client = client if client else Groq(api_key=self.config_ai.API_KEY)
def rewrite_title_with_groq(self, original_content: str) -> str: def rewrite_title(self, original_content: str) -> str:
suggestions = self.client.chat.completions.create( suggestions = self.client.chat.completions.create(
messages=[ messages=[
{ {
@ -51,7 +51,7 @@ class LLMClient:
raise ValueError raise ValueError
return winner_str.strip(" \"'") return winner_str.strip(" \"'")
def rewrite_summary_with_groq(self, orig: Original, improved_title: str) -> str: def rewrite_summary(self, orig: Original, improved_title: str) -> str:
summary = self.client.chat.completions.create( summary = self.client.chat.completions.create(
messages=[ messages=[
{ {