feat(backend): massage collate test to work with mock rerank

cohere-ai · Jan 29, 2025 · c7d7f9c · c7d7f9c
1 parent 0cedfc1
commit c7d7f9c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/backend/tests/unit/tools/test_collate.py b/src/backend/tests/unit/tools/test_collate.py
@@ -49,14 +49,14 @@ async def test_rerank() -> None:
                 "parameters": {"query": "highest mountain"},
                 "name": "retriever",
             },
-            "outputs": [outputs[0]],
+            "outputs": [outputs[0]["text"]],
         },
         {
             "call": {
                 "parameters": {"query": "four components of blood"},
                 "name": "retriever",
             },
-            "outputs": [outputs[1]],
+            "outputs": [outputs[1]["text"]],
         },
         {
             "call": {