Update test_async_client.py

mkozakov · web-flow · commit 0fc503219aea · 2026-03-24T23:59:38.000-04:00
diff --git a/tests/test_async_client.py b/tests/test_async_client.py
@@ -79,7 +79,9 @@ async def test_moved_fn(self) -> None:
     @unittest.skipIf(os.getenv("CO_API_URL") is not None, "Doesn't work in staging.")
     async def test_generate(self) -> None:
         response = await self.co.generate(
+            model="command-a-03-2025",
             prompt='Please explain to me how LLMs work',
+            max_tokens=10
         )
         print(response)
 

Original file line number	Diff line number	Diff line change
`@@ -79,7 +79,9 @@ async def test_moved_fn(self) -> None:`
`79`	`79`	`@unittest.skipIf(os.getenv("CO_API_URL") is not None, "Doesn't work in staging.")`
`80`	`80`	`async def test_generate(self) -> None:`
`81`	`81`	`response = await self.co.generate(`
	`82`	`+ model="command-a-03-2025",`
`82`	`83`	`prompt='Please explain to me how LLMs work',`
	`84`	`+ max_tokens=10`
`83`	`85`	`)`
`84`	`86`	`print(response)`
`85`	`87`