chore: remove deprecated LightningLLM class (#63)

Author: lianakoleva

setup.py

@@ -71,9 +71,8 @@ def _prepare_extras(requirements_dir: str = _PATH_REQUIRES, skip_files: tuple =
     install_requires=_load_requirements(),
     extras_require=_prepare_extras(),
     project_urls={
-        "Bug Tracker": "https://github.com/Lightning-AI/LightningLLM/issues",
-        "Documentation": "https://lightning-ai.github.io/LightningLLM/",
-        "Source Code": "https://github.com/Lightning-AI/LightningLLM",
+        "Bug Tracker": "https://github.com/Lightning-AI/litAI/issues",
+        "Source Code": "https://github.com/Lightning-AI/litAI",
     },
     classifiers=[
         "Environment :: Console",

src/litai/__init__.py

@@ -14,8 +14,8 @@
 """Chat with LLMs through Lightning."""
 
 from litai.__about__ import *  # noqa: F401, F403
-from litai.llm import LLM, LightningLLM  # noqa: F401
+from litai.llm import LLM  # noqa: F401
 from litai.llm_config import Models
 from litai.tools import LitTool, tool
 
-__all__ = ["LLM", "Models", "LitTool", "tool", "LightningLLM"]
+__all__ = ["LLM", "Models", "LitTool", "tool"]

src/litai/llm.py

@@ -530,16 +530,3 @@ def __repr__(self) -> str:
             str: A string representation of the instance.
         """
         return f"<LLM model={self.model} fallback_models={self.fallback_models} max_retries={self.max_retries}"
-
-
-class LightningLLM(LLM):
-    """Alias for the LightningLLM class."""
-
-    def __init__(self, *args: Any, **kwargs: Any) -> None:
-        """Initializes the LightningLLM client."""
-        super().__init__(*args, **kwargs)
-        warnings.warn(
-            "The LightningLLM class is deprecated. Use the LLM class instead.",
-            DeprecationWarning,
-            stacklevel=2,
-        )

tests/__init__.py

@@ -1,4 +1,4 @@
-"""LightningLLM tests."""
+"""LitAI tests."""
 # Copyright The Lightning AI team.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");