Upload 15 files

.dockerignore

@@ -0,0 +1,16 @@
+*.egg-info
+*.pyc
+*.pyo
+*.pyd
+*.pyw
+*.pyz
+.github
+.vscode
+.gitignore
+.git
+.env
+__pycache__
+*.log*
+.data
+build
+trace_data

.env.template

@@ -0,0 +1,59 @@
+# LLM Model Config
+LLM_PROVIDER_GAIA = {YOUR_CONFIG:ant/openai}
+LLM_MODEL_NAME_GAIA = {YOUR_CONFIG}
+LLM_API_KEY_GAIA = {YOUR_CONFIG}
+LLM_BASE_URL_GAIA = {YOUR_CONFIG}
+LLM_TEMPERATURE_GAIA = 0.0
+
+LLM_PROVIDER_WEATHER = {YOUR_CONFIG:ant/openai}
+LLM_MODEL_NAME_WEATHER = {YOUR_CONFIG}
+LLM_API_KEY_WEATHER = {YOUR_CONFIG}
+LLM_BASE_URL_WEATHER = {YOUR_CONFIG}
+LLM_TEMPERATURE_WEATHER = 0.0
+
+# ===============MCP Server Configurations=================
+# Google Search
+GOOGLE_API_KEY={YOUR_CONFIG}
+GOOGLE_CSE_ID={YOUR_CONFIG}
+GOOGLE_SEARCH_ENGINE_ID={YOUR_CONFIG}
+
+# Google Map Server
+GOOGLE_MAPS_SECRET={YOUR_CONFIG}
+
+# GitHub Server
+GITHUB_ACCESS_TOKEN={YOUR_CONFIG}
+
+# E2B Server
+E2B_API_KEY={YOUR_CONFIG}
+
+# Filesystem Server
+FILESYSTEM_SERVER_WORKDIR=/app/aworld/gaia-benchmark/fs
+
+# Browser Use: skip verfication
+SKIP_LLM_API_KEY_VERIFICATION=true
+OPENAI_API_KEY={YOUR_CONFIG}
+
+# Apify Server
+APIFY_API_TOKEN={YOUR_CONFIG}
+
+# Audio Server
+AUDIO_LLM_API_KEY={YOUR_CONFIG}
+AUDIO_LLM_BASE_URL=https://api.zhizengzeng.com/v1
+AUDIO_LLM_MODEL_NAME=gpt-4o-transcribe
+
+# Image Server
+IMAGE_LLM_API_KEY={YOUR_CONFIG}
+IMAGE_LLM_BASE_URL=https://openrouter.ai/api/v1
+IMAGE_LLM_MODEL_NAME=anthropic/claude-3.7-sonnet
+
+# Video Server
+VIDEO_LLM_API_KEY={YOUR_CONFIG}
+VIDEO_LLM_BASE_URL=https://openrouter.ai/api/v1
+VIDEO_LLM_MODEL_NAME=anthropic/claude-3.7-sonnet
+
+#   - Zhizengzeng LLM ModelConfig
+LLM_API_KEY_ZZZ={YOUR_CONFIG}
+LLM_BASE_URL_ZZZ=https://api.zhizengzeng.com/v1
+
+# Weather Agent MCP API KEY, https://dev.qweather.com/
+OPENWEATHER_API_KEY={YOUR_CONFIG}

.gitignore

@@ -0,0 +1,136 @@
+# Python specific
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+build/
+develop-eggs/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+.pytest_cache/
+htmlcov/
+.coverage
+.coverage.*
+.cache
+coverage.xml
+*.cover
+
+# Log files
+*.log
+git-pre-push-hook-warn.log
+logs/
+logging/
+
+# Environment and virtual environments
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+.env
+.env.docker
+
+# IDE specific files
+.idea/
+.vscode/
+.cursor/
+*.swp
+*.swo
+.DS_Store
+.ipynb_checkpoints
+
+# OS generated files
+.DS_Store
+.DS_Store?
+._*
+.Spotlight-V100
+.Trashes
+ehthumbs.db
+Thumbs.db
+
+# Build artifacts
+*.pyc
+*.pyo
+*.pyd
+
+# Documentation
+docs/_build/
+docs/api/
+
+# Data files (customize as needed)
+*.csv
+*.dat
+*.out
+data/raw/
+data/processed/
+data/workspaces/
+gaia-benchmark/
+examples/gaia/gaia_dataset/
+
+# Compiled files
+*.com
+*.class
+*.dll
+*.exe
+*.o
+*.a
+
+# Package files
+*.7z
+*.dmg
+*.gz
+*.iso
+*.jar
+*.rar
+*.tar
+*.zip
+
+# Temporary files
+*.tmp
+*.temp
+.*.swp
+.*.swo
+*.webm
+*.mp4
+*.mp3
+*.avi
+*.mov
+*.flv
+*.wmv
+*.mkv
+*.npy
+*.xlsx
+*.xls
+*.docx
+*.doc
+*.pptx
+*.ppt
+*.pdf
+*.wav
+
+# Node Modules
+node_modules/
+
+aworld.db
+.data/
+examples/gaia/GAIA
+traces*.json
+trace_data
+
+examples/gaia/output
+examples/**/workspaces
+examples/**/data
+aworld/cmd/web/webui/dist

Dockerfile_gaia

@@ -0,0 +1,47 @@
+FROM python:3.11-slim AS base
+
+RUN mkdir -p /app/aworld
+
+WORKDIR /app/aworld
+
+RUN pip config set global.index-url https://mirrors.aliyun.com/pypi/simple/
+
+RUN rm -fv /etc/apt/sources.list.d/* && \
+    echo "deb [trusted=yes] https://mirrors.aliyun.com/debian bookworm main contrib non-free non-free-firmware" > /etc/apt/sources.list && \
+    echo "deb [trusted=yes] https://mirrors.aliyun.com/debian bookworm-updates main contrib non-free non-free-firmware" >> /etc/apt/sources.list && \
+    echo "deb [trusted=yes] https://mirrors.aliyun.com/debian bookworm-backports main contrib non-free non-free-firmware" >> /etc/apt/sources.list && \
+    echo "deb [trusted=yes] https://mirrors.aliyun.com/debian-security bookworm-security main contrib non-free non-free-firmware" >> /etc/apt/sources.list && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+RUN apt-get update && apt-get install -y wget unzip openssh-client procps nodejs npm 
+
+# Config Env
+RUN npx playwright install chrome
+RUN npm install @playwright/mcp @negokaz/excel-mcp-server
+
+# GAIA Docker Image
+FROM base AS gaia_env
+
+# Install aworld
+COPY ./aworld aworld
+COPY ./setup.py setup.py
+RUN python setup.py install
+
+# Install mcp servers requirements
+COPY ./mcp_servers mcp_servers
+RUN pip install -r mcp_servers/requirements.txt
+
+# Copy examples
+COPY ./examples/ examples
+
+# Create GAIA benchmark directory
+RUN mkdir -p gaia-benchmark/fs && \
+    mkdir -p gaia-benchmark/logs && \
+    mkdir -p static
+
+ENV PYTHONPATH=/app/aworld${PYTHONPATH:+:${PYTHONPATH}}
+ENV GAIA_DATASET_PATH=/app/aworld/examples/gaia/GAIA/2023
+ENV LOG_FILE_PATH=/app/aworld/gaia-benchmark/logs
+
+CMD [ "python", "examples/gaia/gaia_agent_server.py" ]

Dockerfile_ray

@@ -0,0 +1,43 @@
+FROM rayproject/ray:latest-py312-cpu as base
+
+USER root
+
+RUN pip config set global.index-url https://mirrors.aliyun.com/pypi/simple/
+
+RUN wget --quiet https://deb.nodesource.com/setup_lts.x && \
+    bash setup_lts.x && \
+    rm -fv setup_lts.x && \
+    apt-get update
+
+RUN apt-get install -y procps iputils-ping nodejs build-essential
+
+RUN apt-get clean
+
+FROM base as ray
+
+USER ray
+
+RUN mkdir -p /home/ray/aworld
+
+WORKDIR /home/ray/aworld
+
+COPY ./aworld aworld
+COPY ./mcp_servers mcp_servers
+COPY ./setup.py setup.py
+
+RUN pip install -e .
+
+RUN pip install -r ./mcp_servers/requirements.txt 
+
+FROM ray
+
+COPY ./examples/ examples
+
+RUN pip install -r ./examples/web/agent_deploy/ray_agent/requirements.txt
+
+# Fix numpy ray compatibility issues
+RUN pip install -U "numpy>=1.19,<2.0" "pyarrow>=14.0.2" "grpcio==1.66.2"
+
+ENV RAY_memory_monitor_refresh_ms=0
+ENV RAY_object_store_memory=1000000000
+ENV RAY_memory_usage_threshold=0.9

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 [Ant Group]
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

README-docker.md

@@ -0,0 +1,62 @@
+# Running GAIA Benchmark in Docker
+
+## Prerequisites
+
+1. **Docker Installation**
+   - Ensure Docker is installed and running on your machine
+   - Verify installation:
+     ```bash
+     docker ps
+     ```
+
+2. **Repository Setup**
+   - Clone the repository:
+     ```bash
+     git clone https://github.com/inclusionAI/AWorld
+     cd AWorld
+     ```
+
+3. **Dataset Preparation**
+   - Download the GAIA dataset from [Hugging Face](https://huggingface.co/datasets/gaia-benchmark/GAIA)
+   - Place it in the correct directory:
+     ```bash
+     git clone [email protected]:datasets/gaia-benchmark/GAIA examples/gaia/GAIA
+     ```
+   - ⚠️ **Note**: You need to configure [Hugging Face SSH Keys](https://huggingface.co/settings/keys) to access the GAIA repository
+
+4. **API Configuration**
+   - Set up your environment variables:
+     ```bash
+     cp .env.template .env
+     ```
+   - Edit `.env` file and replace all `{YOUR_CONFIG}` placeholders with your actual values
+
+## Running the Benchmark
+
+1. **Start the Container**
+   - Build and run the GAIA container:
+     ```bash
+     sh run-gaia.sh
+     ```
+   - Wait for the build to complete and the container to start
+
+2. **Access the Interface**
+   - Open your browser and navigate to [http://127.0.0.1:8080](http://127.0.0.1:8080)
+   - Register a new account
+   - Log in to your account
+
+3. **Run the Benchmark**
+   - Select `gaia_agent` from the top menu
+   - Choose a GAIA question from the list
+   - Click the send button to start the benchmark
+
+## Troubleshooting
+
+   - If you encounter any issues with Docker, ensure it's running properly
+   - For dataset access problems, verify your Hugging Face SSH keys are correctly configured
+   - Check the container logs if the interface is not accessible
+
+## Additional Resources
+
+   - [GAIA Benchmark Documentation](https://huggingface.co/datasets/gaia-benchmark/GAIA)
+   - [Hugging Face SSH Keys Setup Guide](https://huggingface.co/settings/keys)

README-local.md

@@ -0,0 +1,61 @@
+# Running AWorld Agent
+
+## Prerequisites
+
+1. **Conda Environment**
+   - Ensure Conda is installed and configured on your machine
+   - Create a Conda environment:
+     ```bash
+     conda create -n aworld python=3.11
+     conda activate aworld
+     ```
+
+2. **Clone Repository**
+   - Clone the repository and switch to the main branch:
+     ```bash
+     git clone https://github.com/inclusionAI/AWorld
+     cd AWorld
+     ```
+
+3. **Dataset Preparation**
+   - Download the GAIA dataset from [Hugging Face](https://huggingface.co/datasets/gaia-benchmark/GAIA)
+   - Place it in the correct directory:
+     ```bash
+     git clone [email protected]:datasets/gaia-benchmark/GAIA examples/gaia/GAIA
+     ```
+   - ⚠️ **Note**: You need to configure [Hugging Face SSH Keys](https://huggingface.co/settings/keys) to access the GAIA repository
+
+4. **API Configuration**
+   - Set up your environment variables:
+     ```bash
+     cp .env.template .env
+     ```
+   - Edit the `.env` file and replace all `{YOUR_CONFIG}` placeholders with your actual values
+
+## Running the Agent
+
+1. **Start the Web Server**
+   - Build and run the AWorld Agent:
+     ```bash
+     sh run-server.sh
+     ```
+   - Wait for the installation to complete
+
+2. **Access the Interface**
+   - Open your browser and navigate to [http://127.0.0.1:8000](http://127.0.0.1:8000)
+
+## Troubleshooting
+
+   - For dataset access problems, verify that your Hugging Face SSH keys are correctly configured
+   - Set up a pip mirror if necessary
+
+## Develop Your Own Agent
+
+   - Copy `examples/web/agent_deploy/weather_agent` to `examples/web/agent_deploy/{YOUR_AGENT_NAME}`
+   - Write your code in `examples/web/agent_deploy/{YOUR_AGENT_NAME}/agent.py`
+   - Configure the MCP Server in `examples/web/agent_deploy/{YOUR_AGENT_NAME}/mcp.json`
+
+## Additional Resources
+
+   - [GAIA Benchmark Documentation](https://huggingface.co/datasets/gaia-benchmark/GAIA)
+   - [Hugging Face SSH Keys Setup Guide](https://huggingface.co/settings/keys)

docker-compose-gaia.yml

@@ -0,0 +1,40 @@
+services:
+  openwebui:
+    image: ghcr.io/open-webui/open-webui:main
+    ports:
+      - 127.0.0.1:8080:8080
+    restart: on-failure:3
+    volumes:
+      - ./examples/gaia/openwebui-patch/ca81bd47c050_add_config_table.py:/app/backend/open_webui/migrations/versions/ca81bd47c050_add_config_table.py
+      - ./examples/gaia/GAIA:/app/aworld/examples/gaia/GAIA
+    environment:
+      - HF_HUB_OFFLINE=1
+      - ENABLE_OLLAMA_API=False
+      - ENABLE_DIRECT_CONNECTIONS=False
+      - ENABLE_EVALUATION_ARENA_MODELS=False
+      - ENABLE_TITLE_GENERATION=False
+      - ENABLE_AUTOCOMPLETE_GENERATION=False
+      - ENABLE_TAGS_GENERATION=False
+      - OPENAI_API_BASE_URL=http://gaia_agent:8888/v1
+      - OPENAI_API_KEY=xxx
+    deploy:
+      resources:
+        limits:
+          memory: 2G
+
+  gaia_agent:
+    build:
+      context: .
+      dockerfile: Dockerfile_gaia
+    platform: linux/amd64
+    ports:
+      - 127.0.0.1:8888:8888
+    volumes:
+      - ./.env:/app/aworld/.env
+    env_file:
+      - ./.env
+    restart: on-failure:3
+    deploy:
+      resources:
+        limits:
+          memory: 4G

docker-compose-ray.yml

@@ -0,0 +1,78 @@
+services:
+  ray-head:
+    build:
+      context: .
+      dockerfile: Dockerfile_ray
+    platform: linux/amd64
+    container_name: ray-head
+    hostname: ray-head
+    ports:
+      - "8265:8265" # Ray Dashboard
+      - "10001:10001" # Ray Client Server
+      - "6379:6379" # Redis port for Ray GCS
+    command: >
+      ray start --head  --dashboard-host=0.0.0.0  --dashboard-port=8265 --port=6379 --redis-password="" --object-manager-port=8076 --node-manager-port=8077 --min-worker-port=10002 --max-worker-port=19999 --block
+    env_file:
+      - .env
+    networks:
+      - ray-network
+    restart: unless-stopped
+    deploy:
+      resources:
+        limits:
+          memory: 8G
+        reservations:
+          memory: 8G
+
+  ray-worker:
+    build:
+      context: .
+      dockerfile: Dockerfile_ray
+    platform: linux/amd64
+    container_name: ray-worker
+    hostname: ray-worker
+    depends_on:
+      - ray-head
+    command: >
+      ray start  --address=ray-head:6379 --object-manager-port=8076 --node-manager-port=8077 --min-worker-port=10002 --max-worker-port=19999 --block
+    env_file:
+      - .env
+    networks:
+      - ray-network
+    restart: unless-stopped
+    deploy:
+      resources:
+        limits:
+          memory: 4G
+        reservations:
+          memory: 4G
+
+  aworld-web-server:
+    build:
+      context: .
+      dockerfile: Dockerfile_ray
+    platform: linux/amd64
+    container_name: aworld-api-server
+    hostname: aworld-api-server
+    depends_on:
+      - ray-head
+    command: aworld web --port 8000
+    env_file:
+      - .env
+    networks:
+      - ray-network
+    restart: unless-stopped
+    deploy:
+      resources:
+        limits:
+          memory: 4G
+        reservations:
+          memory: 4G
+
+networks:
+  ray-network:
+    driver: bridge
+
+volumes:
+  workspace:
+    driver: local

run-gaia.sh

@@ -0,0 +1,19 @@
+#!/bin/bash
+cd "$(dirname "$0")"
+
+# Check your env file
+if [ ! -f ".env" ]; then
+    echo "Please add your own .env config file from template .env.template before running gaia test!"
+    exit 1
+fi
+
+# Check GAIA dataset
+if [ ! -d "examples/gaia/GAIA" ]; then
+    echo "Please download GAIA dataset from https://huggingface.co/datasets/gaia-benchmark/GAIA and put it in examples/gaia/GAIA"
+    exit 1
+fi
+
+# Build docker image
+docker compose -f docker-compose-gaia.yml build && \
+  docker compose -f docker-compose-gaia.yml up -d && \
+  docker compose -f docker-compose-gaia.yml logs -f

run-ray.sh

@@ -0,0 +1,6 @@
+#!/bin/bash
+cd "$(dirname "$0")"
+
+docker compose -f docker-compose-ray.yml build && \
+  docker compose -f docker-compose-ray.yml up -d && \
+  docker compose -f docker-compose-ray.yml logs -f

run-server.sh

@@ -0,0 +1,8 @@
+#!/bin/bash
+cd "$(dirname "$0")"
+
+pip install -e . && \
+
+pip install -r mcp_servers/requirements.txt && \
+
+cd examples/cmd && aworld web_legacy

run-web.sh

@@ -0,0 +1,8 @@
+#!/bin/bash
+cd "$(dirname "$0")"
+
+pip install -e . && \
+
+pip install -r mcp_servers/requirements.txt && \
+
+cd examples/cmd && aworld web

setup.py

@@ -0,0 +1,253 @@
+# coding: utf-8
+# Copyright (c) 2025 inclusionAI.
+import logging
+import os
+import sys
+import re
+import subprocess
+
+from setuptools import setup, find_packages
+from setuptools.command.sdist import sdist
+from setuptools.command.install import install
+from setuptools.dist import Distribution
+
+from aworld.version_gen import __version__
+
+logger = logging.getLogger("setup")
+
+version_template = """
+# auto generated
+class VersionInfo(object):
+    @property
+    def build_date(self):
+      return "{BUILD_DATE}"
+
+    @property
+    def version(self):
+      return "{BUILD_VERSION}"
+
+    @property
+    def build_user(self):
+      return "{BUILD_USER}"
+"""
+
+
+def check_output(cmd):
+    import subprocess
+
+    output = subprocess.check_output(cmd)
+    return output.decode("utf-8")
+
+
+def get_build_date():
+    import datetime
+    import time
+
+    ts = time.time()
+    return datetime.datetime.fromtimestamp(ts).strftime("%Y-%m-%d %H:%M:%S")
+
+
+def build_version_template():
+    import getpass
+
+    return version_template.format(
+        BUILD_USER=getpass.getuser(),
+        BUILD_VERSION=__version__,
+        BUILD_DATE=get_build_date(),
+    )
+
+
+def call_process(cmd, raise_on_error=True, logging=True):
+    if isinstance(cmd, str):
+        shell = True
+    else:
+        shell = False  # cmd should be list of args
+
+    try:
+        subprocess.check_call(cmd, shell=shell, timeout=60)
+    except subprocess.CalledProcessError as e:
+        if raise_on_error:
+            raise e
+        logger.error(f"Fail to execute {cmd}, {e}")
+        return e.returncode
+
+    if logging:
+        logger.info(f"Successfully execute: {cmd}")
+    return 0
+
+
+class AWorldPackage(sdist):
+    def run(self):
+        from aworld.version_gen import generate_version_info
+
+        home = os.path.join(os.path.dirname(__file__), "aworld")
+        with open(os.path.join(home, "version.py"), "w") as f:
+            version_info = build_version_template()
+            f.write(version_info)
+
+        generate_version_info(scenario="AWORLD_SDIST")
+        sdist.run(self)
+
+
+class AWorldInstaller(install):
+    EXTRA_ENV = "AWORLD_EXTRA"
+    BASE = "framework"
+    BASE_OPT = "optional"
+
+    def __init__(self, *args, **kwargs):
+        super(AWorldInstaller, self).__init__(*args, **kwargs)
+        self._requirements = parse_requirements("aworld/requirements.txt")
+        self._extra = os.getenv(self.EXTRA_ENV)
+        logger.info(f"{os.getcwd()}: Install AWORLD using extra: {self._extra}")
+
+    def run(self):
+        # 1. build wheel using this setup.py, thus using the right install_requires according to ALPS_EXTRA
+        # 2. install this wheel into pip
+        install.run(self)
+
+        reqs = self._requirements.get(self.BASE, [])
+        self._install_reqs(reqs, ignore_error=True)
+
+        # install optional requirements here since pip install doesn't ignore requirement error
+        reqs = self._requirements.get(self.BASE_OPT, [])
+        self._install_reqs(reqs, ignore_error=True)
+
+    def _contains_module(self, module):
+        if self._extra is None:
+            return False
+
+        modules = [mod.strip() for mod in self._extra.split(",")]
+        try:
+            modules.index(module)
+            return True
+        except ValueError:
+            return False
+
+    @staticmethod
+    def _install_reqs(reqs, ignore_error=False, no_deps=False):
+        info = "--no-deps" if no_deps else ""
+        if ignore_error:
+            # install requirements one by one
+            for req in reqs:
+                try:
+                    cmd = f"{sys.executable} -m pip install {info} {req}"
+                    call_process(cmd)
+                    logger.info(f"Installing optional package {req} have succeeded.")
+                except:
+                    logger.warning(
+                        f"Installing optional package {req} is failed, Ignored."
+                    )  # ignore
+        elif reqs:
+            cmd = f"{sys.executable} -m pip install {info} {' '.join(reqs)}"
+            call_process(cmd)
+            logger.info(f"Packages {str(reqs)} have been installed.")
+
+
+def parse_requirements(req_fname):
+    requirements = {}
+    module_name = "unknown"
+    for line in open(req_fname, "r"):
+        match = re.match(r"#+\s+\[(\w+)\]\s+#+", line.strip())
+        if match:
+            # the beginning of a module
+            module_name = match.group(1)
+        else:
+            req = line.strip()
+            if not req or req.startswith("#"):
+                continue
+
+            # it's a requirement, strip trailing comments
+            pos = req.find("#")
+            if pos > 0:
+                req = req[:pos]
+                req = req.strip()
+
+            if module_name not in requirements:
+                requirements[module_name] = []
+            requirements[module_name].append(req)
+
+    return requirements
+
+
+def get_install_requires(extra, requirements):
+    modules = [AWorldInstaller.BASE]
+    if extra is None:
+        # old style of `pip install alps`, install all requirements for compatibility
+        for mod in requirements:
+            if mod in [AWorldInstaller.BASE, AWorldInstaller.BASE_OPT]:
+                continue
+            modules.append(mod)
+    else:
+        for mod in extra.split(","):
+            mod = mod.strip()
+            if mod != AWorldInstaller.BASE:
+                modules.append(mod)
+
+    install_reqs = []
+    for mod in modules:
+        install_reqs.extend(requirements.get(mod, []))
+    return install_reqs
+
+
+def get_python_requires():
+    return ">=3.11"
+
+
+class BinaryDistribution(Distribution):
+    """This class is needed in order to create OS specific wheels."""
+
+    @staticmethod
+    def has_ext_modules():
+        return True
+
+
+requirements = parse_requirements("aworld/requirements.txt")
+extra = os.getenv(AWorldInstaller.EXTRA_ENV, None)
+
+setup(
+    name="aworld",
+    version=__version__,
+    description="Ant Agent Package",
+    url="https://github.com/inclusionAI/AWorld",
+    author="Ant AI",
+    author_email="",
+    long_description="",
+    long_description_content_type="text/markdown",
+    packages=find_packages(
+        where=".",
+        exclude=["tests", "tests.*", "*.tests", "*.tests.*", "*.test", "*.test.*"],
+    ),
+    package_data={
+        "aworld": [
+            "virtual_environments/browsers/script/*.js",
+            "dataset/gaia/gaia.npy",
+            "requirements.txt",
+            "config/*.yaml",
+            "config/*.json",
+            "config/*.tiktoken",
+            "web/templates/*.html",
+        ]
+    },
+    license="MIT",
+    platforms=["any"],
+    keywords=["multi-agent", "agent", "environment", "tool", "sandbox"],
+    cmdclass={
+        "sdist": AWorldPackage,
+        "install": AWorldInstaller,
+    },
+    install_requires=get_install_requires(extra, requirements),
+    python_requires=get_python_requires(),
+    classifiers=[
+        "Development Status :: 5 - Production/Stable",
+        # Indicate who your project is intended for
+        "Intended Audience :: Developers",
+        "Topic :: Software Development :: Build Tools",
+        "Programming Language :: Python :: 3.11",
+        "Programming Language :: Python :: 3.12",
+    ],
+    entry_points={
+        "console_scripts": [
+            "aworld = aworld.__main__:main",
+        ]
+    },
+)