From bf9683cfd0aded30b06711e17e9290318d692285 Mon Sep 17 00:00:00 2001 From: Kenneth Reitz Date: Thu, 31 Oct 2024 15:38:58 -0400 Subject: [PATCH] Refactor code to use syntax sugar for Plugin class --- simplemind/__init__.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/simplemind/__init__.py b/simplemind/__init__.py index 90313de..96c1138 100644 --- a/simplemind/__init__.py +++ b/simplemind/__init__.py @@ -113,6 +113,9 @@ def generate_text( return provider.generate_text(prompt=prompt, llm_model=llm_model, **kwargs) +# Syntax sugar. +Plugin = BasePlugin + __all__ = [ "create_conversation", "find_provider", @@ -121,4 +124,5 @@ __all__ = [ "settings", "BasePlugin", "Session", + "Plugin", ]