| 3671 | |
| 3672 | @staticmethod |
| 3673 | def get_image_urls(messages: List[llama_types.ChatCompletionRequestMessage]): |
| 3674 | image_urls: List[str] = [] |
| 3675 | for message in messages: |
| 3676 | if message["role"] == "user": |
| 3677 | if message["content"] is None: |
| 3678 | continue |
| 3679 | for content in message["content"]: |
| 3680 | if isinstance(content, dict) and "type" in content: |
| 3681 | if content["type"] == "image_url": |
| 3682 | if ( |
| 3683 | isinstance(content["image_url"], dict) |
| 3684 | and "url" in content["image_url"] |
| 3685 | ): |
| 3686 | image_urls.append(content["image_url"]["url"]) |
| 3687 | else: |
| 3688 | image_urls.append(content["image_url"]) |
| 3689 | return image_urls |
| 3690 | |
| 3691 | @classmethod |
| 3692 | def from_pretrained( |