fix: 🐛 await needed for async plans.create (#252)

benshuk · benshuk · web-flow · commit d646e99cdd1d · 2025-01-08T13:20:54.000+01:00
Co-authored-by: benshuk &lt;bens@ai21.com&gt;
diff --git a/ai21/clients/studio/resources/beta/assistant/assistants_plans.py b/ai21/clients/studio/resources/beta/assistant/assistants_plans.py
@@ -78,7 +78,9 @@ async def create(
             **kwargs,
         )
 
-        return self._post(path=f"/assistants/{assistant_id}/{self._module_name}", body=body, response_cls=PlanResponse)
+        return await self._post(
+            path=f"/assistants/{assistant_id}/{self._module_name}", body=body, response_cls=PlanResponse
+        )
 
     async def list(
         self,

Original file line number	Diff line number	Diff line change
`@@ -78,7 +78,9 @@ async def create(`
`78`	`78`	`**kwargs,`
`79`	`79`	`)`
`80`	`80`
`81`		`- return self._post(path=f"/assistants/{assistant_id}/{self._module_name}", body=body, response_cls=PlanResponse)`
	`81`	`+ return await self._post(`
	`82`	`+ path=f"/assistants/{assistant_id}/{self._module_name}", body=body, response_cls=PlanResponse`
	`83`	`+ )`
`82`	`84`
`83`	`85`	`async def list(`
`84`	`86`	`self,`