Spaces:
Sleeping
Sleeping
Made importing LiteLLM optional
Browse filesThe app doesn't need litellm
src/game_reasoning_arena/backends/litellm_backend.py
CHANGED
@@ -3,7 +3,12 @@ LiteLLM backend for API-based inference.
|
|
3 |
"""
|
4 |
|
5 |
import os
|
6 |
-
|
|
|
|
|
|
|
|
|
|
|
7 |
from typing import Any, Optional, Dict
|
8 |
from .base_backend import BaseLLMBackend
|
9 |
|
|
|
3 |
"""
|
4 |
|
5 |
import os
|
6 |
+
try:
|
7 |
+
import litellm # type: ignore
|
8 |
+
except Exception as e:
|
9 |
+
raise ImportError(
|
10 |
+
"LiteLLM backend requested but 'litellm' package is not installed."
|
11 |
+
) from e
|
12 |
from typing import Any, Optional, Dict
|
13 |
from .base_backend import BaseLLMBackend
|
14 |
|