diff --git a/base_telegram_inference_bot.py b/base_telegram_inference_bot.py index 3e4709d..5979c0b 100644 --- a/base_telegram_inference_bot.py +++ b/base_telegram_inference_bot.py @@ -63,7 +63,24 @@ class BaseTelegramInferenceBot(ABC): for function in tool.get_functions(): if function["function"]["name"] == function_name: return tool.execute(function_name, **function_args) - + + @abstractmethod + def get_system_prompt_description(self) -> str: + """Returns a description of the system prompt being used.""" + pass + + @abstractmethod + def get_llm_description(self) -> str: + """Returns a description of the LLM being used.""" + pass + + async def status(self) -> str: # Changed from abstract to concrete + """Provides a status message including prompt and LLM information.""" + prompt_desc = self.get_system_prompt_description() + llm_desc = self.get_llm_description() + # Consider potential async calls if get_... methods were async + # For now, assuming they are synchronous as per design + return f"{prompt_desc}\n{llm_desc}" @abstractmethod async def start(self): @@ -73,10 +90,6 @@ class BaseTelegramInferenceBot(ABC): async def clear(self, user_id): pass - @abstractmethod - async def status(self): - pass - @abstractmethod async def abort_processing(self, user_id): - pass \ No newline at end of file + pass