Apply suggestions from code review

Co-authored-by: andrey-zelenkov <[email protected]>

Author: oseoin

src/f5_ai_gateway_sdk/processor.py

@@ -226,7 +226,7 @@ def __init_subclass__(cls, **kwargs):
                 "The DEPRECATED 'process' method must not be implemented "
                 "alongside 'process_input' or 'process_response'."
             )
-        if is_process_overridden and inspect.iscoroutinefunction(cls.process):
+        if is_process_overridden and inspect.iscoroutinefunction(inspect.unwrap(cls.process)):
             # we don't want to add async capabilities to the deprecated function
             raise TypeError(
                 f"Cannot create concrete class {cls.__name__}. "

tests/test_processor.py

@@ -831,6 +831,47 @@ async def process_response(self):
 
         self.assertIsNotNone(AsyncImplementedProcessor())
 
+    async def test_async_message(self):
+        prompt = TEST_REQ_INPUT.model_dump_json()
+        metadata = {"key": "value"}
+        request = fake_multipart_request(
+            prompt=prompt,
+            metadata=metadata,
+            parameters={"modify": True, "annotate": True},
+        )
+
+        class AsyncInputProcessor(Processor):
+            def __init__(self):
+                super().__init__(
+                    name="async-input-processor",
+                    namespace="fake",
+                    signature=INPUT_ONLY_SIGNATURE,
+                    version="v1",
+                )
+
+            async def process_input(self, prompt, metadata, parameters, request) -> Result:
+                prompt.messages.append(Message(content="Test message"))
+                return Result(modified_prompt=prompt)
+
+        processor = AsyncInputProcessor()
+
+        response = await processor.handle_request(request)
+
+        self.assertStatusCodeEqual(response, HTTP_200_OK)
+
+        content = await self.buffer_response(response)
+        multipart = MultipartDecoderHelper(
+            content=content, content_type=response.headers["Content-Type"]
+        )
+
+        self.assertTrue(
+            multipart.has_prompt(),
+            "prompt should be in the response",
+        )
+
+        multipart_prompt = multipart.prompt
+        assert "Test message" in multipart_prompt.content
+
     def test_async_process_implemented(self):
         expected_message = (
             "Cannot create concrete class AsyncProcessImplementedProcessor. "