J1147
/
ai-on-wechat
关注 1
点赞 0
派生 0
代码 工单 0 合并请求 0 版本发布 0 百科 动态
您最多选择25个主题 主题必须以字母或数字开头，可以包含连字符 (-)，并且长度不得超过35个字符
1202 提交
2 分支
152MB
目录树: 36981bf516
分支列表 标签列表
${ item.name }
创建分支 ${ searchTerm }
从 '36981bf516'
${ noResults }
ai-on-wechat/plugins/kimi4upload/kimi4upload.py

573 行
26KB
原始文件 Blame 文件历史 

  1. import requests

  2. import json

  3. import plugins

  4. from bridge.reply import Reply, ReplyType

  5. from bridge.context import ContextType

  6. from channel.chat_message import ChatMessage

  7. from plugins import *

  8. from common.log import logger

  9. from common.expired_dict import ExpiredDict

  10. import os

  11. import base64

  12. from pathlib import Path

  13. from PIL import Image

  14. import oss2

  15. 

  16. 

  17. EXTENSION_TO_TYPE = {

  18.     'pdf': 'pdf',

  19.     'doc': 'docx', 'docx': 'docx',

  20.     'md': 'md',

  21.     'txt': 'txt',

  22.     'xls': 'excel', 'xlsx': 'excel',

  23.     'csv': 'csv',

  24.     'html': 'html', 'htm': 'html',

  25.     'ppt': 'ppt', 'pptx': 'ppt'

  26. }

  27. 

  28. @plugins.register(

  29.     name="kimi4upload",

  30.     desire_priority=-1,

  31.     desc="A plugin for upload",

  32.     version="0.0.01",

  33.     author="",

  34. )

  35. 

  36. 

  37. class file4upload(Plugin):

  38.     def __init__(self):

  39.         super().__init__()

  40.         try:

  41.             curdir = os.path.dirname(__file__)

  42.             config_path = os.path.join(curdir, "config.json")

  43.             if os.path.exists(config_path):

  44.                 with open(config_path, "r", encoding="utf-8") as f:

  45.                     self.config = json.load(f)

  46.             else:

  47.                 # 使用父类的方法来加载配置

  48.                 self.config = super().load_config()

  49. 

  50.                 if not self.config:

  51.                     raise Exception("config.json not found")

  52.             # 设置事件处理函数

  53.             self.handlers[Event.ON_HANDLE_CONTEXT] = self.on_handle_context

  54.             self.params_cache = ExpiredDict(300)

  55. 

  56.              # 从配置中提取所需的设置

  57.             self.keys = self.config.get("keys", {})

  58.             self.url_sum = self.config.get("url_sum", {})

  59.             self.search_sum = self.config.get("search_sum", {})

  60.             self.file_sum = self.config.get("file_sum", {})

  61.             self.image_sum = self.config.get("image_sum", {})

  62.             self.note = self.config.get("note", {})

  63. 

  64.             self.sum4all_key = self.keys.get("sum4all_key", "")

  65.             self.search1api_key = self.keys.get("search1api_key", "")

  66.             self.gemini_key = self.keys.get("gemini_key", "")

  67.             self.bibigpt_key = self.keys.get("bibigpt_key", "")

  68.             self.outputLanguage = self.keys.get("outputLanguage", "zh-CN")

  69.             self.opensum_key = self.keys.get("opensum_key", "")

  70.             self.open_ai_api_key = self.keys.get("open_ai_api_key", "")

  71.             self.model = self.keys.get("model", "gpt-3.5-turbo")

  72.             self.open_ai_api_base = self.keys.get("open_ai_api_base", "https://api.openai.com/v1")

  73.             self.xunfei_app_id = self.keys.get("xunfei_app_id", "")

  74.             self.xunfei_api_key = self.keys.get("xunfei_api_key", "")

  75.             self.xunfei_api_secret = self.keys.get("xunfei_api_secret", "")

  76.             self.perplexity_key = self.keys.get("perplexity_key", "")

  77.             self.flomo_key = self.keys.get("flomo_key", "")

  78.             # 之前提示

  79.             self.previous_prompt=''

  80. 

  81.             self.file_sum_enabled = self.file_sum.get("enabled", False)

  82.             self.file_sum_service = self.file_sum.get("service", "")

  83.             self.max_file_size = self.file_sum.get("max_file_size", 15000)

  84.             self.file_sum_group = self.file_sum.get("group", True)

  85.             self.file_sum_qa_prefix = self.file_sum.get("qa_prefix", "问")

  86.             self.file_sum_prompt = self.file_sum.get("prompt", "")

  87. 

  88.             self.image_sum_enabled = self.image_sum.get("enabled", False)

  89.             self.image_sum_service = self.image_sum.get("service", "")

  90.             self.image_sum_group = self.image_sum.get("group", True)

  91.             self.image_sum_qa_prefix = self.image_sum.get("qa_prefix", "问")

  92.             self.image_sum_prompt = self.image_sum.get("prompt", "")

  93. 

  94.             # 初始化成功日志

  95.             logger.info("[file4upload] inited.")

  96.         except Exception as e:

  97.             # 初始化失败日志

  98.             logger.warn(f"file4upload init failed: {e}")

  99. 

  100.     def on_handle_context(self, e_context: EventContext):

  101.         context = e_context["context"]

  102.         if context.type not in [ContextType.TEXT, ContextType.SHARING,ContextType.FILE,ContextType.IMAGE]:

  103.             return

  104.         msg: ChatMessage = e_context["context"]["msg"]

  105.         user_id = msg.from_user_id

  106.         content = context.content

  107.         isgroup = e_context["context"].get("isgroup", False)

  108.         # logger.info(f"user_id:{user_id},content:{content},isgroup:{isgroup}")

  109.         # logger.info(f'上下文参数缓存键字典：{self.params_cache.keys}')

  110.         # logger.info(f'user_id in self.params_cache: {user_id in self.params_cache}')

  111.         # 上次提示

  112.         if  context.type == ContextType.TEXT:

  113.             self.previous_prompt=msg.content

  114. 

  115.         if isgroup and not self.file_sum_group:

  116.             # 群聊中忽略处理文件

  117.             logger.info("群聊消息，文件处理功能已禁用")

  118.             return

  119.         logger.info("on_handle_context: 处理上下文开始")

  120.         context.get("msg").prepare()

  121.         # file_path = context.content

  122.         # logger.info(f"on_handle_context: 获取到文件路径 {file_path}")

  123.         

  124.         api_key='sk-5z2L4zy9T1w90j6e3T90ANZdyN2zLWClRwFnBzWgzdrG4onx'

  125. 

  126.         if context.type == ContextType.IMAGE:

  127.             file_path = context.content

  128.             logger.info(f"on_handle_context: 获取到文件路径 {file_path}")

  129.             print(f'处理首次上次的图片,准备抽取文字')

  130.             file_content=self.extract_content_by_llm(file_path,api_key)

  131.             self.params_cache[user_id] = {}

  132.             if file_content is not None:

  133.                 logger.info('图片中抽取文字,使用使用图片的文字请求LLM')

  134.                 

  135.                 messages = [{

  136.                                 "role": "system",

  137.                                 "content": "你是 Kimi，由 Moonshot AI 提供的人工智能助手，你更擅长中文和英文的对话。你会为用户提供安全，有帮助，准确的回答。同时，你会拒绝一切涉及恐怖主义，种族歧视，黄色暴力等问题的回答。Moonshot AI 为专有名词，不可翻译成其他语言。",

  138.                             },{

  139.                                 "role": "system",

  140.                                 "content": file_content,

  141.                             }]

  142.                 self.params_cache[user_id]['last_word_messages']=messages

  143.                 self.params_cache[user_id]['last_image_oss']=None

  144.             else:

  145.                 logger.info('不能抽取文字,使用图片oss请求LLM')

  146.                 # logger.info(f"on_handle_context: 获取到图片路径 {file_path}")

  147.                 # base64_image=self.encode_image_to_base64(file_path)

  148.                 

  149.                 # self.params_cache[user_id]['last_image_oss']=base64_image

  150.                 # self.params_cache[user_id]['last_word_messages']=None

  151.                 access_key_id = 'LTAI5tRTG6pLhTpKACJYoPR5'

  152.                 access_key_secret = 'E7dMzeeMxq4VQvLg7Tq7uKf3XWpYfN'

  153.                 # OSS区域对应的Endpoint

  154.                 endpoint = 'http://oss-cn-shanghai.aliyuncs.com'  # 根据你的区域选择

  155.                 # Bucket名称

  156.                 bucket_name = 'cow-agent'

  157.                 local_file_path=file_path

  158.                 oss_file_name=f'cow/{os.path.basename(file_path)}'

  159.                 logger.info(f'oss_file_name:{oss_file_name}\n local_file_path :{local_file_path}')

  160.                 file_url = upload_oss(access_key_id, access_key_secret, endpoint, bucket_name, local_file_path, oss_file_name)

  161.                 logger.info(f'写入图片缓存oss 地址{file_url}')

  162.                 self.params_cache[user_id]['last_image_oss']=file_url

  163.                 self.params_cache[user_id]['last_word_messages']=None

  164. 

  165. 

  166.             if self.previous_prompt == '':

  167.                 reply = Reply()

  168.                 reply.type = ReplyType.TEXT

  169.                 # reply.content = f"{remove_markdown(reply_content)}\n\n💬5min内输入{self.file_sum_qa_prefix}+问题，可继续追问" 

  170.                 reply.content = f"您刚刚上传了一张图片，请问我有什么可以帮您的呢？"

  171.                 e_context["reply"] = reply

  172.                 e_context.action = EventAction.BREAK

  173.                 return

  174.             else:

  175.                 e_context.action = EventAction.CONTINUE

  176. 

  177.         if context.type == ContextType.FILE:

  178.             file_path = context.content

  179.             logger.info(f"on_handle_context: 获取到文件路径 {file_path}")

  180.             print(f'处理首次上次的文件')

  181.             file_content=self.extract_content_by_llm(file_path,api_key)

  182.             if file_content is not None:

  183.                 self.params_cache[user_id] = {}

  184.                 messages = [{

  185.                                 "role": "system",

  186.                                 "content": "你是 Kimi，由 Moonshot AI 提供的人工智能助手，你更擅长中文和英文的对话。你会为用户提供安全，有帮助，准确的回答。同时，你会拒绝一切涉及恐怖主义，种族歧视，黄色暴力等问题的回答。Moonshot AI 为专有名词，不可翻译成其他语言。",

  187.                             },{

  188.                                 "role": "system",

  189.                                 "content": file_content,

  190.                             }]

  191.                 self.params_cache[user_id]['last_word_messages']=messages

  192.                 self.params_cache[user_id]['last_image_oss']=None

  193. 

  194.             if self.previous_prompt == '':

  195.                 reply = Reply()

  196.                 reply.type = ReplyType.TEXT

  197.                 # reply.content = f"{remove_markdown(reply_content)}\n\n💬5min内输入{self.file_sum_qa_prefix}+问题，可继续追问" 

  198.                 reply.content = f"您刚刚上传了一个文件，请问我有什么可以帮您的呢？如可以问“请总结分析这份报告文件，同时，提供治疗和健康建议。”"

  199.                 e_context["reply"] = reply

  200.                 e_context.action = EventAction.BREAK

  201.                 return

  202.             else:

  203.                 e_context.action = EventAction.CONTINUE

  204.                 

  205.         # if user_id in self.params_cache and (self.params_cache[user_id]['last_word_messages']!=None):

  206.         if user_id in self.params_cache:

  207.             if 'last_word_messages' in self.params_cache[user_id] and self.params_cache[user_id]['last_word_messages'] is not None: 

  208.                 print(f'缓存处理已经上传的文件')

  209.                 # last_word_messages=self.params_cache[user_id]['last_word_messages']

  210.                 # cache_messages=last_word_messages[:2]

  211.                 cache_messages=self.params_cache[user_id]['last_word_messages']

  212.                 messages = [

  213.                     *cache_messages, 

  214.                     {

  215.                         "role": "user",

  216.                         "content": self.previous_prompt ,#msg.content,

  217.                     },

  218.                 ]

  219.                 self.handle_file_upload(messages, e_context)

  220. 

  221.         # if user_id in self.params_cache and ('last_image_oss' in self.params_cache[user_id]  or self.params_cache[user_id]['last_image_oss']!=None):

  222.         if user_id in self.params_cache:

  223.             if 'last_image_oss' in self.params_cache[user_id] and self.params_cache[user_id]['last_image_oss'] is not None:           

  224.                 print(f'缓存处理已经oss图片的文件')

  225.                 file_url=self.params_cache[user_id]['last_image_oss']

  226.                 messages = [{

  227.                                     "role": "system",

  228.                                     "content": "你是一个能能描述任何图片的智能助手",

  229.                                 },

  230.                             {

  231.                                 "role": "user",

  232.                                 "content": f'{file_url}\n{self.previous_prompt}',

  233.                         }]

  234.                 # messages=[

  235.                 #     {

  236.                 #         "role": "user",

  237.                 #         "content": [

  238.                 #             {

  239.                 #                 "type": "image_url",

  240.                 #                 "image_url": {

  241.                 #                     "url": f"{file_url}"

  242.                 #                 }

  243.                 #             },

  244.                 #             {

  245.                 #                 "type": "text",

  246.                 #                 "text": f"{self.previous_prompt}"

  247.                 #             }

  248.                 #         ]

  249.                 #     }

  250.                 # ]

  251. 

  252. 

  253.                 self.handle_images_oos(messages, e_context)           

  254.             

  255.     def handle_file_upload(self, messages, e_context):

  256.         logger.info("handle_file: 向LLM发送内容总结请求")

  257.         

  258.         msg: ChatMessage = e_context["context"]["msg"]

  259.         user_id = msg.from_user_id

  260.         user_params = self.params_cache.get(user_id, {})

  261.         prompt = user_params.get('prompt', self.file_sum_prompt)

  262.         

  263.         self.params_cache[user_id] = {}

  264. 

  265. 

  266.         try:

  267.             api_key = "sk-5z2L4zy9T1w90j6e3T90ANZdyN2zLWClRwFnBzWgzdrG4onx"

  268.             # base_url = "https://api.moonshot.cn/v1",

  269.             api_url = "https://api.moonshot.cn/v1/chat/completions"

  270.             headers = {

  271.                 'Content-Type': 'application/json',

  272.                 'Authorization': f'Bearer {api_key}'

  273.             }

  274. 

  275.             data={

  276.                 "model": "moonshot-v1-128k",

  277.                 "messages":messages,

  278.                 # "temperature": 0.3

  279.             }

  280.             response = requests.post(url=api_url, headers=headers, data=json.dumps(data))

  281.             logger.info(f'handle_file_upload: 请求文件内容{json.dumps(messages, ensure_ascii=False)}')

  282.             response.raise_for_status()

  283.             response_data = response.json()

  284.             if "choices" in response_data and len(response_data["choices"]) > 0:

  285.                     first_choice = response_data["choices"][0]

  286.                     if "message" in first_choice and "content" in first_choice["message"]:

  287.                         response_content = first_choice["message"]["content"].strip()  # 获取响应内容

  288.                         reply_content = response_content.replace("\\n", "\n")  # 替换 \\n 为 \n

  289.                         # self.params_cache[user_id]['last_word_messages']=messages

  290.                         # if self.params_cache[user_id]['last_word_messages']!=None:

  291.                         #     self.params_cache[user_id]['last_word_messages']=messages

  292. 

  293.                         self.previous_prompt =''

  294.                     else:

  295.                         logger.error("Content not found in the response")

  296.                         reply_content = "Content not found in the LLM API response"

  297.             else:

  298.                 logger.error("No choices available in the response")

  299.                 reply_content = "No choices available in the LLM API response"

  300.         except requests.exceptions.RequestException as e:

  301.             logger.error(f"Error calling LLM API: {e}")

  302.             reply_content = f"An error occurred while calling LLM API"

  303. 

  304.         reply = Reply()

  305.         reply.type = ReplyType.TEXT

  306.         reply.content = f"{remove_markdown(reply_content)}"

  307.         e_context["reply"] = reply

  308.         e_context.action = EventAction.BREAK_PASS

  309.       

  310. 

  311.     def handle_images_base64(self, messages, e_context):

  312.         logger.info("handle_file: 向LLM发送内容总结请求")

  313.         

  314.         msg: ChatMessage = e_context["context"]["msg"]

  315.         user_id = msg.from_user_id

  316.         user_params = self.params_cache.get(user_id, {})

  317.         prompt = user_params.get('prompt', self.file_sum_prompt)

  318.         

  319.         

  320. 

  321. 

  322.         try:

  323.             # api_key = "sk-5z2L4zy9T1w90j6e3T90ANZdyN2zLWClRwFnBzWgzdrG4onx"

  324.             # # base_url = "https://api.moonshot.cn/v1",

  325.             # api_url = "https://api.moonshot.cn/v1/chat/completions"

  326. 

  327.             

  328. 

  329.             # api_key = "sk-5dyg7PMUNeoSqHH807453eB06f434c34Ba6fB4764aC8358c"

  330.             # api_url = "http://106.15.182.218:3001/v1/chat/completions"

  331.             # headers = {

  332.             #     'Content-Type': 'application/json',

  333.             #     'Authorization': f'Bearer {api_key}'

  334.             # }

  335. 

  336.             # data={

  337.             #     "model": "moonshot-v1-128k",

  338.             #     "messages":messages,

  339.             #     # "temperature": 0.3

  340.             # }

  341.             # response = requests.post(url=api_url, headers=headers, json=data)

  342.             

  343.             base64_image=self.encode_image_to_base64('tmp/240926-164856.png')

  344.             api_key = self.open_ai_api_key

  345.             api_base = f"{self.open_ai_api_base}/chat/completions"

  346.             logger.info(api_base)

  347.             payload = {

  348.                 "model": "moonshot-v1-128k",

  349.                 "messages": [

  350.                     {

  351.                         "role": "user",

  352.                         "content": [

  353.                             {

  354.                                 "type": "text",

  355.                                 "text": self.previous_prompt

  356.                             },

  357.                             {

  358.                                 "type": "image_url",

  359.                                 "image_url": {

  360.                                     "url": f"data:image/jpeg;base64,{base64_image}"

  361.                                 }

  362.                             }

  363.                         ]

  364.                     }

  365.                 ],

  366.                 "max_tokens": 3000

  367.             }

  368. 

  369.             # payload = {

  370.             #     "model": "moonshot-v1-128k",

  371.             #     "messages": messages,

  372.             #     "max_tokens": 3000

  373.             # }

  374.             headers = {

  375.                 "Content-Type": "application/json",

  376.                 "Authorization": f"Bearer {api_key}"

  377.             }

  378.             logger.info('开始')

  379.             response = requests.post(api_base, headers=headers, json=payload)

  380. 

  381. 

  382.             # logger.info(f'handle_file_upload: 请求文件内容{json.dumps(messages, ensure_ascii=False)}')

  383.             response.raise_for_status()

  384.             response_data = response.json()

  385.             if "choices" in response_data and len(response_data["choices"]) > 0:

  386.                     first_choice = response_data["choices"][0]

  387.                     if "message" in first_choice and "content" in first_choice["message"]:

  388.                         response_content = first_choice["message"]["content"].strip()  # 获取响应内容

  389.                         reply_content = response_content.replace("\\n", "\n")  # 替换 \\n 为 \n

  390.                         # self.params_cache[user_id]['last_word_messages']=messages

  391.                         # if self.params_cache[user_id]['last_word_messages']!=None:

  392.                         #     self.params_cache[user_id]['last_word_messages']=messages

  393. 

  394.                         self.previous_prompt =''

  395.                     else:

  396.                         logger.error("Content not found in the response")

  397.                         reply_content = "Content not found in the LLM API response"

  398.             else:

  399.                 logger.error("No choices available in the response")

  400.                 reply_content = "No choices available in the LLM API response"

  401.         except requests.exceptions.RequestException as e:

  402.             logger.error(f"Error calling LLM API: {e}")

  403.             reply_content = f"An error occurred while calling LLM API"

  404. 

  405.         reply = Reply()

  406.         reply.type = ReplyType.TEXT

  407.         reply.content = f"{remove_markdown(reply_content)}"

  408.         e_context["reply"] = reply

  409.         e_context.action = EventAction.BREAK_PASS

  410.       

  411.     def handle_images_oos(self, messages, e_context):

  412.         logger.info("handle_file: 向LLM发送内容总结请求")

  413.         

  414.         msg: ChatMessage = e_context["context"]["msg"]

  415.         user_id = msg.from_user_id

  416.         user_params = self.params_cache.get(user_id, {})

  417.         prompt = user_params.get('prompt', self.file_sum_prompt)

  418.         

  419.         self.params_cache[user_id] = {}

  420. 

  421. 

  422.         try:

  423.             api_key = self.open_ai_api_key

  424.             api_base = f"{self.open_ai_api_base}/chat/completions"

  425.             logger.info(api_base)

  426.             payload = {

  427.                 "model": "7374349217580056592",

  428.                 "messages":messages,

  429.                 "max_tokens": 3000

  430.             }

  431. 

  432.             # payload = {

  433.             #     "model": "moonshot-v1-128k",

  434.             #     "messages": messages,

  435.             #     "max_tokens": 3000

  436.             # }

  437.             headers = {

  438.                 "Content-Type": "application/json",

  439.                 "Authorization": f"Bearer {api_key}"

  440.             }

  441.             # logger.info('开始')

  442.             response = requests.post(api_base, headers=headers, json=payload)

  443.             logger.info(f'handle_file_upload: 请求文件内容{json.dumps(messages, ensure_ascii=False)}')

  444.             response.raise_for_status()

  445.             response_data = response.json()

  446.             if "choices" in response_data and len(response_data["choices"]) > 0:

  447.                     first_choice = response_data["choices"][0]

  448.                     if "message" in first_choice and "content" in first_choice["message"]:

  449.                         response_content = first_choice["message"]["content"].strip()  # 获取响应内容

  450.                         reply_content = response_content.replace("\\n", "\n")  # 替换 \\n 为 \n

  451.                         # self.params_cache[user_id]['last_word_messages']=messages

  452.                         # if self.params_cache[user_id]['last_word_messages']!=None:

  453.                         #     self.params_cache[user_id]['last_word_messages']=messages

  454. 

  455.                         self.previous_prompt =''

  456.                     else:

  457.                         logger.error("Content not found in the response")

  458.                         reply_content = "Content not found in the LLM API response"

  459.             else:

  460.                 logger.error("No choices available in the response")

  461.                 reply_content = "No choices available in the LLM API response"

  462.         except requests.exceptions.RequestException as e:

  463.             logger.error(f"Error calling LLM API: {e}")

  464.             reply_content = f"An error occurred while calling LLM API"

  465. 

  466.         reply = Reply()

  467.         reply.type = ReplyType.TEXT

  468.         reply.content = f"{remove_markdown(reply_content)}"

  469.         e_context["reply"] = reply

  470.         e_context.action = EventAction.BREAK_PASS

  471.     

  472.     def extract_content_by_llm(self, file_path: str, api_key: str) -> str:

  473.         logger.info(f'大模型开始抽取文字')

  474.         try:

  475.             headers = {

  476.                 'Authorization': f'Bearer {api_key}'

  477.             }

  478.             data = {

  479.                 'purpose': 'file-extract',

  480.             }

  481.             file_name=os.path.basename(file_path)

  482.             files = {

  483.                 'file': (file_name, open(Path(file_path), 'rb')),

  484.             }

  485.             print(files)

  486.             api_url='https://api.moonshot.cn/v1/files'

  487.             response = requests.post(api_url, headers=headers, files=files, data=data)

  488.             # print(response.text)

  489.             response_data = response.json()

  490.             file_id = response_data.get('id')

  491.             # print(f'文件id:{file_id}')

  492.             response=requests.get(url=f"https://api.moonshot.cn/v1/files/{file_id}/content", headers=headers)

  493.             print(response.text)

  494.             response_data = response.json()

  495.             content = response_data.get('content')

  496.             return content

  497.         except requests.exceptions.RequestException as e:

  498.             logger.error(f"Error calling LLM API: {e}")

  499.             return None

  500. 

  501.     def encode_image_to_base64(self, image_path):

  502.         logger.info(f"开始处理图片: {image_path}")

  503.         try:

  504.             with Image.open(image_path) as img:

  505.                 logger.info(f"成功打开图片. 原始大小: {img.size}")

  506.                 if img.width > 1024:

  507.                     new_size = (1024, int(img.height*1024/img.width))

  508.                     img = img.resize(new_size)

  509.                     img.save(image_path)  # 保存调整大小后的图片

  510.                     logger.info(f"调整图片大小至: {new_size}")

  511. 

  512.             with open(image_path, "rb") as image_file:

  513.                 img_byte_arr = image_file.read()

  514.                 logger.info(f"读取图片完成. 大小: {len(img_byte_arr)} 字节")

  515. 

  516.             encoded = base64.b64encode(img_byte_arr).decode('ascii')

  517.             logger.info(f"Base64编码完成. 编码后长度: {len(encoded)}")

  518.             return encoded

  519.         except Exception as e:

  520.             logger.error(f"图片编码过程中发生错误: {str(e)}", exc_info=True)

  521.             raise

  522. 

  523. def remove_markdown(text):

  524.     # 替换Markdown的粗体标记

  525.     text = text.replace("**", "")

  526.     # 替换Markdown的标题标记

  527.     text = text.replace("### ", "").replace("## ", "").replace("# ", "")

  528.     return text

  529. 

  530. 

  531. def upload_oss(access_key_id, access_key_secret, endpoint, bucket_name, local_file_path, oss_file_name, expiration_days=7):

  532.     """

  533.     上传文件到阿里云OSS并设置生命周期规则，同时返回文件的公共访问地址。

  534.     

  535.     :param access_key_id: 阿里云AccessKey ID

  536.     :param access_key_secret: 阿里云AccessKey Secret

  537.     :param endpoint: OSS区域对应的Endpoint

  538.     :param bucket_name: OSS中的Bucket名称

  539.     :param local_file_path: 本地文件路径

  540.     :param oss_file_name: OSS中的文件存储路径

  541.     :param expiration_days: 文件保存天数，默认7天后删除

  542.     :return: 文件的公共访问地址

  543.     """

  544.     

  545.     # 创建Bucket实例

  546.     auth = oss2.Auth(access_key_id, access_key_secret)

  547.     bucket = oss2.Bucket(auth, endpoint, bucket_name)

  548. 

  549.     ### 1. 设置生命周期规则 ###

  550.     rule_id = f'delete_after_{expiration_days}_days'  # 规则ID

  551.     prefix = oss_file_name.split('/')[0] + '/'  # 设置规则应用的前缀为文件所在目录

  552. 

  553.     # 定义生命周期规则

  554.     rule = oss2.models.LifecycleRule(rule_id, prefix, status=oss2.models.LifecycleRule.ENABLED,

  555.                                      expiration=oss2.models.LifecycleExpiration(days=expiration_days))

  556. 

  557.     # 设置Bucket的生命周期

  558.     lifecycle = oss2.models.BucketLifecycle([rule])

  559.     bucket.put_bucket_lifecycle(lifecycle)

  560. 

  561.     print(f"已设置生命周期规则：文件将在{expiration_days}天后自动删除")

  562. 

  563.     ### 2. 上传文件到OSS ###

  564.     bucket.put_object_from_file(oss_file_name, local_file_path)

  565. 

  566.     ### 3. 构建公共访问URL ###

  567.     file_url = f"http://{bucket_name}.{endpoint.replace('http://', '')}/{oss_file_name}"

  568. 

  569.     print(f"文件上传成功，公共访问地址：{file_url}")

  570.     

  571.     return file_url

  572.