Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion src/llama_stack_provider_ragas/constants.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,5 +29,5 @@
# Kubeflow ConfigMap keys and defaults for base image resolution
RAGAS_PROVIDER_IMAGE_CONFIGMAP_NAME = "trustyai-service-operator-config"
RAGAS_PROVIDER_IMAGE_CONFIGMAP_KEY = "ragas-provider-image"
DEFAULT_RAGAS_PROVIDER_IMAGE = "quay.io/trustyai/llama-stack-provider-ragas:latest"
DEFAULT_RAGAS_PROVIDER_IMAGE = "registry.access.redhat.com/ubi9/python-312:latest"
KUBEFLOW_CANDIDATE_NAMESPACES = ["redhat-ods-applications", "opendatahub"]
10 changes: 8 additions & 2 deletions src/llama_stack_provider_ragas/remote/kubeflow/components.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,10 @@ def get_base_image() -> str:
return DEFAULT_RAGAS_PROVIDER_IMAGE


@dsl.component(base_image=get_base_image())
@dsl.component(
base_image=get_base_image(),
packages_to_install=["llama-stack-provider-ragas[remote]"],
)
def retrieve_data_from_llama_stack(
dataset_id: str,
llama_stack_base_url: str,
Expand All @@ -72,7 +75,10 @@ def retrieve_data_from_llama_stack(
df.to_json(output_dataset.path, orient="records", lines=True)


@dsl.component(base_image=get_base_image())
@dsl.component(
base_image=get_base_image(),
packages_to_install=["llama-stack-provider-ragas[remote]"],
)
def run_ragas_evaluation(
model: str,
sampling_params: dict,
Expand Down