Upload 3 files

aworld/cmd/__init__.py

@@ -0,0 +1,22 @@
+# coding: utf-8
+# Copyright (c) 2025 inclusionAI.
+
+from .data_model import (
+    BaseAWorldAgent,
+    ChatCompletionRequest,
+    ChatCompletionResponse,
+    ChatCompletionChoice,
+    ChatCompletionMessage,
+    AgentModel,
+    SessionModel,
+)
+
+__all__ = [
+    "BaseAWorldAgent",
+    "ChatCompletionRequest",
+    "ChatCompletionResponse",
+    "ChatCompletionChoice",
+    "ChatCompletionMessage",
+    "AgentModel",
+    "SessionModel",
+]

aworld/cmd/cli.py

@@ -0,0 +1,47 @@
+import click
+
+
+
+@click.group()
+def main(*args, **kwargs):
+    print(
+        """\
+    AWorld CLI Help:
+        aworld web: run aworld web ui server
+        aworld api: run aworld api server
+        aworld web_legacy: run aworld web agent (legacy), Streamlit Web UI
+        aworld help: show help"""
+    )
+
+
+@main.command("web")
+@click.option(
+    "--port", type=int, default=8000, help="Port to run the AWorld api server"
+)
+@click.argument("args", nargs=-1)
+def main_web(port, args=None, **kwargs):
+    from .web import web_server
+    web_server.run_server(port, args, **kwargs)
+
+
+@main.command("api")
+@click.option(
+    "--port", type=int, default=8000, help="Port to run the AWorld api server"
+)
+@click.argument("args", nargs=-1)
+def main_api(port, args=None, **kwargs):
+    from .web import api_server
+    api_server.run_server(port, args, **kwargs)
+
+
+@main.command("web_legacy")
+@click.option(
+    "--port", type=int, default=8000, help="Port to run the AWorld agent web app"
+)
+@click.argument("args", nargs=-1)
+def main_web_legacy(port, args=None, **kwargs):
+    from .web_legacy import web_server
+    web_server.run_web_server(port, args, **kwargs)
+
+if __name__ == "__main__":
+    main()

aworld/cmd/data_model.py

@@ -0,0 +1,78 @@
+import datetime
+import uuid
+from abc import abstractmethod
+from typing import Any, AsyncGenerator, List, Optional
+from pydantic import BaseModel, Field
+
+from aworld.output.base import Output
+
+
+class ChatCompletionMessage(BaseModel):
+    role: str = Field(..., description="The role of the message")
+    content: str = Field(..., description="The content of the message")
+    trace_id: str = Field(None, description="The trace id")
+
+
+class ChatCompletionRequest(BaseModel):
+    user_id: str = Field(None, description="The user id")
+    session_id: str = Field(
+        None,
+        description="The session id, if not provided, a new session will be created",
+    )
+    query_id: str = Field(None, description="The query id")
+    trace_id: str = Field(None, description="The trace id")
+    model: str = Field(..., description="The model to use")
+    messages: List[ChatCompletionMessage] = Field(
+        ..., description="The messages to send to the agent"
+    )
+
+
+class ChatCompletionChoice(BaseModel):
+    index: int = 0
+    delta: ChatCompletionMessage = Field(
+        ..., description="The delta message from the agent"
+    )
+
+
+class ChatCompletionResponse(BaseModel):
+    object: str = "chat.completion.chunk"
+    id: str = uuid.uuid4().hex
+    choices: List[ChatCompletionChoice] = Field(
+        ..., description="The choices from the agent"
+    )
+
+
+class AgentModel(BaseModel):
+    id: str = Field(..., description="The agent id")
+    name: Optional[str] = Field(None, description="The agent name")
+    description: Optional[str] = Field(None, description="The agent description")
+    path: str = Field(..., description="The agent path")
+    instance: Any = Field(..., description="The agent module instance", exclude=True)
+
+
+class BaseAWorldAgent:
+    @abstractmethod
+    def name(self) -> str:
+        pass
+
+    @abstractmethod
+    def description(self) -> str:
+        pass
+
+    @abstractmethod
+    async def run(
+        self, prompt: str = None, request: ChatCompletionRequest = None
+    ) -> AsyncGenerator[Output, None]:
+        pass
+
+
+class SessionModel(BaseModel):
+    user_id: str = Field(..., description="The user id")
+    session_id: str = Field(..., description="The session id")
+    name: str = Field(None, description="The session name")
+    description: str = Field(None, description="The session description")
+    created_at: datetime.datetime = Field(None, description="The session created at")
+    updated_at: datetime.datetime = Field(None, description="The session updated at")
+    messages: List[ChatCompletionMessage] = Field(
+        None, description="The messages in the session"
+    )