Update main.py

main.py

@@ -279,7 +279,7 @@ class Blackbox:
             if messages and isinstance(messages[0]['content'], list):
                 # Prepend prefix to the first text message
                 for content_part in messages[0]['content']:
-                    if content_part.get('type') == 'text' and not content_part['text'].startswith(prefix):
+                    if isinstance(content_part, dict) and content_part.get('type') == 'text' and not content_part['text'].startswith(prefix):
                         logger.debug(f"Adding prefix '{prefix}' to the first text message.")
                         content_part['text'] = f"{prefix} {content_part['text']}"
                         break
@@ -292,7 +292,7 @@ class Blackbox:
             last_message = messages[-1]
             if isinstance(last_message['content'], list):
                 for content_part in last_message['content']:
-                    if content_part.get('type') == 'text':
+                    if isinstance(content_part, dict) and content_part.get('type') == 'text':
                         content_part['role'] = 'user'
             else:
                 last_message['id'] = random_id
@@ -355,7 +355,7 @@ class Blackbox:
                                 logger.error("Image URL not found in the response.")
                                 raise Exception("Image URL not found in the response")
                         else:
-                            async for chunk in response.content.iter_chunks():
+                            async for chunk, info in response.content.iter_chunks():
                                 if chunk:
                                     decoded_chunk = chunk.decode(errors='ignore')
                                     decoded_chunk = re.sub(r'\$@\$v=[^$]+\$@\$', '', decoded_chunk)
@@ -452,10 +452,10 @@ async def chat_completions(
                 # Convert list of content parts to a structured format
                 combined_content = []
                 for part in msg.content:
-                    if isinstance(part, TextContent):
-                        combined_content.append({"type": part.type, "text": part.text})
-                    elif isinstance(part, ImageURLContent):
-                        combined_content.append({"type": part.type, "image_url": part.image_url})
+                    if isinstance(part, dict) and part.get('type') == 'text':
+                        combined_content.append({"type": part['type'], "text": part['text']})
+                    elif isinstance(part, dict) and part.get('type') == 'image_url':
+                        combined_content.append({"type": part['type'], "image_url": part['image_url']})
                 processed_messages.append({"role": msg.role, "content": combined_content})
             else:
                 processed_messages.append({"role": msg.role, "content": msg.content})