diff --git a/docs/features/plugin/migration/index.mdx b/docs/features/plugin/migration/index.mdx
index eca9784..f39a591 100644
--- a/docs/features/plugin/migration/index.mdx
+++ b/docs/features/plugin/migration/index.mdx
@@ -180,12 +180,12 @@ class ChatBotService:
 
     async def handle_request(
         self,
-        user_input: dict,
+        body: dict,
         __request__: Request,
         __task__=None,
     ) -> str:
         # Call the unified endpoint for chat completion
-        response = await chat_completion(__request__, user_input)
+        response = await chat_completion(__request__, body)
         return response
 ```