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

feat: Add single rewrite method to LLM models

Browse source
This commit is contained in:
Marty Oehme 2025-06-10 21:27:51 +02:00
parent 333d825cb7
commit 63131b7f11
Signed by: Marty
GPG key ID: 4E535BC19C61886E
2 changed files with 13 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
prophet/domain/llm.py
View file

@ -1,10 +1,11 @@
from typing import Protocol from typing import Protocol
from prophet.domain.improvement import Improvement
from prophet.domain.original import Original from prophet.domain.original import Original
class LLMClient(Protocol): class LLMClient(Protocol):
def get_alternative_title_suggestions(self, original_content: str) -> str: def rewrite(self, original: Original) -> Improvement:
raise NotImplementedError raise NotImplementedError
def rewrite_title( def rewrite_title(
@ -16,3 +17,6 @@ class LLMClient(Protocol):
self, original: Original, improved_title: str | None = None self, original: Original, improved_title: str | None = None
) -> str: ) -> str:
raise NotImplementedError raise NotImplementedError
def get_alternative_title_suggestions(self, original_content: str) -> str:
raise NotImplementedError

8
prophet/infra/llm_groq.py
View file

@ -20,6 +20,14 @@ class GroqClient(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)
@override
def rewrite(self, original: Original) -> Improvement:
suggestions = self.get_alternative_title_suggestions(original.title)
new_title = self.rewrite_title(original.title, suggestions)
new_summary = self.rewrite_summary(original, new_title)
return Improvement(original=original, title=new_title, summary=new_summary)
@override @override
def get_alternative_title_suggestions( def get_alternative_title_suggestions(
self, original_content: str, custom_prompt: str | None = None self, original_content: str, custom_prompt: str | None = None