fix: check litellm version

Author: zakahan

veadk/agent.py

@@ -50,6 +50,7 @@
 from veadk.tracing.base_tracer import BaseTracer
 from veadk.utils.logger import get_logger
 from veadk.utils.patches import patch_asyncio, patch_tracer
+from veadk.utils.misc import check_litellm_version
 from veadk.version import VERSION
 
 patch_tracer()
@@ -159,28 +160,7 @@ def model_post_init(self, __context: Any) -> None:
         if not self.model:
             if self.enable_responses:
                 min_version = "1.79.3"
-                from packaging.version import InvalidVersion
-                from packaging.version import parse as parse_version
-                import pkg_resources
-
-                try:
-                    installed = parse_version(
-                        pkg_resources.get_distribution("litellm").version
-                    )
-                except pkg_resources.DistributionNotFound:
-                    raise ImportError(
-                        "litellm installation not detected, please install it first: pip install litellm>=1.79.3"
-                    ) from None
-                except InvalidVersion as e:
-                    raise ValueError(
-                        f"Invalid format of litellm version number:{e}"
-                    ) from None
-                required = parse_version(min_version)
-                if installed < required:
-                    raise ValueError(
-                        "You have used `enable_responses=True`. If you want to use the `responses_api`, please install the relevant support:"
-                        "\npip install veadk-python[responses]"
-                    )
+                check_litellm_version(min_version)
 
                 from veadk.models.ark_llm import ArkLlm
 

veadk/utils/misc.py

@@ -182,3 +182,35 @@ def get_agent_dir():
         full_path = os.getcwd()
 
     return full_path
+
+
+def check_litellm_version(min_version: str):
+    """
+    Check if the installed litellm version meets the minimum requirement.
+
+    Args:
+        min_version (str): The minimum required version of litellm.
+    """
+    try:
+        from packaging.version import InvalidVersion
+        from packaging.version import parse as parse_version
+        import pkg_resources
+
+        try:
+            installed = parse_version(pkg_resources.get_distribution("litellm").version)
+        except pkg_resources.DistributionNotFound:
+            raise ImportError(
+                "litellm installation not detected, please install it first: pip install litellm>=1.79.3"
+            ) from None
+        except InvalidVersion as e:
+            raise ValueError(f"Invalid format of litellm version number:{e}") from None
+        required = parse_version(min_version)
+        if installed < required:
+            raise ValueError(
+                "You have used `enable_responses=True`. If you want to use the `responses_api`, please install the relevant support:"
+                "\npip install veadk-python[responses]"
+            )
+    except ImportError:
+        raise ImportError(
+            "packaging or pkg_resources not found. Please install them: pip install packaging setuptools"
+        )