feat: Add single rewrite method to LLM models

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

View file

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