Upload 4 files

aworld/__init__.py

@@ -0,0 +1,15 @@
+# coding: utf-8
+# Copyright (c) 2025 inclusionAI.
+import os
+
+try:
+    from aworld.utils.import_package import import_package
+
+    import_package("dotenv", install_name="python-dotenv")
+    from dotenv import load_dotenv
+
+    sucess = load_dotenv()
+    if not sucess:
+        load_dotenv(os.path.join(os.getcwd(), ".env"))
+except Exception as e:
+    print(e)

aworld/__main__.py

@@ -0,0 +1,7 @@
+# coding: utf-8
+# Copyright (c) 2025 inclusionAI.
+
+from .cmd.cli import main
+
+if __name__ == "__main__":
+    main()

aworld/runner.py

@@ -0,0 +1,106 @@
+# coding: utf-8
+# Copyright (c) 2025 inclusionAI.
+import asyncio
+from typing import List, Dict, Union
+
+from aworld.config import RunConfig
+from aworld.config.conf import TaskConfig
+from aworld.agents.llm_agent import Agent
+from aworld.core.agent.swarm import Swarm
+from aworld.core.common import Config
+from aworld.core.task import Task, TaskResponse, Runner
+from aworld.output import StreamingOutputs
+from aworld.runners.utils import choose_runners, execute_runner
+from aworld.utils.common import sync_exec
+
+
+class Runners:
+    """Unified entrance to the utility class of the runnable task of execution."""
+
+    @staticmethod
+    def streamed_run_task(task: Task) -> StreamingOutputs:
+        """Run the task in stream output."""
+        if not task.conf:
+            task.conf = TaskConfig()
+
+        streamed_result = StreamingOutputs(
+            input=task.input,
+            usage={},
+            is_complete=False
+        )
+        task.outputs = streamed_result
+
+        streamed_result._run_impl_task = asyncio.create_task(
+            Runners.run_task(task)
+        )
+        return streamed_result
+
+    @staticmethod
+    async def run_task(task: Union[Task, List[Task]], run_conf: RunConfig = None) -> Dict[str, TaskResponse]:
+        """Run tasks for some complex scenarios where agents cannot be directly used.
+
+        Args:
+            task: User task define.
+            run_conf:
+        """
+        if isinstance(task, Task):
+            task = [task]
+
+            runners: List[Runner] = await choose_runners(task)
+            return await execute_runner(runners, run_conf)
+
+    @staticmethod
+    def sync_run_task(task: Union[Task, List[Task]], run_conf: Config = None) -> Dict[str, TaskResponse]:
+        return sync_exec(Runners.run_task, task=task, run_conf=run_conf)
+
+    @staticmethod
+    def sync_run(
+            input: str,
+            agent: Agent = None,
+            swarm: Swarm = None,
+            tool_names: List[str] = [],
+            session_id: str = None
+    ) -> TaskResponse:
+        return sync_exec(
+            Runners.run,
+            input=input,
+            agent=agent,
+            swarm=swarm,
+            tool_names=tool_names,
+            session_id=session_id
+        )
+
+    @staticmethod
+    async def run(
+            input: str,
+            agent: Agent = None,
+            swarm: Swarm = None,
+            tool_names: List[str] = [],
+            session_id: str = None
+    ) -> TaskResponse:
+        """Run agent directly with input and tool names.
+
+        Args:
+            input: User query.
+            agent: An agent with AI model configured, prompts, tools, mcp servers and other agents.
+            swarm: Multi-agent topo.
+            tool_names: Tool name list.
+            session_id: Session id.
+
+        Returns:
+            TaskResponse: Task response.
+        """
+        if agent and swarm:
+            raise ValueError("`agent` and `swarm` only choose one.")
+
+        if not input:
+            raise ValueError('`input` is empty.')
+
+        if agent:
+            agent.task = input
+            swarm = Swarm(agent)
+
+        task = Task(input=input, swarm=swarm, tool_names=tool_names,
+                    event_driven=swarm.event_driven, session_id=session_id)
+        res = await Runners.run_task(task)
+        return res.get(task.id)

aworld/version_gen.py

@@ -0,0 +1,36 @@
+# coding: utf-8
+# Copyright (c) 2025 inclusionAI.
+
+import os
+import getpass
+import time
+
+__version__ = '0.2.2'
+
+version_template = \
+    """# auto generated
+class VersionInfo:
+    BUILD_DATE = "{BUILD_DATE}"
+    BUILD_VERSION = "{BUILD_VERSION}"
+    BUILD_USER = "{BUILD_USER}" 
+    SCENARIO = "{SCENARIO}"
+"""
+
+
+def generate_version_info(directory_path: str = None, scenario: str = "", version: str = None):
+    if directory_path is None:
+        directory_path = os.path.dirname(__file__)
+    with open(os.path.join(directory_path, "version_info.py"), "w") as f:
+        version_info = _build_version_template(scenario=scenario, version=version)
+        f.write(version_info)
+
+
+def _build_version_template(scenario: str = "", version: str = None) -> str:
+    if version is None:
+        version = __version__
+    return version_template.format(
+        BUILD_USER=getpass.getuser(),
+        BUILD_VERSION=version,
+        BUILD_DATE=time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()),
+        SCENARIO=scenario,
+    )