SAGE-Bench

Sleeping

+# %%
+import datetime as dt
+import os
+import re
+from logging import Logger
+from multiprocessing import Pool
+from multiprocessing.pool import ThreadPool
+from pathlib import Path
+from typing import List, Union
+import oss2
+from loguru import logger
+from oss2.credentials import EnvironmentVariableCredentialsProvider
+from tqdm import tqdm
+from compassflow.constants import DATADIR
+from compassflow.oss.oss import OssBucket
+from compassflow.utils import starstarmap
+class OSSFileManager:
+    def __init__(
+        self,
+        oss_access_key_id: str = None,
+        oss_access_key_secret: str = None,
+        region: str = "http://oss-cn-shanghai.aliyuncs.com",
+        bucket_name: str = "opencompass",
+        oss_block_name: str = None,
+        logger: Logger = logger,
+    ) -> None:
+        """OSS File Manager
+        Args:
+            oss_access_key_id (str, optional): _description_. Defaults to None.
+            oss_access_key_secret (str, optional): _description_. Defaults to None.
+            region (_type_, optional): _description_. Defaults to 'http://oss-cn-shanghai.aliyuncs.com'.
+            bucket_name (str, optional): _description_. Defaults to 'opencompass'.
+            oss_block_name: oss_block_name which is defined in the prefect
+            logger (Logger, optional): _description_. Defaults to logger.
+        """
+        self.logger = logger
+        if oss_block_name is not None:
+            oss_bucket = OssBucket.load(oss_block_name)
+            self._bucket = oss_bucket._get_bucket()
+            return
+        # 阿里云账号AccessKey拥有所有API的访问权限，风险很高。强烈建议您创建并使用RAM账号进行API访问或日常运维，请登录RAM控制台创建RAM账号。
+        if oss_access_key_id is not None and oss_access_key_secret is not None:
+            os.environ["OSS_ACCESS_KEY_ID"] = oss_access_key_id
+            os.environ["OSS_ACCESS_KEY_SECRET"] = oss_access_key_secret
+        if (
+            os.getenv("OSS_ACCESS_KEY_ID") is None
+            or os.getenv("OSS_ACCESS_KEY_SECRET") is None
+        ):
+            raise ValueError("Access Key ID and Access Key Secret cannot be empty.")
+        auth = oss2.ProviderAuth(EnvironmentVariableCredentialsProvider())
+        # Endpoint以杭州为例，其它Region请按实际情况填写。
+        # 填写Bucket名称，例如examplebucket。
+        bucket = oss2.Bucket(
+            auth=auth,
+            endpoint=region,
+            bucket_name=bucket_name,
+        )
+        self._bucket = bucket
+    def list_latest_files_by_date(
+        self,
+        object_dir: str = "",
+        delimiter: str = "/",
+        start_date: Union[str, dt.date, dt.datetime] = None,
+        end_date: Union[str, dt.date, dt.datetime] = None,
+        max_num_files: int = 10,
+        date_pattern: str = r"([0-9]{4}-[0-9]{2}-[0-9]{2})",
+        file_date_format: str = "%Y-%m-%d",
+        suffix: str = "",
+    ) -> List[Union[str, Path]]:
+        """List the latest files by date in an OSS bucket directory
+        Args:
+            object_dir (str, optional): _description_. Defaults to ''.
+            delimiter (str, optional): _description_. Defaults to '/'.
+            start_date (Union[str, dt.date], optional): _description_. Defaults to None.
+            end_date (Union[str, dt.date], optional): _description_. Defaults to None.
+            max_num_files (int, optional): _description_. Defaults to 10.
+            date_pattern (str, optional): _description_. Defaults to r'^([0-9]{4}-[0-9]{2}-[0-9]{2})'.
+            suffix (str, optional): _description_. Defaults to ''.
+        Returns:
+            List[Union[str, Path]]: _description_
+        """
+        if start_date is not None:
+            if isinstance(start_date, str):
+                start_date = dt.datetime.strptime(start_date.replace("-", ""), "%Y%m%d")
+            if isinstance(start_date, dt.date):
+                start_date = dt.datetime(
+                    start_date.year, start_date.month, start_date.day, 0, 0, 0
+                )
+        if end_date is not None:
+            if isinstance(end_date, str):
+                end_date = dt.datetime.strptime(end_date.replace("-", ""), "%Y%m%d")
+            if isinstance(end_date, dt.date):
+                end_date = dt.datetime(
+                    end_date.year, end_date.month, end_date.day, 0, 0, 0
+                )
+        object_iter = oss2.ObjectIterator(
+            bucket=self._bucket, prefix=object_dir, delimiter=delimiter
+        )
+        root_dir = Path(object_dir)
+        filenames = []
+        for filename in object_iter:
+            filename = filename.key.replace(object_dir, "")
+            # print(filename)
+            if filename.endswith(suffix):
+                # Match date pattern in filename
+                date_search = re.search(date_pattern, filename)
+                if date_search:
+                    file_date = dt.datetime.strptime(
+                        date_search.group(1), file_date_format
+                    )
+                else:
+                    self.logger.warning(
+                        f"date pattern doesn't match, skipping file {filename}"
+                    )
+                    continue
+                # Check if file date within specified range
+                if start_date is not None:
+                    if start_date > file_date:
+                        continue
+                if end_date is not None:
+                    if end_date < file_date:
+                        continue
+                filepath = root_dir / filename
+                # name_tstamp_tuple = (filepath, os.path.getmtime(filepath))
+                # filenames.append(name_tstamp_tuple)
+                filenames.append(str(filepath))
+        # sort by tstamp
+        # filenames = sorted(filenames, key=lambda x: x[1])
+        filenames = sorted(filenames)
+        # filenames = [x[0] for x in filenames]
+        max_num_files = max_num_files or len(filenames)
+        filenames = filenames[-max_num_files:]
+        self.logger.info(f"{filenames=}")
+        return filenames
+    def download_object_to_file(
+        self,
+        oss_file_path: str | Path,
+        local_file_path: str | Path,
+        replace: bool = False,
+        make_dir: bool = False,
+        print_logs: bool = True,
+    ):
+        """Download a single OSS object to local file.
+        Args:
+            oss_file_path (str): _description_
+            local_file_path (str): _description_
+            replace (bool, optional): _description_. Defaults to False.
+            make_dir (bool, optional): Whether to create intermediate dirs if they don't exist. Defaults to False.
+            print_logs: bool, optional): Whether to print logs. Defaults to True.
+        """
+        if isinstance(local_file_path, str):
+            local_file_path = Path(local_file_path)
+        if not replace:
+            if local_file_path.exists():
+                if print_logs:
+                    err_msg = f"{local_file_path} already exists, skipping file..."
+                    self.logger.info(err_msg)
+                return
+        if print_logs:
+            if local_file_path.exists():
+                err_msg = f"{local_file_path} already exists, replacing file..."
+                self.logger.info(err_msg)
+        if make_dir:
+            os.makedirs(local_file_path.parent, exist_ok=True)
+        self._bucket.get_object_to_file(
+            key=str(oss_file_path),
+            filename=local_file_path,
+        )
+    def download_objects_to_files(
+        self,
+        file_download_mapping: list[tuple[str | Path, str | Path]],
+        oss_base_dir: str = None,
+        local_base_dir: str = None,
+        replace: bool = True,
+        num_threads: int = 1,
+        **kwargs,
+    ) -> None:
+        """Download objects from OSS to local storage.
+        Args:
+            file_download_mapping (list[tuple[str | Path, str | Path]]): A list of file path pairs that maps the OSS file path (to download) to the local file path (download location).
+            oss_base_dir (str): OSS directory path to be prepended to all OSS file paths.
+            local_base_dir (str, optional): Base directory path to be prepended to all local file paths.
+            replace (bool, optional): _description_. Defaults to True.
+            num_threads (int, optional): _description_. Defaults to 1.
+            **kwargs: Additional keyword arguments passed to `list_latest_files_by_date`
+        """
+        if isinstance(oss_base_dir, str):
+            oss_base_dir = Path(oss_base_dir)
+        if isinstance(local_base_dir, str):
+            local_base_dir = Path(local_base_dir)
+        if not isinstance(file_download_mapping, list):
+            raise TypeError("file_download_mapping must be a list of 2 value tuples.")
+        for item in file_download_mapping:
+            if not isinstance(item, tuple):
+                raise TypeError(
+                    "Each item in the file_download_mapping list must be a 2 value tuple."
+                )
+            if len(item) != 2:
+                raise ValueError(
+                    "Each tuple in the file_download_mapping list must be length 2."
+                )
+        if num_threads == 1:
+            for local_file_path, oss_file_path in file_download_mapping:
+                if local_base_dir is not None:
+                    local_file_path = local_base_dir / local_file_path
+                if oss_base_dir is not None:
+                    oss_file_path = oss_base_dir / oss_file_path
+                self.download_object_to_file(
+                    oss_file_path=oss_file_path,
+                    local_file_path=local_file_path,
+                    replace=replace,
+                )
+            return
+        # Start multithreaded process if num_threads > 1
+        with ThreadPool(num_threads) as p:
+            pool_args_list = []
+            for local_file_path, oss_file_path in file_download_mapping:
+                if local_base_dir is not None:
+                    local_file_path = local_base_dir / local_file_path
+                if oss_base_dir is not None:
+                    oss_file_path = oss_base_dir / oss_file_path
+                args_dict = dict(
+                    oss_file_path=oss_file_path,
+                    local_file_path=local_file_path,
+                    replace=replace,
+                    print_logs=True,
+                )
+                pool_args_list.append(args_dict)
+            ret_all = list(
+                tqdm(
+                    starstarmap(
+                        pool=p,
+                        fn=self.download_object_to_file,
+                        kwargs_iter=pool_args_list,
+                    ),
+                    total=len(file_download_mapping),
+                )
+            )
+    def download_latest_objects_to_dir(
+        self,
+        oss_object_dir: str,
+        local_dir: str,
+        start_date: Union[str, dt.date, dt.datetime] = None,
+        end_date: Union[str, dt.date, dt.datetime] = None,
+        date_pattern: str = r"([0-9]{4}-[0-9]{2}-[0-9]{2})",
+        file_date_format: str = "%Y-%m-%d",
+        max_num_files: int = 5,
+        replace: bool = True,
+        make_dir: bool = True,
+        delimiter: str = "/",
+        num_threads: int = 1,
+        suffix: str = "",
+        **kwargs,
+    ) -> None:
+        """Download the latest objects from oss to local directory
+        Args:
+            oss_object_dir (str): _description_
+            local_dir (str): _description_
+            start_date (Union[str, dt.date, dt.datetime], optional): _description_. Defaults to None.
+            end_date (Union[str, dt.date, dt.datetime], optional): _description_. Defaults to None.
+            max_num_files (int, optional): _description_. Defaults to 5.
+            replace (bool, optional): _description_. Defaults to True.
+            delimiter (str, optional): _description_. Defaults to '/'.
+            num_threads (int, optional): _description_. Defaults to 1.
+            suffix (str, optional): _description_. Defaults to ''.
+            **kwargs: Additional keyword arguments passed to `list_latest_files_by_date`
+        """
+        if isinstance(local_dir, str):
+            local_dir = Path(local_dir)
+        oss_file_list = self.list_latest_files_by_date(
+            object_dir=oss_object_dir,
+            delimiter=delimiter,
+            start_date=start_date,
+            end_date=end_date,
+            date_pattern=date_pattern,
+            file_date_format=file_date_format,
+            max_num_files=max_num_files,
+            suffix=suffix,
+            **kwargs,
+        )
+        if num_threads == 1:
+            for oss_file_path in oss_file_list:
+                file_name = Path(oss_file_path).name
+                local_file_path = local_dir / file_name
+                self.download_object_to_file(
+                    oss_file_path=oss_file_path,
+                    local_file_path=local_file_path,
+                    replace=replace,
+                    make_dir=make_dir,
+                )
+            return
+        # Start multithreaded process if num_threads > 1
+        with ThreadPool(num_threads) as p:
+            pool_args_list = []
+            for oss_file_path in oss_file_list:
+                file_name = Path(oss_file_path).name
+                local_file_path = local_dir / file_name
+                args_dict = dict(
+                    oss_file_path=str(oss_file_path),
+                    local_file_path=str(local_file_path),
+                    replace=replace,
+                    make_dir=make_dir,
+                    print_logs=True,
+                )
+                pool_args_list.append(args_dict)
+            ret_all = list(
+                tqdm(
+                    starstarmap(
+                        pool=p,
+                        fn=self.download_object_to_file,
+                        kwargs_iter=pool_args_list,
+                    ),
+                    total=len(oss_file_list),
+                )
+            )
+    def upload_file_to_object(
+        self,
+        local_file_path: str,
+        oss_file_path: str | Path,
+        replace: bool = False,
+        print_logs: bool = True,
+    ):
+        """Upload a single local file to OSS
+        Args:
+            oss_file_path (str): _description_
+            local_file_path (str): _description_
+            replace (bool, optional): _description_. Defaults to False.
+        """
+        if isinstance(local_file_path, Path):
+            local_file_path = str(local_file_path)
+        if isinstance(oss_file_path, Path):
+            oss_file_path = str(oss_file_path)
+        # Check if file already exists
+        is_file_exists = self._bucket.object_exists(
+            key=oss_file_path,
+        )
+        if is_file_exists:
+            if replace:
+                if print_logs:
+                    err_msg = f"{oss_file_path} already exists, replacing file..."
+                    self.logger.info(err_msg)
+                self._bucket.put_object_from_file(
+                    key=str(oss_file_path),
+                    filename=local_file_path,
+                )
+            else:
+                if print_logs:
+                    err_msg = f"{oss_file_path} already exists, skipping file..."
+                    self.logger.info(err_msg)
+                return
+        self._bucket.put_object_from_file(
+            key=oss_file_path,
+            filename=local_file_path,
+        )
+    def upload_files_to_objects(
+        self,
+        file_upload_mapping: list[tuple[str | Path, str | Path]],
+        local_base_dir: str = None,
+        oss_base_dir: str = None,
+        replace: bool = True,
+        num_threads: int = 1,
+        **kwargs,
+    ) -> None:
+        """Upload files from local storage to OSS.
+        Args:
+            file_upload_mapping (list[tuple[str | Path, str | Path]]): A list of file path pairs that maps the local file path (to upload) to the OSS file path (upload location).
+            oss_base_dir (str): OSS directory path to be prepended to all OSS file paths.
+            local_base_dir (str, optional): Base directory path to be prepended to all local file paths.
+            replace (bool, optional): _description_. Defaults to True.
+            num_threads (int, optional): _description_. Defaults to 1.
+            **kwargs: Additional keyword arguments passed to `list_latest_files_by_date`
+        """
+        if isinstance(oss_base_dir, str):
+            oss_base_dir = Path(oss_base_dir)
+        if isinstance(local_base_dir, str):
+            local_base_dir = Path(local_base_dir)
+        if not isinstance(file_upload_mapping, list):
+            raise TypeError("file_upload_mapping must be a list of 2 value tuples.")
+        for item in file_upload_mapping:
+            if not isinstance(item, tuple):
+                raise TypeError(
+                    "Each item in the file_upload_mapping list must be a 2 value tuple."
+                )
+            if len(item) != 2:
+                raise ValueError(
+                    "Each tuple in the file_upload_mapping list must be length 2."
+                )
+        if num_threads == 1:
+            for local_file_path, oss_file_path in file_upload_mapping:
+                if local_base_dir is not None:
+                    local_file_path = local_base_dir / local_file_path
+                if oss_base_dir is not None:
+                    oss_file_path = oss_base_dir / oss_file_path
+                self.upload_file_to_object(
+                    oss_file_path=oss_file_path,
+                    local_file_path=local_file_path,
+                    replace=replace,
+                )
+            return
+        # Start multithreaded process if num_threads > 1
+        with ThreadPool(num_threads) as p:
+            pool_args_list = []
+            for local_file_path, oss_file_path in file_upload_mapping:
+                if local_base_dir is not None:
+                    local_file_path = local_base_dir / local_file_path
+                if oss_base_dir is not None:
+                    oss_file_path = oss_base_dir / oss_file_path
+                args_dict = dict(
+                    oss_file_path=oss_file_path,
+                    local_file_path=local_file_path,
+                    replace=replace,
+                    print_logs=True,
+                )
+                pool_args_list.append(args_dict)
+            ret_all = list(
+                tqdm(
+                    starstarmap(
+                        pool=p,
+                        fn=self.upload_file_to_object,
+                        kwargs_iter=pool_args_list,
+                    ),
+                    total=len(file_upload_mapping),
+                )
+            )
+    @property
+    def bucket(self) -> oss2.Bucket:
+        return self._bucket
+# %%
+if __name__ == "__main__":
+    # %% Initialize
+    oss_file_manager = OSSFileManager(logger=logger)
+    # %% List the latest files by date (based on file suffix) in an OSS directory
+    oss_file_manager.list_latest_files_by_date(
+        "compass-arena/dev/data/conversations/", max_num_files=5
+    )

src/oss/oss_submission_handler.py ADDED Viewed

	@@ -0,0 +1,230 @@

+#!/usr/bin/env python3
+"""
+OSS提交处理器 - 替换原有的git/http提交方式
+在HuggingFace Spaces中直接将提交文件上传到OSS
+"""
+import os
+import sys
+import json
+from datetime import datetime
+from pathlib import Path
+from typing import Dict, Any
+# 添加上级目录到路径以导入oss_file_manager
+sys.path.append(os.path.join(os.path.dirname(__file__), '..', 'spaces'))
+from oss_file_manager import OSSFileManager
+class OSSSubmissionHandler:
+    """OSS提交处理器 - 将用户提交直接上传到OSS"""
+    def __init__(self, oss_submission_path: str = "atlas_eval/submissions/"):
+        """
+        初始化OSS提交处理器
+        Args:
+            oss_submission_path: OSS中存储提交文件的路径
+        """
+        self.oss_path = oss_submission_path
+        self.oss_manager = OSSFileManager()
+        print(f"📁 OSS提交路径: oss://opencompass/{oss_submission_path}")
+    def format_error(self, msg: str) -> str:
+        """格式化错误消息"""
+        return f"<p style='color: red; font-size: 16px;'>{msg}</p>"
+    def format_success(self, msg: str) -> str:
+        """格式化成功消息"""
+        return f"<p style='color: green; font-size: 16px;'>{msg}</p>"
+    def format_warning(self, msg: str) -> str:
+        """格式化警告消息"""
+        return f"<p style='color: orange; font-size: 16px;'>{msg}</p>"
+    def validate_sage_submission(self, submission_data: Dict[str, Any]) -> tuple[bool, str]:
+        """验证SAGE基准提交格式"""
+        # 检查必需的顶级字段
+        required_fields = ["submission_org", "submission_email", "predictions"]
+        for field in required_fields:
+            if field not in submission_data:
+                return False, f"缺少必需字段: {field}"
+        # 验证邮箱格式（基本验证）
+        email = submission_data["submission_email"]
+        if "@" not in email or "." not in email:
+            return False, "邮箱格式无效"
+        # 验证predictions
+        predictions = submission_data["predictions"]
+        if not isinstance(predictions, list) or len(predictions) == 0:
+            return False, "predictions必须是非空列表"
+        for i, prediction in enumerate(predictions):
+            # 检查必需的prediction字段
+            pred_required_fields = ["original_question_id", "content", "reasoning_content"]
+            for field in pred_required_fields:
+                if field not in prediction:
+                    return False, f"预测{i}中缺少字段: {field}"
+            # 验证content数组
+            content = prediction["content"]
+            reasoning_content = prediction["reasoning_content"]
+            if not isinstance(content, list) or len(content) != 4:
+                return False, f"预测{i}的content必须是包含4个项目的列表"
+            if not isinstance(reasoning_content, list) or len(reasoning_content) != 4:
+                return False, f"预测{i}的reasoning_content必须是包含4个项目的列表"
+            # 验证question ID
+            if not isinstance(prediction["original_question_id"], int):
+                return False, f"预测{i}的question ID必须是整数"
+        return True, "提交格式有效"
+    def generate_submission_filename(self, submission_data: Dict[str, Any]) -> str:
+        """生成提交文件名"""
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        org_name = submission_data["submission_org"].replace(" ", "_").replace("/", "_").replace("\\", "_")
+        return f"submission_{org_name}_{timestamp}.json"
+    def upload_to_oss(self, submission_data: Dict[str, Any], filename: str) -> tuple[bool, str]:
+        """上传提交文件到OSS"""
+        try:
+            # 创建临时本地文件
+            temp_file = f"/tmp/{filename}"
+            with open(temp_file, 'w', encoding='utf-8') as f:
+                json.dump(submission_data, f, indent=2, ensure_ascii=False)
+            # 上传到OSS
+            oss_file_path = f"{self.oss_path}{filename}"
+            print(f"⬆️ 上传到OSS: {oss_file_path}")
+            self.oss_manager.upload_file_to_object(
+                local_file_path=temp_file,
+                oss_file_path=oss_file_path,
+                replace=True
+            )
+            # 清理临时文件
+            os.remove(temp_file)
+            print(f"✅ OSS上传成功: {oss_file_path}")
+            return True, f"oss://opencompass/{oss_file_path}"
+        except Exception as e:
+            print(f"❌ OSS上传失败: {e}")
+            return False, str(e)
+    def process_sage_submission(self, submission_file_or_data, org_name=None, email=None) -> str:
+        """
+        处理SAGE基准提交文件 - OSS模式
+        替换原有的git/http方式，直接上传到OSS
+        """
+        try:
+            # 处理输入参数 - 可能是文件路径或者已经的数据
+            if submission_file_or_data is None:
+                return self.format_error("❌ 没有提供提交数据。")
+            # 如果是字符串，认为是文件路径
+            if isinstance(submission_file_or_data, str):
+                try:
+                    with open(submission_file_or_data, 'r', encoding='utf-8') as f:
+                        content = f.read()
+                    # 解析JSON
+                    submission_data = json.loads(content)
+                except Exception as e:
+                    return self.format_error(f"❌ 读取文件时出错: {str(e)}")
+            # 如果是字典，直接使用
+            elif isinstance(submission_file_or_data, dict):
+                submission_data = submission_file_or_data
+            else:
+                return self.format_error("❌ 无效的提交数据格式。")
+            # 如果表单提供了组织名和邮箱，使用表单数据
+            if org_name and email:
+                submission_data["submission_org"] = org_name.strip()
+                submission_data["submission_email"] = email.strip()
+            # 验证提交格式
+            is_valid, message = self.validate_sage_submission(submission_data)
+            if not is_valid:
+                return self.format_error(f"❌ 提交验证失败: {message}")
+            # 生成文件名
+            filename = self.generate_submission_filename(submission_data)
+            # 上传到OSS
+            success, result = self.upload_to_oss(submission_data, filename)
+            if not success:
+                return self.format_error(f"❌ 上传到OSS失败: {result}")
+            # 生成成功消息
+            org = submission_data["submission_org"]
+            email_addr = submission_data["submission_email"]
+            num_predictions = len(submission_data["predictions"])
+            success_msg = self.format_success(f"""
+🎉 <strong>提交成功！</strong><br><br>
+📋 <strong>提交信息:</strong><br>
+• 组织: {org}<br>
+• 邮箱: {email_addr}<br>
+• 预测数量: {num_predictions} 个问题<br>
+• 文件名: {filename}<br><br>
+🚀 <strong>存储位置:</strong><br>
+{result}<br><br>
+⚡ <strong>评测状态:</strong><br>
+您的提交已成功上传到云存储，自动评测系统将在5-15分钟内开始处理。<br><br>
+⏳ <strong>评测流程:</strong><br>
+1. 🔍 系统自动检测到新提交<br>
+2. ⬇️ 下载并验证提交格式<br>
+3. 🔬 使用LLM-as-Judge进行全面评估<br>
+4. 📊 计算各科目及总体准确率<br>
+5. 🏆 自动更新到排行榜<br><br>
+🕐 <strong>预计时间:</strong><br>
+评测完成时间约5-15分钟，取决于当前队列长度。<br>
+请稍后刷新排行榜查看结果。<br><br>
+🧪 感谢您参与SAGE科学推理基准测试！
+            """)
+            return success_msg
+        except Exception as e:
+            return self.format_error(f"❌ 提交处理失败: {str(e)}")
+# 兼容性函数 - 保持与原有代码的接口一致
+def process_sage_submission_simple(submission_file, org_name=None, email=None) -> str:
+    """
+    处理SAGE基准提交文件 - OSS模式
+    这是一个兼容性函数，保持与原有simple_submit.py的接口一致
+    """
+    handler = OSSSubmissionHandler()
+    return handler.process_sage_submission(submission_file, org_name, email)
+def format_error(msg):
+    return f"<p style='color: red; font-size: 16px;'>{msg}</p>"
+def format_success(msg):
+    return f"<p style='color: green; font-size: 16px;'>{msg}</p>"
+def format_warning(msg):
+    return f"<p style='color: orange; font-size: 16px;'>{msg}</p>"
+if __name__ == "__main__":
+    # 测试代码
+    print("🧪 测试OSS提交处理器")
+    # 检查环境变量
+    required_env_vars = ["OSS_ACCESS_KEY_ID", "OSS_ACCESS_KEY_SECRET"]
+    missing_vars = [var for var in required_env_vars if not os.getenv(var)]
+    if missing_vars:
+        print(f"❌ 缺少必需的环境变量: {missing_vars}")
+        exit(1)
+    handler = OSSSubmissionHandler()
+    print("✅ OSS提交处理器初始化成功")

src/submission/simple_submit.py CHANGED Viewed

@@ -12,9 +12,8 @@ from typing import Dict, Any
 from pathlib import Path
 # 导入OSS提交处理器
-sys.path.append(os.path.join(os.path.dirname(__file__), '..', '..', '..', 'oss_sage_evaluator'))
 try:
-    from oss_submission_handler import OSSSubmissionHandler
     OSS_AVAILABLE = True
 except ImportError as e:
     print(f"⚠️ OSS模块不可用，将使用备用模式: {e}")

 from pathlib import Path
 # 导入OSS提交处理器
 try:
+    from src.oss.oss_submission_handler import OSSSubmissionHandler
     OSS_AVAILABLE = True
 except ImportError as e:
     print(f"⚠️ OSS模块不可用，将使用备用模式: {e}")