update docstrings

deepsense-ai · Dec 5, 2024 · fc68a6d · fc68a6d
1 parent 7564110
commit fc68a6d
Show file tree

Hide file tree

Showing 3 changed files with 0 additions and 9 deletions.
diff --git a/packages/ragbits-core/src/ragbits/core/embeddings/litellm.py b/packages/ragbits-core/src/ragbits/core/embeddings/litellm.py
@@ -35,9 +35,6 @@ def __init__(
                 for more information, follow the instructions for your specific vendor in the\
                 [LiteLLM documentation](https://docs.litellm.ai/docs/embedding/supported_embedding).
             api_version: The API version for the call.
-
-        Raises:
-            ImportError: If the 'litellm' extra requirements are not installed.
         """
         super().__init__()
         self.model = model

diff --git a/packages/ragbits-core/src/ragbits/core/llms/clients/litellm.py b/packages/ragbits-core/src/ragbits/core/llms/clients/litellm.py
@@ -57,9 +57,6 @@ def __init__(
             api_key: API key used to authenticate with the LLM API.
             api_version: API version of the LLM API.
             use_structured_output: Whether to request a structured output from the model. Default is False.
-
-        Raises:
-            ImportError: If the 'litellm' extra requirements are not installed.
         """
         super().__init__(model_name)
         self.base_url = base_url

diff --git a/packages/ragbits-core/src/ragbits/core/llms/litellm.py b/packages/ragbits-core/src/ragbits/core/llms/litellm.py
@@ -42,9 +42,6 @@ def __init__(
             use_structured_output: Whether to request a
                 [structured output](https://docs.litellm.ai/docs/completion/json_mode#pass-in-json_schema)
                 from the model. Default is False. Can only be combined with models that support structured output.
-
-        Raises:
-            ImportError: If the 'litellm' extra requirements are not installed.
         """
         super().__init__(model_name, default_options)
         self.base_url = base_url