Spaces:
Sleeping
Sleeping
| from abc import ABC , abstractmethod | |
| class LLMInterface(ABC): | |
| # to force used this method | |
| def set_generation_model(self, model_id: str): | |
| pass | |
| def set_embedding_model(self, model_id: str, embeddig_size:int): | |
| pass | |
| def generate_text(self, prompt: str, max_output_token: int=None, | |
| chat_history: list=[],temperature: float = None): | |
| pass | |
| def embed_text(self, text: str, document_type: str= None ): | |
| pass | |
| def embed_text(self, texts: list, document_type: str = None): | |
| pass | |
| def construct_prompt(self, prompt: str, role: str): # call to handle prompt before generate text method usedit | |
| pass | |