From 28e67cc3fa7da7cc6f0e8ffbe0780f990233be56 Mon Sep 17 00:00:00 2001
From: zhanghaoyu <zhy15935467108@163.com>
Date: Sat, 15 Mar 2025 12:37:56 +0800
Subject: [PATCH] 1. modify IMAGE_URL_PATTERN 2. modify import

---
 app/services/chat/message_converter.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/app/services/chat/message_converter.py b/app/services/chat/message_converter.py
index e29e143..651c6b7 100644
--- a/app/services/chat/message_converter.py
+++ b/app/services/chat/message_converter.py
@@ -3,9 +3,11 @@
 from abc import ABC, abstractmethod
 import re
 from typing import Any, Dict, List, Optional
+import requests
+import base64
 
 SUPPORTED_ROLES = ["user", "model", "system"]
-IMAGE_URL_PATTERN = r'(https://img\.picgo\.net/[^\s)\]\"\']+)'
+IMAGE_URL_PATTERN = r'\[image\]\((.*?)\)'
 
 
 class MessageConverter(ABC):
@@ -39,8 +41,6 @@ def _convert_image_to_base64(url: str) -> str:
     Returns:
         str: base64编码的图片数据
     """
-    import requests
-    import base64
     response = requests.get(url)
     if response.status_code == 200:
         # 将图片内容转换为base64