fix: types

Author: JohnRichard4096

src/amrita_core/chatmanager.py

@@ -401,7 +401,7 @@ def __init__(
         preset: ModelPreset | None = None,
         auto_create_session: bool = False,
         hook_args: tuple[Any, ...] = (),
-        hook_kwargs: dict[str, Any] = {},
+        hook_kwargs: dict[str, Any] | None = None,
         queue_size: int = 25,
         overflow_queue_size: int = 45,
     ) -> None:
@@ -444,7 +444,7 @@ def __init__(
         )
         # Hook args
         self._hook_args = hook_args
-        self._hook_kwargs = hook_kwargs
+        self._hook_kwargs = hook_kwargs or {}
 
         # Initialize async queue for streaming responses
         self._response_queue = asyncio.Queue(queue_size)

src/amrita_core/protocol.py

@@ -106,9 +106,7 @@ def __init__(self, image: str | BytesIO | bytes):
         super().__init__("image")
         self.image: str | BytesIO | bytes = image
 
-    async def get_image(
-        self, headers: dict[str, None] | None = None
-    ) -> BytesIO | bytes:
+    async def get_image(self, headers: dict[str, Any] | None = None) -> BytesIO | bytes:
         if isinstance(self.image, str):
             self.image = await self.curl_image(headers)
         return self.image
@@ -123,7 +121,8 @@ async def curl_image(self, extra_headers: dict | None = None) -> bytes:
                 async with session.get(self.image) as response:
                     if response.status != 200:
                         raise ValueError(f"Failed to download image from {self.image}")
-                    obj = await response.read()
+                    bt = await response.read()
+                    obj = base64.b64encode(bt)
                     return obj
         raise ValueError("Image must be a URL to use this method")
 
@@ -135,7 +134,7 @@ def get_content(self) -> str:
         image_type = get_image_format(self.image)
         if not image_type:
             return "[Unsupported image format]"
-        return f"![](data:image/{image_type};base64,{base64.b64encode(self.image).decode()})"
+        return f"![](data:image/{image_type};base64,{self.image.decode('utf-8')})"
 
     async def save_to(self, path: Path, headers: dict | None = None):
         async with aiofiles.open(path, "wb") as f: