phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-20 06:08:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

more scoring function for rag

Browse source
This commit is contained in:
Xi Yan 2024-12-19 16:53:39 -08:00
parent b94ab8d013
commit 9aa4a405ca
8 changed files with 132 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_stack/providers/tests/scoring/test_scoring.py
View file

@ -60,7 +60,7 @@ class TestScoring:
f"{provider_id} provider does not support scoring without params"
)
await register_dataset(datasets_impl)
await register_dataset(datasets_impl, for_rag=True)
response = await datasets_impl.list_datasets()
assert len(response) == 1
@ -112,7 +112,7 @@ class TestScoring:
scoring_stack[Api.datasets],
scoring_stack[Api.models],
)
await register_dataset(datasets_impl)
await register_dataset(datasets_impl, for_rag=True)
response = await datasets_impl.list_datasets()
assert len(response) == 1
@ -173,7 +173,7 @@ class TestScoring:
scoring_stack[Api.datasets],
scoring_stack[Api.models],
)
await register_dataset(datasets_impl)
await register_dataset(datasets_impl, for_rag=True)
rows = await datasetio_impl.get_rows_paginated(
dataset_id="test_dataset",
rows_in_page=3,