Fix LLMManager doc-string

google · Feb 3, 2025 · 364ec10 · 364ec10
1 parent e2abfdb
commit 364ec10
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/timesketch/lib/llms/manager.py b/timesketch/lib/llms/manager.py
@@ -18,6 +18,8 @@
 
 
 class LLMManager:
+    """The manager for LLM providers."""
+
     _class_registry = {}
 
     @classmethod
@@ -61,10 +63,12 @@ def create_provider(cls, feature_name: str = None, **kwargs) -> LLMProvider:
         """
         Create an instance of the provider for the given feature.
 
-        If a configuration exists for the feature in current_app.config["LLM_PROVIDER_CONFIGS"],
-        use it; otherwise, fall back to the configuration under the "default" key.
+        If a configuration exists for the feature in
+        current_app.config["LLM_PROVIDER_CONFIGS"], use it; otherwise,
+        fall back to the configuration under the "default" key.
 
-        The configuration is expected to be a dict with exactly one key: the provider name.
+        The configuration is expected to be a dict with exactly one key:
+        the provider name.
         """
         llm_configs = current_app.config.get("LLM_PROVIDER_CONFIGS", {})
         if feature_name and feature_name in llm_configs: