refactoring LLM adapters and service layer

This commit is contained in:
Adam Wilson
2025-06-12 20:25:18 -06:00
parent 0cdce03879
commit 3aca7df000
5 changed files with 43 additions and 67 deletions
@@ -1,10 +1,14 @@
import abc
from src.text_generation.services.language_models.abstract_language_model_response_service import AbstractLanguageModelResponseService
from src.text_generation.adapters.llm.language_model import LanguageModel
class AbstractLanguageModelResponseService(abc.ABC):
@abc.abstractmethod
def invoke(self, user_input: str) -> str:
raise NotImplementedError
class LanguageModelResponseService(AbstractLanguageModelResponseService):
def __call__(self, *args, **kwds):
pass
class TextGenerationResponseService(AbstractLanguageModelResponseService):
def invoke(self, user_prompt: str) -> str:
if not user_prompt:
raise ValueError(f"Parameter 'user_prompt' cannot be empty or None")
llm = LanguageModel()
response = llm.invoke(user_prompt=user_prompt)
return response