diff --git a/src/routers/generate_3D.py b/src/routers/generate_3D.py
index d913532..b5a83a4 100755
--- a/src/routers/generate_3D.py
+++ b/src/routers/generate_3D.py
@@ -255,8 +255,9 @@ async def triop_api_img_to_3D(request_data: Tripo3dApiModel, background_tasks: B
     else:
         task_resp = await create_multi_task(input_data=request_data)
 
-    if task_resp.get("state") == "success":
-        api_task_id = task_resp.get("task_id")
+    if task_resp.get("code") == 0:
+        api_task_id = task_resp.get("data").get("task_id")
+        logger.info(f"{request_data, request_data.task_id, api_task_id, request_data.callback_url}")
         background_tasks.add_task(get_task_result_async, request_data, request_data.task_id, api_task_id, request_data.callback_url)
         result = {
             "state": "success",
@@ -272,6 +273,7 @@ async def triop_api_img_to_3D(request_data: Tripo3dApiModel, background_tasks: B
             "message": task_resp.get("message"),
             "error": task_resp.get("message")
         }
+        logger.info(data)
         return ResponseModel(data=data, code=500, msg=task_resp.get("message", ""))
 
 # @router.post("/img_to_3D")