Fix break

cohere-ai · Nov 28, 2024 · fe9d0fc · fe9d0fc
1 parent 64209fd
commit fe9d0fc
Showing 1 changed file with 4 additions and 10 deletions.
diff --git a/src/test/cloud-tests/rerank-2.test.ts b/src/test/cloud-tests/rerank-2.test.ts
@@ -5,16 +5,10 @@ skipIfTestTypeIsnt("rerank-2")("rerank-2", () => {
     test.concurrent("works", async () => {
         const rerank = await cohere.v2.rerank({
             documents: [
-                { text: "Carson City is the capital city of the American state of Nevada." },
-                {
-                    text: "The Commonwealth of the Northern Mariana Islands is a group of islands in the Pacific Ocean. Its capital is Saipan.",
-                },
-                {
-                    text: "Washington, D.C. (also known as simply Washington or D.C., and officially as the District of Columbia) is the capital of the United States. It is a federal district.",
-                },
-                {
-                    text: "Capital punishment (the death penalty) has existed in the United States since beforethe United States was a country. As of 2017, capital punishment is legal in 30 of the 50 states.",
-                },
+                "Carson City is the capital city of the American state of Nevada.",
+                "The Commonwealth of the Northern Mariana Islands is a group of islands in the Pacific Ocean. Its capital is Saipan.",
+                "Washington, D.C. (also known as simply Washington or D.C., and officially as the District of Columbia) is the capital of the United States. It is a federal district.",
+                "Capital punishment (the death penalty) has existed in the United States since beforethe United States was a country. As of 2017, capital punishment is legal in 30 of the 50 states."
             ],
             query: "What is the capital of the United States?",
             topN: 3,