fix quickstarts

Author: xzrderek

tests/github_actions/quickstart.py

@@ -14,14 +14,18 @@
 import pytest
 
 from eval_protocol.data_loader.dynamic_data_loader import DynamicDataLoader
-from eval_protocol.models import EvaluationRow, Message
+from eval_protocol.models import EvaluationRow, InputMetadata
 from eval_protocol.pytest import evaluation_test
 from eval_protocol.pytest.github_action_rollout_processor import GithubActionRolloutProcessor
 
 
 def rows() -> List[EvaluationRow]:
-    row = EvaluationRow(messages=[Message(role="user", content="What is the capital of France?")])
-    return [row, row, row]
+    return [
+        EvaluationRow(input_metadata=InputMetadata(row_id=str(i)))
+        for i in range(
+            3
+        )  # In this example we use index to associate rows. Dataset is assumed to be accessible to the worker.
+    ]
 
 
 @pytest.mark.skipif(os.environ.get("CI") == "true", reason="Only run this test locally (skipped in CI)")

tests/remote_server/quickstart.py

@@ -47,7 +47,7 @@ async def test_remote_rollout_and_fetch_fireworks(row: EvaluationRow) -> Evaluat
     End-to-end test with Vercel production server:
     - Uses deployed Vercel serverless function (no manual startup needed)
     - trigger remote rollout via RemoteRolloutProcessor (calls init/status)
-    - fetch traces from Langfuse via Fireworks tracing proxy (uses default FireworksTracingAdapter)
+    - fetch traces from Fireworks tracing (uses default FireworksTracingAdapter)
     - FAIL if no traces found or rollout_id missing
     """
     assert row.messages[0].content == "What is the capital of France?", "Row should have correct message content"