fix batch response item issues

Author: Ndiritu

src/msgraph_core/requests/batch_response_item.py

@@ -1,15 +1,11 @@
-from typing import Optional, Dict, Any
+from typing import Optional, Dict, Any, Callable
 from io import BytesIO
 
 from kiota_abstractions.serialization import Parsable, ParsableFactory
 from kiota_abstractions.serialization import ParseNode
 from kiota_abstractions.serialization import SerializationWriter
 
 
-class StreamInterface(BytesIO):
-    pass
-
-
 class BatchResponseItem(Parsable):
 
     def __init__(self) -> None:
@@ -24,7 +20,7 @@ def __init__(self) -> None:
 
     @property
     def id(self) -> Optional[str]:
-        """ 
+        """
         Get the ID of the response
         :return: The ID of the response
         :rtype: Optional[str]
@@ -69,7 +65,7 @@ def status(self) -> Optional[int]:
 
     @status.setter
     def status(self, status_code: Optional[int]) -> None:
-        """ 
+        """
         Set the status code of the response
         :param status_code: The status code of the response
         :type status_code: Optional[int]
@@ -104,7 +100,7 @@ def body(self) -> Optional[BytesIO]:
         return self._body
 
     @body.setter
-    def body(self, body: Optional[StreamInterface]) -> None:
+    def body(self, body: Optional[BytesIO]) -> None:
         """
         Set the body of the response
         :param body: The body of the response
@@ -138,15 +134,15 @@ def create_from_discriminator_value(
             raise TypeError("parse_node cannot be null")
         return BatchResponseItem()
 
-    def get_field_deserializers(self) -> Dict[str, Any]:
+    def get_field_deserializers(self) -> Dict[str, Callable[[ParseNode], None]]:
         """
         Gets the deserialization information for this object.
 
         """
         return {
             "id": lambda x: setattr(self, "id", x.get_str_value()),
             "status": lambda x: setattr(self, "status", x.get_int_value()),
-            "headers": lambda x: setattr(self, "headers", x.try_get_anything(x._json_node)),
+            "headers": lambda x: setattr(self, "headers", x.try_get_anything(x._json_node)), # type: ignore # need interface to return a dictionary
             "body": lambda x: setattr(self, "body", x.get_bytes_value()),
         }
 
@@ -157,5 +153,8 @@ def serialize(self, writer: SerializationWriter) -> None:
         writer.write_str_value('id', self._id)
         writer.write_str_value('atomicity_group', self._atomicity_group)
         writer.write_int_value('status', self._status)
-        writer.write_collection_of_primitive_values('headers', self._headers)
-        writer.write_bytes_value('body', self._body)
+        writer.write_collection_of_primitive_values('headers', self._headers) # type: ignore # need method to serialize dicts
+        if self._body:
+            writer.write_bytes_value('body', self._body.getvalue())
+        else:
+            writer.write_bytes_value('body', None)