Upload 3 files

AWorld-main/aworlddistributed/client/aworld_client_run_gaia.py

@@ -0,0 +1,91 @@
+# Initialize AworldTaskClient with server endpoints
+import asyncio
+import logging
+from datetime import datetime
+import os
+import random
+import uuid
+
+from aworld.utils.common import get_local_ip
+
+from client.aworld_client import AworldTask, AworldTaskClient
+
+AWORLD_TASK_CLIENT = AworldTaskClient(
+    know_hosts = ["localhost:9999"]
+)
+
+
+async def _run_gaia_task(gaia_task: AworldTask, delay: int, background: bool = False) -> None:
+    """Run a single Gaia task with the given question ID.
+
+    Args:
+        gaia_task_id: The ID of the question to process
+    """
+    global AWORLD_TASK_CLIENT
+    await asyncio.sleep(delay)
+
+    # Submit task to Aworld server
+    await AWORLD_TASK_CLIENT.submit_task(gaia_task, background=background)
+
+    # Get and print task result
+    task_result = await AWORLD_TASK_CLIENT.get_task_state(task_id=gaia_task.task_id)
+    if not background:
+        logging.info(f"execute task_result#{gaia_task.task_id} is {task_result.data if task_result else None}")
+    else:
+        logging.info(f"submit task_result#{gaia_task.task_id} background success, please use task_id get task_result await a moment")
+
+
+
+async def _batch_run_gaia_task(gaia_tasks: list[AworldTask]) -> None:
+    """Run multiple Gaia tasks in parallel.
+
+    """
+    tasks = [
+        _run_gaia_task(gaia_task, index * 3, background=True)
+        for index, gaia_task in enumerate(gaia_tasks)
+    ]
+    await asyncio.gather(*tasks)
+
+
+CUSTOM_SYSTEM_PROMPT = f""" **PLEASE CUSTOM IT **"""
+
+if __name__ == '__main__':
+    gaia_task_ids = ['c61d22de-5f6c-4958-a7f6-5e9707bd3466']
+    gaia_tasks = []
+    custom_mcp_servers = [
+            # "e2b-server",
+            "e2b-code-server",
+            "terminal-controller",
+            "excel",
+            # "filesystem",
+            "calculator",
+            "ms-playwright",
+            "audio_server",
+            "image_server",
+            "google-search",
+            # "video_server",
+            # "search_server",
+            # "download_server",
+            # "document_server",
+            # "youtube_server",
+            # "reasoning_server",
+        ]
+
+    for gaia_task_id in gaia_task_ids:
+        task_id = datetime.now().strftime("%Y%m%d%H%M%S") + "_" + gaia_task_id + "_" + str(uuid.uuid4())
+        gaia_tasks.append(
+            AworldTask(
+                task_id=task_id,
+                agent_id="gaia_agent",
+                agent_input=gaia_task_id,
+                session_id="session_id",
+                user_id=os.getenv("USER", "SYSTEM"),
+                client_id=get_local_ip(),
+                mcp_servers=custom_mcp_servers,
+                max_retries=5,
+                llm_custom_input="你好"
+                # llm_model_name="gpt-4o",
+                # task_system_prompt=CUSTOM_SYSTEM_PROMPT
+            )
+        )
+    asyncio.run(_batch_run_gaia_task(gaia_tasks))

AWorld-main/aworlddistributed/client/aworld_client_run_web.py

@@ -0,0 +1,55 @@
+# Initialize AworldTaskClient with server endpoints
+import asyncio
+import random
+import uuid
+
+from base import AworldTask
+from client.aworld_client import AworldTaskClient
+
+AWORLD_TASK_CLIENT = AworldTaskClient(
+    know_hosts=["localhost:9999"]
+)
+
+
+async def _run_web_task(web_question_id: str) -> None:
+    """Run a single Web task with the given question ID.
+
+    Args:
+        web_question_id: The ID of the question to process
+    """
+    global AWORLD_TASK_CLIENT
+    task_id = str(uuid.uuid4())
+
+    # Submit task to Aworld server
+    await AWORLD_TASK_CLIENT.submit_task(
+        AworldTask(
+            task_id=task_id,
+            agent_id="playwright_agent",
+            agent_input=web_question_id,
+            session_id="session_id",
+            user_id="SYSTEM"
+        )
+    )
+
+    # Get and print task result
+    task_result = await AWORLD_TASK_CLIENT.get_task_state(task_id=task_id)
+    print(task_result)
+
+
+async def _batch_run_web_task(start_i: int, end_i: int) -> None:
+    """Run multiple Web tasks in parallel.
+
+    Args:
+        start_i: Starting question ID
+        end_i: Ending question ID
+    """
+    tasks = [
+        _run_web_task(str(i))
+        for i in range(start_i, end_i + 1)
+    ]
+    await asyncio.gather(*tasks)
+
+
+if __name__ == '__main__':
+    # Run batch processing for questions 1-5
+    asyncio.run(_batch_run_web_task(25, 25))

AWorld-main/aworlddistributed/client/aworld_download_background_tasks.py

@@ -0,0 +1,28 @@
+import asyncio
+
+from client.aworld_client import AworldTaskClient
+
+
+async def download_with_timerange(know_hosts: list[str], start_time, end_time, save_path):
+    # create client
+    client = AworldTaskClient(know_hosts = know_hosts)
+
+    # 1. download task results to file
+    file_path = await client.download_task_results(
+        start_time=start_time,
+        end_time=end_time,
+        save_path=save_path
+    )
+
+    # 2. parse local jsonl file
+    local_results = client.parse_task_results_file(save_path)
+
+    # 3. analyze results data
+    for result in local_results:
+        print(f"Submit User ID: {result['user_id']}, Task ID: {result['task_id']},Status: {result['status']}, Replays: {result['result_data']['replays_file'] if result['result_data'] else ''}")
+
+if __name__ == '__main__':
+    asyncio.run(download_with_timerange(know_hosts= ["http://localhost:9999"],
+                    start_time="2025-06-12 00:00:00",
+                    end_time="2025-06-12 23:59:59",
+                    save_path="results/january_tasks.jsonl"))