1. modify IMAGE_URL_PATTERN

2. modify import
This commit is contained in:
zhanghaoyu
2025-03-15 12:37:56 +08:00
parent d99a0bde93
commit 28e67cc3fa
+3 -3
View File
@@ -3,9 +3,11 @@
from abc import ABC, abstractmethod from abc import ABC, abstractmethod
import re import re
from typing import Any, Dict, List, Optional from typing import Any, Dict, List, Optional
import requests
import base64
SUPPORTED_ROLES = ["user", "model", "system"] SUPPORTED_ROLES = ["user", "model", "system"]
IMAGE_URL_PATTERN = r'(https://img\.picgo\.net/[^\s)\]\"\']+)' IMAGE_URL_PATTERN = r'\[image\]\((.*?)\)'
class MessageConverter(ABC): class MessageConverter(ABC):
@@ -39,8 +41,6 @@ def _convert_image_to_base64(url: str) -> str:
Returns: Returns:
str: base64编码的图片数据 str: base64编码的图片数据
""" """
import requests
import base64
response = requests.get(url) response = requests.get(url)
if response.status_code == 200: if response.status_code == 200:
# 将图片内容转换为base64 # 将图片内容转换为base64