abstracts and fakes

This commit is contained in:
Adam Wilson
2025-06-12 20:23:31 -06:00
parent 0b6b7b79b9
commit 0cdce03879
3 changed files with 25 additions and 0 deletions

View File

@@ -0,0 +1,7 @@
import abc
class AbstractLanguageModel(abc.ABC):
@abc.abstractmethod
def invoke(self, user_prompt: str) -> str:
raise NotImplementedError

View File

@@ -0,0 +1,7 @@
import abc
class AbstractLanguageModelResponseService(abc.ABC):
@abc.abstractmethod
def invoke(self, user_prompt: str) -> str:
raise NotImplementedError

View File

@@ -0,0 +1,11 @@
from src.text_generation.services.language_models.abstract_language_model_response_service import AbstractLanguageModelResponseService
class FakeLanguageModelResponseService(AbstractLanguageModelResponseService):
def invoke(self, user_prompt: str) -> str:
if not user_prompt:
raise ValueError(f"Parameter 'user_prompt' cannot be empty or None")
return "fake language model response"