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

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.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(
messages=[
{
@ -51,7 +51,7 @@ class LLMClient:
raise ValueError
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(
messages=[
{