From 364ec10fe4647d53c5e1a38c5b45cdae298e77dc Mon Sep 17 00:00:00 2001 From: Maarten van Dantzig Date: Mon, 3 Feb 2025 20:41:36 +0100 Subject: [PATCH] Fix LLMManager doc-string --- timesketch/lib/llms/manager.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/timesketch/lib/llms/manager.py b/timesketch/lib/llms/manager.py index 0ceb17878f..5412abcec6 100644 --- 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: