mirror of
https://github.com/lightbroker/llmsecops-research.git
synced 2026-03-18 08:16:19 +00:00
abstracts and fakes
This commit is contained in:
@@ -0,0 +1,7 @@
|
||||
import abc
|
||||
|
||||
|
||||
class AbstractLanguageModel(abc.ABC):
|
||||
@abc.abstractmethod
|
||||
def invoke(self, user_prompt: str) -> str:
|
||||
raise NotImplementedError
|
||||
@@ -0,0 +1,7 @@
|
||||
import abc
|
||||
|
||||
|
||||
class AbstractLanguageModelResponseService(abc.ABC):
|
||||
@abc.abstractmethod
|
||||
def invoke(self, user_prompt: str) -> str:
|
||||
raise NotImplementedError
|
||||
@@ -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"
|
||||
Reference in New Issue
Block a user