feat: optimize agent configuration and memory

channel/dingtalk/dingtalk_message.py

@@ -1,4 +1,5 @@
 import os
+import re
 
 import requests
 from dingtalk_stream import ChatbotMessage
@@ -8,6 +9,7 @@ from channel.chat_message import ChatMessage
 # -*- coding=utf-8 -*-
 from common.log import logger
 from common.tmp_dir import TmpDir
+from config import conf
 
 
 class DingTalkMessage(ChatMessage):
@@ -37,15 +39,67 @@ class DingTalkMessage(ChatMessage):
             self.content = event.extensions['content']['recognition'].strip()
             self.ctype = ContextType.TEXT
         elif (self.message_type == 'picture') or (self.message_type == 'richText'):
-            self.ctype = ContextType.IMAGE
             # 钉钉图片类型或富文本类型消息处理
             image_list = event.get_image_list()
-            if len(image_list) > 0:
+            
+            if self.message_type == 'picture' and len(image_list) > 0:
+                # 单张图片消息：下载到工作空间，用于文件缓存
+                self.ctype = ContextType.IMAGE
                 download_code = image_list[0]
                 download_url = image_download_handler.get_image_download_url(download_code)
-                self.content = download_image_file(download_url, TmpDir().path())
+                
+                # 下载到工作空间 tmp 目录
+                workspace_root = os.path.expanduser(conf().get("agent_workspace", "~/cow"))
+                tmp_dir = os.path.join(workspace_root, "tmp")
+                os.makedirs(tmp_dir, exist_ok=True)
+                
+                image_path = download_image_file(download_url, tmp_dir)
+                if image_path:
+                    self.content = image_path
+                    self.image_path = image_path  # 保存图片路径用于缓存
+                    logger.info(f"[DingTalk] Downloaded single image to {image_path}")
+                else:
+                    self.content = "[图片下载失败]"
+                    self.image_path = None
+            
+            elif self.message_type == 'richText' and len(image_list) > 0:
+                # 富文本消息：下载所有图片并附加到文本中
+                self.ctype = ContextType.TEXT
+                
+                # 下载到工作空间 tmp 目录
+                workspace_root = os.path.expanduser(conf().get("agent_workspace", "~/cow"))
+                tmp_dir = os.path.join(workspace_root, "tmp")
+                os.makedirs(tmp_dir, exist_ok=True)
+                
+                # 提取富文本中的文本内容
+                text_content = ""
+                if self.rich_text_content:
+                    # rich_text_content 是一个 RichTextContent 对象，需要从中提取文本
+                    text_list = event.get_text_list()
+                    if text_list:
+                        text_content = "".join(text_list).strip()
+                
+                # 下载所有图片
+                image_paths = []
+                for download_code in image_list:
+                    download_url = image_download_handler.get_image_download_url(download_code)
+                    image_path = download_image_file(download_url, tmp_dir)
+                    if image_path:
+                        image_paths.append(image_path)
+                
+                # 构建消息内容：文本 + 图片路径
+                content_parts = []
+                if text_content:
+                    content_parts.append(text_content)
+                for img_path in image_paths:
+                    content_parts.append(f"[图片: {img_path}]")
+                
+                self.content = "\n".join(content_parts) if content_parts else "[富文本消息]"
+                logger.info(f"[DingTalk] Received richText with {len(image_paths)} image(s): {self.content}")
             else:
-                logger.debug(f"[Dingtalk] messageType :{self.message_type} , imageList isEmpty")
+                self.ctype = ContextType.IMAGE
+                self.content = "[未找到图片]"
+                logger.debug(f"[DingTalk] messageType: {self.message_type}, imageList isEmpty")
 
         if self.is_group:
             self.from_user_id = event.conversation_id
@@ -59,27 +113,95 @@ class DingTalkMessage(ChatMessage):
 
 
 def download_image_file(image_url, temp_dir):
-    headers = {
-        'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36'
-    }
-    # 设置代理
-    # self.proxies
-    # , proxies=self.proxies
-    response = requests.get(image_url, headers=headers, stream=True, timeout=60 * 5)
-    if response.status_code == 200:
-
-        # 生成文件名
-        file_name = image_url.split("/")[-1].split("?")[0]
-
-        # 检查临时目录是否存在，如果不存在则创建
-        if not os.path.exists(temp_dir):
-            os.makedirs(temp_dir)
-
-        # 将文件保存到临时目录
-        file_path = os.path.join(temp_dir, file_name)
-        with open(file_path, 'wb') as file:
-            file.write(response.content)
-        return file_path
+    """
+    下载图片文件
+    支持两种方式：
+    1. 普通 HTTP(S) URL
+    2. 钉钉 downloadCode: dingtalk://download/{download_code}
+    """
+    # 检查临时目录是否存在，如果不存在则创建
+    if not os.path.exists(temp_dir):
+        os.makedirs(temp_dir)
+    
+    # 处理钉钉 downloadCode
+    if image_url.startswith("dingtalk://download/"):
+        download_code = image_url.replace("dingtalk://download/", "")
+        logger.info(f"[DingTalk] Downloading image with downloadCode: {download_code[:20]}...")
+        
+        # 需要从外部传入 access_token，这里先用一个临时方案
+        # 从 config 获取 dingtalk_client_id 和 dingtalk_client_secret
+        from config import conf
+        client_id = conf().get("dingtalk_client_id")
+        client_secret = conf().get("dingtalk_client_secret")
+        
+        if not client_id or not client_secret:
+            logger.error("[DingTalk] Missing dingtalk_client_id or dingtalk_client_secret")
+            return None
+        
+        # 获取 access_token
+        token_url = "https://oapi.dingtalk.com/gettoken"
+        token_params = {
+            "appkey": client_id,
+            "appsecret": client_secret
+        }
+        
+        try:
+            token_response = requests.get(token_url, params=token_params, timeout=10)
+            token_data = token_response.json()
+            
+            if token_data.get("errcode") == 0:
+                access_token = token_data.get("access_token")
+                
+                # 下载图片
+                download_url = f"https://oapi.dingtalk.com/robot/messageFiles/download"
+                download_params = {
+                    "access_token": access_token,
+                    "downloadCode": download_code
+                }
+                
+                response = requests.get(download_url, params=download_params, stream=True, timeout=60)
+                if response.status_code == 200:
+                    # 生成文件名（使用 download_code 的 hash，避免特殊字符）
+                    import hashlib
+                    file_hash = hashlib.md5(download_code.encode()).hexdigest()[:16]
+                    file_name = f"{file_hash}.png"
+                    file_path = os.path.join(temp_dir, file_name)
+                    
+                    with open(file_path, 'wb') as file:
+                        file.write(response.content)
+                    
+                    logger.info(f"[DingTalk] Image downloaded successfully: {file_path}")
+                    return file_path
+                else:
+                    logger.error(f"[DingTalk] Failed to download image: {response.status_code}")
+                    return None
+            else:
+                logger.error(f"[DingTalk] Failed to get access token: {token_data}")
+                return None
+        except Exception as e:
+            logger.error(f"[DingTalk] Exception downloading image: {e}")
+            return None
+    
+    # 普通 HTTP(S) URL
     else:
-        logger.info(f"[Dingtalk] Failed to download image file, {response.content}")
-        return None
+        headers = {
+            'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36'
+        }
+        
+        try:
+            response = requests.get(image_url, headers=headers, stream=True, timeout=60 * 5)
+            if response.status_code == 200:
+                # 生成文件名
+                file_name = image_url.split("/")[-1].split("?")[0]
+                
+                # 将文件保存到临时目录
+                file_path = os.path.join(temp_dir, file_name)
+                with open(file_path, 'wb') as file:
+                    file.write(response.content)
+                return file_path
+            else:
+                logger.info(f"[Dingtalk] Failed to download image file, {response.content}")
+                return None
+        except Exception as e:
+            logger.error(f"[Dingtalk] Exception downloading image: {e}")
+            return None