lcipolina commited on
Commit
432399d
·
verified ·
1 Parent(s): cdbde5a

Made importing LiteLLM optional

Browse files

The 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
- import litellm
 
 
 
 
 
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