-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathprovider.py
More file actions
37 lines (25 loc) · 997 Bytes
/
provider.py
File metadata and controls
37 lines (25 loc) · 997 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
import logging
from .inline.provider import get_provider_spec as get_inline_provider_spec
logger = logging.getLogger(__name__)
def _has_remote_dependencies() -> bool:
"""Check if remote dependencies are available."""
try:
import kfp # noqa: F401
import kubernetes # noqa: F401
import s3fs # noqa: F401
from kfp import kubernetes as kfp_kubernetes # noqa: F401
return True
except ImportError:
return False
def get_provider_spec():
providers = [get_inline_provider_spec()]
if _has_remote_dependencies():
from .remote.provider import get_provider_spec as get_remote_provider_spec
providers.append(get_remote_provider_spec())
else:
logger.info(
"Remote provider dependencies not found, returning inline provider only. "
"Enable remote evaluation with 'pip install llama-stack-provider-ragas[remote]'."
)
return providers
__all__ = ["get_provider_spec"]