mirror of
https://github.com/snailyp/gemini-balance.git
synced 2026-09-05 07:26:39 +08:00
1. modify IMAGE_URL_PATTERN
2. modify import
This commit is contained in:
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user